From 262222589cce669b9a54560bae711718fd34848a Mon Sep 17 00:00:00 2001
From: josh <josh@elixxir.io>
Date: Thu, 19 May 2022 13:28:46 -0700
Subject: [PATCH] Update tests in UD/

---
 ud/lookup_test.go |  2 +-
 ud/search_test.go |  2 +-
 ud/utils_test.go  | 10 ++++++++--
 3 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/ud/lookup_test.go b/ud/lookup_test.go
index 23311707d..132905cc0 100644
--- a/ud/lookup_test.go
+++ b/ud/lookup_test.go
@@ -83,7 +83,7 @@ func TestManager_Lookup(t *testing.T) {
 	p := single.RequestParams{
 		Timeout:             timeout,
 		MaxResponseMessages: 1,
-		CmixParam:           cmix.GetDefaultCMIXParams(),
+		CmixParams:          cmix.GetDefaultCMIXParams(),
 	}
 
 	// Run the lookup
diff --git a/ud/search_test.go b/ud/search_test.go
index b9bcf5b09..4bbc98e44 100644
--- a/ud/search_test.go
+++ b/ud/search_test.go
@@ -68,7 +68,7 @@ func TestManager_Search(t *testing.T) {
 	p := single.RequestParams{
 		Timeout:             timeout,
 		MaxResponseMessages: 1,
-		CmixParam:           cmix.GetDefaultCMIXParams(),
+		CmixParams:          cmix.GetDefaultCMIXParams(),
 	}
 
 	_, _, err = Search(m.network, m.events, prng, m.e2e.GetGroup(),
diff --git a/ud/utils_test.go b/ud/utils_test.go
index 675acc924..d402f57df 100644
--- a/ud/utils_test.go
+++ b/ud/utils_test.go
@@ -78,7 +78,7 @@ func newTestManager(t *testing.T) (*Manager, *testNetworkManager) {
 
 	// Create our Manager object
 	m := &Manager{
-		e2e:    mockE2e{},
+		e2e:    mockE2e{grp: getGroup()},
 		events: event.NewEventManager(),
 		user:   mockUser{testing: t, key: key},
 		store:  udStore,
@@ -398,7 +398,13 @@ func (m mockComms) GetHost(hostId *id.ID) (*connect.Host, bool) {
 	return m.udHost, true
 }
 
-type mockE2e struct{}
+type mockE2e struct {
+	grp *cyclic.Group
+}
+
+func (m mockE2e) GetHistoricalDHPubkey() *cyclic.Int {
+	return m.grp.NewInt(6)
+}
 
 func (m mockE2e) GetReceptionID() *id.ID {
 	//TODO implement me
-- 
GitLab