test: add operator to handlers that use probeUtils

Signed-off-by: sukkyun2 <[email protected]>
litmuschaos · Aug 27, 2024 · 2308669 · 2308669
1 parent 84d8657
commit 2308669
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 5 deletions.
diff --git a/chaoscenter/graphql/server/pkg/chaos_experiment/handler/fuzz_tests/handler_fuzz_test.go b/chaoscenter/graphql/server/pkg/chaos_experiment/handler/fuzz_tests/handler_fuzz_test.go
@@ -2,6 +2,7 @@ package fuzz_tests
 
 import (
 	"context"
+	dbSchemaProbe "github.com/litmuschaos/litmus/chaoscenter/graphql/server/pkg/database/mongodb/probe"
 	"testing"
 
 	"github.com/litmuschaos/litmus/chaoscenter/graphql/server/pkg/chaos_experiment/handler"
@@ -45,8 +46,9 @@ func NewMockServices() *MockServices {
 		chaosExperimentOperator    = dbChaosExperiment.NewChaosExperimentOperator(mongodbMockOperator)
 		chaosExperimentRunOperator = dbChaosExperimentRun.NewChaosExperimentRunOperator(mongodbMockOperator)
 		chaosExperimentService     = new(chaosExperimentMocks.ChaosExperimentService)
+		probeOperator              = dbSchemaProbe.NewChaosProbeOperator(mongodbMockOperator)
 	)
-	var chaosExperimentHandler = handler.NewChaosExperimentHandler(chaosExperimentService, chaosExperimentRunService, infrastructureService, gitOpsService, chaosExperimentOperator, chaosExperimentRunOperator, mongodbMockOperator)
+	var chaosExperimentHandler = handler.NewChaosExperimentHandler(chaosExperimentService, chaosExperimentRunService, infrastructureService, gitOpsService, chaosExperimentOperator, chaosExperimentRunOperator, probeOperator, mongodbMockOperator)
 	return &MockServices{
 		ChaosExperimentService:     chaosExperimentService,
 		ChaosExperimentRunService:  chaosExperimentRunService,

diff --git a/chaoscenter/graphql/server/pkg/chaos_experiment/handler/handler_test.go b/chaoscenter/graphql/server/pkg/chaos_experiment/handler/handler_test.go
@@ -3,6 +3,7 @@ package handler
 import (
 	"context"
 	"errors"
+	dbSchemaProbe "github.com/litmuschaos/litmus/chaoscenter/graphql/server/pkg/database/mongodb/probe"
 	"reflect"
 	"testing"
 
@@ -46,8 +47,9 @@ func NewMockServices() *MockServices {
 		chaosExperimentOperator    = dbChaosExperiment.NewChaosExperimentOperator(mongodbMockOperator)
 		chaosExperimentRunOperator = dbChaosExperimentRun.NewChaosExperimentRunOperator(mongodbMockOperator)
 		chaosExperimentService     = new(chaosExperimentMocks.ChaosExperimentService)
+		probeOperator              = dbSchemaProbe.NewChaosProbeOperator(mongodbMockOperator)
 	)
-	var chaosExperimentHandler = NewChaosExperimentHandler(chaosExperimentService, chaosExperimentRunService, infrastructureService, gitOpsService, chaosExperimentOperator, chaosExperimentRunOperator, mongodbMockOperator)
+	var chaosExperimentHandler = NewChaosExperimentHandler(chaosExperimentService, chaosExperimentRunService, infrastructureService, gitOpsService, chaosExperimentOperator, chaosExperimentRunOperator, probeOperator, mongodbMockOperator)
 	return &MockServices{
 		ChaosExperimentService:     chaosExperimentService,
 		ChaosExperimentRunService:  chaosExperimentRunService,

diff --git a/chaoscenter/graphql/server/pkg/chaos_experiment/ops/service_test.go b/chaoscenter/graphql/server/pkg/chaos_experiment/ops/service_test.go
@@ -3,6 +3,7 @@ package ops
 import (
 	"context"
 	"errors"
+	dbSchemaProbe "github.com/litmuschaos/litmus/chaoscenter/graphql/server/pkg/database/mongodb/probe"
 	"io"
 	"log"
 	"math/rand"
@@ -28,10 +29,11 @@ import (
 
 var (
 	mongodbMockOperator        = new(dbMocks.MongoOperator)
+	probeOperator              = dbSchemaProbe.NewChaosProbeOperator(mongodbMockOperator)
 	infraOperator              = dbChaosInfra.NewInfrastructureOperator(mongodbMockOperator)
 	chaosExperimentOperator    = dbChaosExperiment.NewChaosExperimentOperator(mongodbMockOperator)
 	chaosExperimentRunOperator = dbChaosExperimentRun.NewChaosExperimentRunOperator(mongodbMockOperator)
-	probeService               = probe.NewProbeService()
+	probeService               = probe.NewProbeService(probeOperator)
 )
 
 var chaosExperimentRunTestService = NewChaosExperimentService(chaosExperimentOperator, infraOperator, chaosExperimentRunOperator, probeService)

diff --git a/chaoscenter/graphql/server/pkg/chaos_experiment_run/handler/handler_test.go b/chaoscenter/graphql/server/pkg/chaos_experiment_run/handler/handler_test.go
@@ -3,6 +3,7 @@ package handler
 import (
 	"context"
 	"errors"
+	dbSchemaProbe "github.com/litmuschaos/litmus/chaoscenter/graphql/server/pkg/database/mongodb/probe"
 	"io"
 	"log"
 	"os"
@@ -36,11 +37,12 @@ var (
 	infrastructureService      = new(chaosInfraMocks.InfraService)
 	chaosExperimentRunService  = new(choasExperimentRunMocks.ChaosExperimentRunService)
 	gitOpsService              = new(dbGitOpsMocks.GitOpsService)
+	probeOperator              = dbSchemaProbe.NewChaosProbeOperator(mongodbMockOperator)
 	chaosExperimentOperator    = dbChaosExperiment.NewChaosExperimentOperator(mongodbMockOperator)
 	chaosExperimentRunOperator = dbChaosExperimentRun.NewChaosExperimentRunOperator(mongodbMockOperator)
 )
 
-var chaosExperimentRunHandler = NewChaosExperimentRunHandler(chaosExperimentRunService, infrastructureService, gitOpsService, chaosExperimentOperator, chaosExperimentRunOperator, mongodbMockOperator)
+var chaosExperimentRunHandler = NewChaosExperimentRunHandler(chaosExperimentRunService, infrastructureService, gitOpsService, chaosExperimentOperator, chaosExperimentRunOperator, probeOperator, mongodbMockOperator)
 
 // TestMain is the entry point for testing
 func TestMain(m *testing.M) {
@@ -56,6 +58,7 @@ func TestNewChaosExperimentRunHandler(t *testing.T) {
 		gitOpsService              gitops.Service
 		chaosExperimentOperator    *dbChaosExperiment.Operator
 		chaosExperimentRunOperator *dbChaosExperimentRun.Operator
+		probeOperator              *dbSchemaProbe.Operator
 		mongodbOperator            mongodb.MongoOperator
 	}
 	tests := []struct {
@@ -71,6 +74,7 @@ func TestNewChaosExperimentRunHandler(t *testing.T) {
 				gitOpsService:              gitOpsService,
 				chaosExperimentOperator:    chaosExperimentOperator,
 				chaosExperimentRunOperator: chaosExperimentRunOperator,
+				probeOperator:              probeOperator,
 				mongodbOperator:            mongodbMockOperator,
 			},
 			want: &ChaosExperimentRunHandler{
@@ -79,13 +83,14 @@ func TestNewChaosExperimentRunHandler(t *testing.T) {
 				gitOpsService:              gitOpsService,
 				chaosExperimentOperator:    chaosExperimentOperator,
 				chaosExperimentRunOperator: chaosExperimentRunOperator,
+				probeOperator:              probeOperator,
 				mongodbOperator:            mongodbMockOperator,
 			},
 		},
 	}
 	for _, tc := range tests {
 		t.Run(tc.name, func(t *testing.T) {
-			if got := NewChaosExperimentRunHandler(tc.args.chaosExperimentRunService, tc.args.infrastructureService, tc.args.gitOpsService, tc.args.chaosExperimentOperator, tc.args.chaosExperimentRunOperator, tc.args.mongodbOperator); !reflect.DeepEqual(got, tc.want) {
+			if got := NewChaosExperimentRunHandler(tc.args.chaosExperimentRunService, tc.args.infrastructureService, tc.args.gitOpsService, tc.args.chaosExperimentOperator, tc.args.chaosExperimentRunOperator, tc.args.probeOperator, tc.args.mongodbOperator); !reflect.DeepEqual(got, tc.want) {
 				t.Errorf("NewChaosExperimentRunHandler() = %v, want %v", got, tc.want)
 			}
 		})