From 72a62f1292037f0f82382a94d14d40ff62539a1f Mon Sep 17 00:00:00 2001 From: Roman Khimov Date: Fri, 10 Jan 2020 15:24:32 +0300 Subject: [PATCH] result: add trivial test for peers management functions --- pkg/rpc/result/peers_test.go | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 pkg/rpc/result/peers_test.go diff --git a/pkg/rpc/result/peers_test.go b/pkg/rpc/result/peers_test.go new file mode 100644 index 000000000..6e5a9339e --- /dev/null +++ b/pkg/rpc/result/peers_test.go @@ -0,0 +1,26 @@ +package result + +import ( + "testing" + + "github.com/stretchr/testify/require" +) + +func TestGetPeers(t *testing.T) { + gp := NewGetPeers() + require.Equal(t, 0, len(gp.Unconnected)) + require.Equal(t, 0, len(gp.Connected)) + require.Equal(t, 0, len(gp.Bad)) + + gp.AddUnconnected([]string{"1.1.1.1:53", "8.8.8.8:53", "9.9.9.9:53"}) + gp.AddConnected([]string{"192.168.0.1:10333"}) + gp.AddBad([]string{"127.0.0.1:20333"}) + + require.Equal(t, 3, len(gp.Unconnected)) + require.Equal(t, 1, len(gp.Connected)) + require.Equal(t, 1, len(gp.Bad)) + require.Equal(t, "192.168.0.1", gp.Connected[0].Address) + require.Equal(t, "10333", gp.Connected[0].Port) + require.Equal(t, "127.0.0.1", gp.Bad[0].Address) + require.Equal(t, "20333", gp.Bad[0].Port) +}