From e63b6531994adcf976d8e15c2b791682b8531e7d Mon Sep 17 00:00:00 2001
From: tobi <31960611+tsmethurst@users.noreply.github.com>
Date: Tue, 15 Mar 2022 15:01:19 +0100
Subject: [performance] Add dereference shortcuts to avoid making http calls to
 self (#430)

* update transport (controller) to allow shortcuts

* go fmt

* expose underlying sig transport to allow test sigs
---
 testrig/testmodels.go          | 8 ++++----
 testrig/transportcontroller.go | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

(limited to 'testrig')

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,
-- 
cgit v1.3