diff --git a/api/client/client_test.go b/api/client/client_test.go new file mode 100644 index 000000000..5d290e852 --- /dev/null +++ b/api/client/client_test.go @@ -0,0 +1,24 @@ +package client + +import ( + "reflect" + "testing" + + "github.com/harmony-one/harmony/p2p" +) + +func TestClient(t *testing.T) { + leaders := map[uint32]p2p.Peer{ + 0: p2p.Peer{ + IP: "127.0.0.1", + Port: "90", + }, + } + client := NewClient(nil, &leaders) + + leadersGot := client.GetLeaders() + + if len(leadersGot) != 1 || !reflect.DeepEqual(leaders[0], leadersGot[0]) { + t.Errorf("expected: %v, got: %v\n", leaders[0], leadersGot[0]) + } +}