From fe7e08939d7741cf3ae8d2f804bb3b4975b7d0fb Mon Sep 17 00:00:00 2001 From: pavelkrolevets Date: Fri, 4 Oct 2024 13:02:34 +0300 Subject: [PATCH] update test --- integration_test/health_check_test.go | 31 +++++++++++++++++++++++++-- pkgs/initiator/initiator.go | 2 +- 2 files changed, 30 insertions(+), 3 deletions(-) diff --git a/integration_test/health_check_test.go b/integration_test/health_check_test.go index 192bf204..bea5ef1b 100644 --- a/integration_test/health_check_test.go +++ b/integration_test/health_check_test.go @@ -1,6 +1,8 @@ package integration_test import ( + "io" + "os" "strings" "testing" @@ -38,11 +40,36 @@ func TestHealthCheck(t *testing.T) { RootCmd.Short = "ssv-dkg-test" RootCmd.Version = version cli_initiator.HealthCheck.Version = version - t.Run("test 4 operators health check", func(t *testing.T) { - args := []string{"ping", "--ip", strings.Join(ips, ",")} + t.Run("test 1 operator health check: positive", func(t *testing.T) { + rescueStdout := os.Stdout + r, w, _ := os.Pipe() + os.Stdout = w + args := []string{"ping", "--ip", ips[0]} RootCmd.SetArgs(args) err := RootCmd.Execute() require.NoError(t, err) + w.Close() + out, _ := io.ReadAll(r) + os.Stdout = rescueStdout + t.Log(string(out)) + require.True(t, strings.Contains(string(out), "operator online and healthy: multisig ready 👌 and connected to ethereum network")) + resetFlags(RootCmd) + }) + t.Run("test 1 operator health check: negative", func(t *testing.T) { + servers[0].HttpSrv.Close() + rescueStdout := os.Stdout + r, w, _ := os.Pipe() + os.Stdout = w + args := []string{"ping", "--ip", ips[0]} + RootCmd.SetArgs(args) + err := RootCmd.Execute() + require.NoError(t, err) + w.Close() + out, _ := io.ReadAll(r) + os.Stdout = rescueStdout + t.Log(string(out)) + require.True(t, strings.Contains(string(out), "operator not healthy")) + require.True(t, strings.Contains(string(out), "connection refused")) resetFlags(RootCmd) }) for _, srv := range servers { diff --git a/pkgs/initiator/initiator.go b/pkgs/initiator/initiator.go index abee3fa2..78d4a595 100644 --- a/pkgs/initiator/initiator.go +++ b/pkgs/initiator/initiator.go @@ -805,7 +805,7 @@ func (c *Initiator) processPongMessage(res wire.PongResult) error { c.Logger.Info("🟢 operator online and healthy: multisig ready 👌 but NOT connected to ethereum network 🚫", zap.Uint64("ID", pong.ID), zap.String("IP", res.IP), zap.String("Version", string(signedPongMsg.Message.Version)), zap.String("Public key", string(pong.PubKey))) } } else { - c.Logger.Info("🟢 operator online: but NOT multisig ready 🚫", zap.Uint64("ID", pong.ID), zap.String("IP", res.IP), zap.String("Version", string(signedPongMsg.Message.Version)), zap.String("Public key", string(pong.PubKey))) + c.Logger.Error("🔴 operator online: but NOT multisig ready", zap.Uint64("ID", pong.ID), zap.String("IP", res.IP), zap.String("Version", string(signedPongMsg.Message.Version)), zap.String("Public key", string(pong.PubKey))) } return nil }