@@ -19,38 +19,37 @@ package ccm
1919import (
2020 "context"
2121 "fmt"
22+ "net/http"
2223
2324 . "github.com/onsi/ginkgo/v2"
2425 . "github.com/onsi/gomega"
26+ oapiError "github.com/stackitcloud/stackit-sdk-go/core/oapierror"
2527
26- "github.com/stackitcloud/cloud-provider-stackit/pkg/stackit"
28+ stackitclientmock "github.com/stackitcloud/cloud-provider-stackit/pkg/stackit/client/mock "
2729 iaas "github.com/stackitcloud/stackit-sdk-go/services/iaas/v2api"
28-
2930 "go.uber.org/mock/gomock"
3031 corev1 "k8s.io/api/core/v1"
3132 metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
3233)
3334
3435var _ = Describe ("Node Controller" , func () {
3536 var (
36- nodeMockClient * stackit. MockNodeClient
37+ nodeMockClient * stackitclientmock. MockIaaSClient
3738 instance * Instances
3839
39- projectID string
40- region string
41- serverID string
40+ region string
41+ serverID string
4242 )
4343
4444 BeforeEach (func () {
45- projectID = "my-project"
4645 region = "eu01"
4746 serverID = "my-server"
4847
4948 ctrl := gomock .NewController (GinkgoT ())
50- nodeMockClient = stackit . NewMockNodeClient (ctrl )
49+ nodeMockClient = stackitclientmock . NewMockIaaSClient (ctrl )
5150
5251 var err error
53- instance , err = NewInstance (nodeMockClient , projectID , "eu01" )
52+ instance , err = NewInstance (nodeMockClient , "eu01" )
5453 Expect (err ).NotTo (HaveOccurred ())
5554 })
5655
@@ -68,7 +67,7 @@ var _ = Describe("Node Controller", func() {
6867
6968 Describe ("InstanceExists" , func () {
7069 It ("does not error if instance not found" , func () {
71- nodeMockClient .EXPECT ().ListServers (gomock .Any (), projectID , region ).Return (& []iaas.Server {}, nil )
70+ nodeMockClient .EXPECT ().ListServers (gomock .Any ()).Return (& []iaas.Server {}, nil )
7271
7372 node := & corev1.Node {
7473 ObjectMeta : metav1.ObjectMeta {Name : "foo" },
@@ -80,7 +79,7 @@ var _ = Describe("Node Controller", func() {
8079 })
8180
8281 It ("successfully get the instance when provider ID not there" , func () {
83- nodeMockClient .EXPECT ().ListServers (gomock .Any (), projectID , region ).Return (& []iaas.Server {
82+ nodeMockClient .EXPECT ().ListServers (gomock .Any ()).Return (& []iaas.Server {
8483 {
8584 Name : "foo" ,
8685 },
@@ -96,7 +95,7 @@ var _ = Describe("Node Controller", func() {
9695 })
9796
9897 It ("successfully get the instance when provider ID is there" , func () {
99- nodeMockClient .EXPECT ().GetServer (gomock .Any (), projectID , region , serverID ).Return (& iaas.Server {
98+ nodeMockClient .EXPECT ().GetServer (gomock .Any (), serverID ).Return (& iaas.Server {
10099 Name : "foo" ,
101100 }, nil )
102101
@@ -113,7 +112,7 @@ var _ = Describe("Node Controller", func() {
113112 })
114113
115114 It ("successfully get the instance when old provider ID is there" , func () {
116- nodeMockClient .EXPECT ().GetServer (gomock .Any (), projectID , region , serverID ).Return (& iaas.Server {
115+ nodeMockClient .EXPECT ().GetServer (gomock .Any (), serverID ).Return (& iaas.Server {
117116 Name : "foo" ,
118117 }, nil )
119118
@@ -130,7 +129,7 @@ var _ = Describe("Node Controller", func() {
130129 })
131130
132131 It ("successfully get the instance when old regional provider ID is there" , func () {
133- nodeMockClient .EXPECT ().GetServer (gomock .Any (), projectID , region , serverID ).Return (& iaas.Server {
132+ nodeMockClient .EXPECT ().GetServer (gomock .Any (), serverID ).Return (& iaas.Server {
134133 Name : "foo" ,
135134 }, nil )
136135
@@ -147,7 +146,7 @@ var _ = Describe("Node Controller", func() {
147146 })
148147
149148 It ("error when list server fails" , func () {
150- nodeMockClient .EXPECT ().ListServers (gomock .Any (), projectID , region ).Return (nil , fmt .Errorf ("failed due to some reason" ))
149+ nodeMockClient .EXPECT ().ListServers (gomock .Any ()).Return (nil , fmt .Errorf ("failed due to some reason" ))
151150
152151 node := & corev1.Node {
153152 ObjectMeta : metav1.ObjectMeta {Name : "foo" },
@@ -158,7 +157,7 @@ var _ = Describe("Node Controller", func() {
158157 })
159158
160159 It ("does not error when get server instance not found" , func () {
161- nodeMockClient .EXPECT ().GetServer (gomock .Any (), projectID , region , serverID ).Return (nil , stackit . ErrorNotFound )
160+ nodeMockClient .EXPECT ().GetServer (gomock .Any (), serverID ).Return (nil , & oapiError. GenericOpenAPIError { StatusCode : http . StatusNotFound } )
162161
163162 node := & corev1.Node {
164163 ObjectMeta : metav1.ObjectMeta {Name : "foo" },
@@ -175,7 +174,7 @@ var _ = Describe("Node Controller", func() {
175174
176175 Describe ("InstanceShutdown" , func () {
177176 It ("successfully gets the instance status with provider ID" , func () {
178- nodeMockClient .EXPECT ().ListServers (gomock .Any (), projectID , region ).Return (& []iaas.Server {
177+ nodeMockClient .EXPECT ().ListServers (gomock .Any ()).Return (& []iaas.Server {
179178 {
180179 Name : "foo" ,
181180 Status : new (instanceStopping ),
@@ -192,7 +191,7 @@ var _ = Describe("Node Controller", func() {
192191 })
193192
194193 It ("successfully gets the instance status without provider ID" , func () {
195- nodeMockClient .EXPECT ().GetServer (gomock .Any (), projectID , region , serverID ).Return (& iaas.Server {
194+ nodeMockClient .EXPECT ().GetServer (gomock .Any (), serverID ).Return (& iaas.Server {
196195 Name : "foo" ,
197196 Status : new ("ACTIVE "),
198197 }, nil )
@@ -210,7 +209,7 @@ var _ = Describe("Node Controller", func() {
210209 })
211210
212211 It ("fails if server not found" , func () {
213- nodeMockClient .EXPECT ().ListServers (gomock .Any (), projectID , region ).Return (nil , stackit . ErrorNotFound )
212+ nodeMockClient .EXPECT ().ListServers (gomock .Any ()).Return (nil , & oapiError. GenericOpenAPIError { StatusCode : http . StatusNotFound } )
214213
215214 node := & corev1.Node {
216215 ObjectMeta : metav1.ObjectMeta {Name : "foo" },
@@ -224,7 +223,7 @@ var _ = Describe("Node Controller", func() {
224223
225224 Describe ("InstanceMetadata" , func () {
226225 It ("does not error if instance not found" , func () {
227- nodeMockClient .EXPECT ().ListServers (gomock .Any (), projectID , region ).Return (& []iaas.Server {}, nil )
226+ nodeMockClient .EXPECT ().ListServers (gomock .Any ()).Return (& []iaas.Server {}, nil )
228227
229228 node := & corev1.Node {
230229 ObjectMeta : metav1.ObjectMeta {Name : "foo" },
@@ -236,7 +235,7 @@ var _ = Describe("Node Controller", func() {
236235 })
237236
238237 It ("successfully get all the metadata values" , func () {
239- nodeMockClient .EXPECT ().ListServers (gomock .Any (), projectID , region ).Return (& []iaas.Server {
238+ nodeMockClient .EXPECT ().ListServers (gomock .Any ()).Return (& []iaas.Server {
240239 {
241240 Name : "foo" ,
242241 Id : new (serverID ),
@@ -271,7 +270,7 @@ var _ = Describe("Node Controller", func() {
271270 })
272271
273272 It ("errors when list server fails" , func () {
274- nodeMockClient .EXPECT ().ListServers (gomock .Any (), projectID , region ).Return (nil , fmt .Errorf ("failed due to some reason" ))
273+ nodeMockClient .EXPECT ().ListServers (gomock .Any ()).Return (nil , fmt .Errorf ("failed due to some reason" ))
275274
276275 node := & corev1.Node {
277276 ObjectMeta : metav1.ObjectMeta {Name : "foo" },
0 commit comments