From b2cfa3f98c7421b1f05af0ee625adf3b73acbfa6 Mon Sep 17 00:00:00 2001
From: benjamin <ben@elixxir.io>
Date: Thu, 27 Oct 2022 15:03:53 -0700
Subject: [PATCH] fixed tests

---
 cmix/nodes/registrar_test.go | 4 ++--
 cmix/nodes/utils_test.go     | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/cmix/nodes/registrar_test.go b/cmix/nodes/registrar_test.go
index 922adfad6..55b684495 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 96eacda72..9f9657db1 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)
 	}
-- 
GitLab