From eb4fe931156279575531c1db6abe3480ac490f33 Mon Sep 17 00:00:00 2001 From: Jake Taylor <jake@elixxir.io> Date: Tue, 27 Apr 2021 13:06:33 -0500 Subject: [PATCH] fix test --- network/gateway/hostpool_test.go | 28 ++++++++++------------------ 1 file changed, 10 insertions(+), 18 deletions(-) diff --git a/network/gateway/hostpool_test.go b/network/gateway/hostpool_test.go index 59530e69e..8181e6d49 100644 --- a/network/gateway/hostpool_test.go +++ b/network/gateway/hostpool_test.go @@ -589,32 +589,24 @@ func TestHostPool_ForceAdd(t *testing.T) { t.Fatalf("Failed to create mock host pool: %v", err) } - // Construct a list of new gateways to add - newGatewayLen := 10 - newGateways := make([]*id.ID, newGatewayLen) - for i := 0; i < newGatewayLen; i++ { - gwId := id.NewIdFromUInt(uint64(100+i), id.Gateway, t) - // Add mock gateway to manager - _, err = manager.AddHost(gwId, "", nil, connect.GetDefaultHostParams()) - if err != nil { - t.Fatalf("Could not add mock host to manager: %v", err) - } - newGateways[i] = gwId + // Construct a new gateway to add + gwId := id.NewIdFromUInt(uint64(100), id.Gateway, t) + // Add mock gateway to manager + _, err = manager.AddHost(gwId, "", nil, connect.GetDefaultHostParams()) + if err != nil { + t.Fatalf("Could not add mock host to manager: %v", err) } - // forceAdd list of gateways - err = testPool.forceAdd(newGateways) + // forceAdd gateway + err = testPool.forceAdd(gwId) if err != nil { t.Errorf("Could not add gateways: %v", err) } // check that gateways have been added to the map - for _, gw := range newGateways { - if _, ok := testPool.hostMap[*gw]; !ok { - t.Errorf("Failed to forcefully add new gateway ID: %v", gw) - } + if _, ok := testPool.hostMap[*gwId]; !ok { + t.Errorf("Failed to forcefully add new gateway ID: %v", gwId) } - } // Unit test which only adds information to ndf -- GitLab