diff --git a/cmix/nodes/registrar_test.go b/cmix/nodes/registrar_test.go index 922adfad68bc2b52f999a92561c834803be4f16b..55b684495decc278b4eb3147c3b0398676879901 100644 --- a/cmix/nodes/registrar_test.go +++ b/cmix/nodes/registrar_test.go @@ -38,7 +38,7 @@ func TestLoadRegistrar_New(t *testing.T) { nodeChan := make(chan commNetwork.NodeGateway, InputChanLen) r, err := LoadRegistrar(session, sender, &MockClientComms{}, - rngGen, nodeChan) + rngGen, nodeChan, func() int { return 100 }) if err != nil { t.Fatalf("Failed to create new registrar: %+v", err) } @@ -71,7 +71,7 @@ func TestLoadRegistrar_Load(t *testing.T) { // Load the store and check its attributes r, err := LoadRegistrar( - testR.session, testR.sender, testR.comms, testR.rng, testR.c) + testR.session, testR.sender, testR.comms, testR.rng, testR.c, func() int { return 100 }) if err != nil { t.Fatalf("Unable to load store: %+v", err) } diff --git a/cmix/nodes/utils_test.go b/cmix/nodes/utils_test.go index 96eacda72794e3f66c1b4513e7f359624cf0cbe1..9f9657db116623508c86db323425feaf293982a9 100644 --- a/cmix/nodes/utils_test.go +++ b/cmix/nodes/utils_test.go @@ -101,7 +101,7 @@ func makeTestRegistrar(mockComms *MockClientComms, t *testing.T) *registrar { nodeChan := make(chan commNetwork.NodeGateway, InputChanLen) r, err := LoadRegistrar( - session, sender, mockComms, rngGen, nodeChan) + session, sender, mockComms, rngGen, nodeChan, func() int { return 100 }) if err != nil { t.Fatalf("Failed to create new registrar: %+v", err) }