From c73e285ad13f6addf99e102fc3fc506556494edb Mon Sep 17 00:00:00 2001
From: Jake Taylor <jake@elixxir.io>
Date: Tue, 18 Aug 2020 13:56:12 -0500
Subject: [PATCH] fix tests

---
 client/mockServer_test.go | 9 ++++-----
 node/register_test.go     | 2 +-
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/client/mockServer_test.go b/client/mockServer_test.go
index 73079cac..3ec1f5dc 100644
--- a/client/mockServer_test.go
+++ b/client/mockServer_test.go
@@ -15,7 +15,6 @@ import (
 	pb "gitlab.com/elixxir/comms/mixmessages"
 	"gitlab.com/elixxir/comms/testutils"
 	"gitlab.com/xx_network/comms/connect"
-	"gitlab.com/xx_network/primitives/id"
 	"gitlab.com/xx_network/primitives/ndf"
 	"sync"
 )
@@ -53,8 +52,8 @@ func getNextAddress() string {
 type MockRegistration struct {
 }
 
-func (s *MockRegistration) RegisterNode(NodeID *id.ID,
-	NodeTLSCert, GatewayTLSCert, RegistrationCode, Addr, Addr2 string) error {
+func (s *MockRegistration) RegisterNode(salt []byte, serverAddr, serverTlsCert, gatewayAddr,
+	gatewayTlsCert, registrationCode string) error {
 	return nil
 }
 
@@ -85,8 +84,8 @@ func (s *MockRegistration) CheckRegistration(msg *pb.RegisteredNodeCheck) (*pb.R
 type MockRegistrationError struct {
 }
 
-func (s *MockRegistrationError) RegisterNode(NodeID *id.ID,
-	NodeTLSCert, GatewayTLSCert, RegistrationCode, Addr, Addr2 string) error {
+func (s *MockRegistrationError) RegisterNode(salt []byte, serverAddr, serverTlsCert, gatewayAddr,
+	gatewayTlsCert, registrationCode string) error {
 	return nil
 }
 
diff --git a/node/register_test.go b/node/register_test.go
index 2bee3309..f9f6f064 100644
--- a/node/register_test.go
+++ b/node/register_test.go
@@ -32,7 +32,7 @@ func TestSendNodeRegistration(t *testing.T) {
 		t.Errorf("Unable to call NewHost: %+v", err)
 	}
 
-	msgs := &pb.NodeRegistration{ID: testId.Bytes()}
+	msgs := &pb.NodeRegistration{Salt: testId.Bytes()}
 	err = server.SendNodeRegistration(host, msgs)
 	if err != nil {
 		t.Errorf("SendNodeTopology: Error received: %s", err)
-- 
GitLab