Merge pull request #5 from chatwork/fix_test

some fix
chatwork · Nov 24, 2023 · 0e9cd8a · 0e9cd8a
2 parents 44b0b54 + 9118ae0
commit 0e9cd8a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/cmd/datadog-agent/datadog-agent.go b/cmd/datadog-agent/datadog-agent.go
@@ -25,6 +25,7 @@ type DatadogAgent struct {
 	ApiKey      string
 	AppKey      string
 	ClusterName string
+	WaitTime    time.Duration
 }
 
 func NewDatadogAgent(debug bool, logger func() *logrus.Entry, chatwork *notify.Chatwork) (*DatadogAgent, error) {
@@ -58,6 +59,7 @@ func NewDatadogAgent(debug bool, logger func() *logrus.Entry, chatwork *notify.C
 		ApiKey:      apiKey,
 		AppKey:      appKey,
 		ClusterName: clusterName,
+		WaitTime:    3 * 60 * time.Second,
 	}, nil
 }
 
@@ -102,7 +104,7 @@ func (d *DatadogAgent) checkMetrics() error {
 	query := fmt.Sprintf("avg:kubernetes.cpu.user.total{env:%s}", d.ClusterName)
 
 	d.Logger().Info("Waiting metrics...")
-	time.Sleep((60 * 3) * time.Second)
+	time.Sleep(d.WaitTime)
 
 	d.Logger().Infof("Querying metrics with query: %s", query)
 	d.Chatwork.AddMessage(fmt.Sprintf("Querying metrics with query: %s", query))

diff --git a/cmd/datadog-agent/datadog-agent_test.go b/cmd/datadog-agent/datadog-agent_test.go
@@ -4,6 +4,7 @@ import (
 	"errors"
 	"os"
 	"testing"
+	"time"
 
 	"github.com/chatwork/kibertas/cmd"
 	"github.com/chatwork/kibertas/config"
@@ -42,6 +43,7 @@ func TestCheck(t *testing.T) {
 		ApiKey:      "",
 		AppKey:      "",
 		ClusterName: "",
+		WaitTime:    1 * time.Second,
 	}
 
 	err = datadogAgent.Check()