update collectionNode mock to use testing.TB interface

onflow · Jan 17, 2025 · e075836 · e075836
1 parent 97b7ba9
commit e075836
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/engine/access/rpc/connection/node_mock.go b/engine/access/rpc/connection/node_mock.go
@@ -95,16 +95,16 @@ func newCollectionNode(tb testing.TB) *collectionNode {
 	}
 }
 
-func (cn *collectionNode) start(t *testing.T) {
+func (cn *collectionNode) start(tb testing.TB) {
 	if cn.handler == nil {
-		t.Fatalf("collectionNode must be initialized using newCollectionNode")
+		tb.Fatalf("collectionNode must be initialized using newCollectionNode")
 	}
 
-	cn.setupNode(t)
+	cn.setupNode(tb)
 	access.RegisterAccessAPIServer(cn.server, cn.handler)
-	cn.node.start(t)
+	cn.node.start(tb)
 }
 
-func (cn *collectionNode) stop(t *testing.T) {
-	cn.node.stop(t)
+func (cn *collectionNode) stop(tb testing.TB) {
+	cn.node.stop(tb)
 }