diff options
Diffstat (limited to 'testrig')
| -rw-r--r-- | testrig/testmodels.go | 8 | ||||
| -rw-r--r-- | testrig/transportcontroller.go | 2 |
2 files changed, 5 insertions, 5 deletions
diff --git a/testrig/testmodels.go b/testrig/testmodels.go index b95f5fbb6..a959bac51 100644 --- a/testrig/testmodels.go +++ b/testrig/testmodels.go @@ -1749,8 +1749,8 @@ func GetSignatureForActivity(activity pub.Activity, pubKeyID string, privkey cry panic(err) } - // trigger the delivery function, which will trigger the 'do' function of the recorder above - if err := tp.Deliver(context.Background(), bytes, destination); err != nil { + // trigger the delivery function for the underlying signature transport, which will trigger the 'do' function of the recorder above + if err := tp.SigTransport().Deliver(context.Background(), bytes, destination); err != nil { panic(err) } @@ -1781,8 +1781,8 @@ func GetSignatureForDereference(pubKeyID string, privkey crypto.PrivateKey, dest panic(err) } - // trigger the delivery function, which will trigger the 'do' function of the recorder above - if _, err := tp.Dereference(context.Background(), destination); err != nil { + // trigger the dereference function for the underlying signature transport, which will trigger the 'do' function of the recorder above + if _, err := tp.SigTransport().Dereference(context.Background(), destination); err != nil { panic(err) } diff --git a/testrig/transportcontroller.go b/testrig/transportcontroller.go index 7408de967..90eab5ab3 100644 --- a/testrig/transportcontroller.go +++ b/testrig/transportcontroller.go @@ -39,7 +39,7 @@ import ( // PER TEST rather than per suite, so that the do function can be set on a test by test (or even more granular) // basis. func NewTestTransportController(client pub.HttpClient, db db.DB) transport.Controller { - return transport.NewController(db, &federation.Clock{}, client) + return transport.NewController(db, NewTestFederatingDB(db), &federation.Clock{}, client) } // NewMockHTTPClient returns a client that conforms to the pub.HttpClient interface, |