adapted unit tests

keptn · Aug 24, 2022 · a87fe4f · a87fe4f
1 parent 2b2d9ce
commit a87fe4f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/pkg/sdk/internal/api/initializer_test.go b/pkg/sdk/internal/api/initializer_test.go
@@ -3,6 +3,7 @@ package api
 import (
 	"fmt"
 	keptnapi "github.com/keptn/go-utils/pkg/api/utils"
+	keptnapiv2 "github.com/keptn/go-utils/pkg/api/utils/v2"
 	"github.com/keptn/go-utils/pkg/sdk/connector/logger"
 	"github.com/keptn/go-utils/pkg/sdk/internal/config"
 	"github.com/stretchr/testify/require"
@@ -40,7 +41,8 @@ func Test_Initialize(t *testing.T) {
 		require.NotNil(t, result.EventSenderCallback)
 		require.NotNil(t, result.KeptnAPI)
 		require.IsType(t, &keptnapi.APISet{}, result.KeptnAPI)
-		require.NotNil(t, result.ResourceHandler)
+		require.NotNil(t, result.KeptnAPIV2)
+		require.IsType(t, &keptnapiv2.APISet{}, result.KeptnAPIV2)
 	})
 	t.Run("Internal Use case - internal api set is used", func(t *testing.T) {
 		env := config.EnvConfig{}
@@ -50,7 +52,8 @@ func Test_Initialize(t *testing.T) {
 		require.NotNil(t, result.EventSenderCallback)
 		require.NotNil(t, result.KeptnAPI)
 		require.IsType(t, &keptnapi.InternalAPISet{}, result.KeptnAPI)
-		require.NotNil(t, result.ResourceHandler)
+		require.NotNil(t, result.KeptnAPIV2)
+		require.IsType(t, &keptnapiv2.InternalAPISet{}, result.KeptnAPIV2)
 	})
 	t.Run("HTTP client creation fails", func(t *testing.T) {
 		env := config.EnvConfig{KeptnAPIEndpoint: "http://endpoint"}
@@ -66,6 +69,7 @@ func Test_Initialize(t *testing.T) {
 		require.NotNil(t, result.EventSenderCallback)
 		require.NotNil(t, result.KeptnAPI)
 		require.IsType(t, &keptnapi.APISet{}, result.KeptnAPI)
-		require.NotNil(t, result.ResourceHandler)
+		require.NotNil(t, result.KeptnAPIV2)
+		require.IsType(t, &keptnapiv2.APISet{}, result.KeptnAPIV2)
 	})
 }
diff --git a/pkg/sdk/keptn_test.go b/pkg/sdk/keptn_test.go
@@ -35,6 +35,7 @@ func Test_NewKeptn(t *testing.T) {
 		require.IsType(t, keptnSDK.taskRegistry.Get("event2.type").taskHandler, &TaskHandlerMock{})
 		require.NotNil(t, keptnSDK.GetResourceHandler())
 		require.NotNil(t, keptnSDK.APIV1())
+		require.NotNil(t, keptnSDK.APIV2())
 
 	})
 }