@@ -5,20 +5,15 @@ import (
5
5
"crypto/sha1"
6
6
"fmt"
7
7
8
- . "github.com/onsi/ginkgo/v2"
9
- . "github.com/onsi/gomega"
10
-
11
- "github.com/stretchr/testify/mock"
12
-
13
- corev1 "k8s.io/api/core/v1"
14
- "k8s.io/apimachinery/pkg/runtime"
15
- kubeclientfakes "k8s.io/client-go/kubernetes/fake"
16
- kubeclienttesting "k8s.io/client-go/testing"
8
+ cfntypes "github.com/aws/aws-sdk-go-v2/service/cloudformation/types"
9
+ ekstypes "github.com/aws/aws-sdk-go-v2/service/eks/types"
17
10
18
11
"github.com/aws/aws-sdk-go-v2/aws"
19
- cfntypes "github.com/aws/aws-sdk-go-v2/service/cloudformation/types"
20
12
"github.com/aws/aws-sdk-go-v2/service/eks"
21
- ekstypes "github.com/aws/aws-sdk-go-v2/service/eks/types"
13
+ "github.com/stretchr/testify/mock"
14
+
15
+ . "github.com/onsi/ginkgo/v2"
16
+ . "github.com/onsi/gomega"
22
17
23
18
"github.com/weaveworks/eksctl/pkg/actions/podidentityassociation"
24
19
api "github.com/weaveworks/eksctl/pkg/apis/eksctl.io/v1alpha5"
@@ -30,7 +25,7 @@ import (
30
25
var _ = Describe ("Pod Identity Deleter" , func () {
31
26
type deleteEntry struct {
32
27
podIdentityAssociations []api.PodIdentityAssociation
33
- mockCalls func (stackManager * managerfakes.FakeStackManager , clientSet * kubeclientfakes. Clientset , eksAPI * mocksv2.EKS )
28
+ mockCalls func (stackManager * managerfakes.FakeStackManager , eksAPI * mocksv2.EKS )
34
29
35
30
expectedCalls func (stackManager * managerfakes.FakeStackManager , eksAPI * mocksv2.EKS )
36
31
expectedErr string
@@ -45,23 +40,14 @@ var _ = Describe("Pod Identity Deleter", func() {
45
40
return nil
46
41
}
47
42
}
48
- mockClientSet := func (clientSet * kubeclientfakes.Clientset ) {
49
- clientSet .PrependReactor ("delete" , "serviceaccounts" , func (action kubeclienttesting.Action ) (bool , runtime.Object , error ) {
50
- return true , nil , nil
51
- })
52
- clientSet .PrependReactor ("get" , "serviceaccounts" , func (action kubeclienttesting.Action ) (bool , runtime.Object , error ) {
53
- return true , & corev1.ServiceAccount {}, nil
54
- })
55
- }
56
- mockCalls := func (stackManager * managerfakes.FakeStackManager , clientSet * kubeclientfakes.Clientset , eksAPI * mocksv2.EKS , podID podidentityassociation.Identifier ) {
43
+ mockCalls := func (stackManager * managerfakes.FakeStackManager , eksAPI * mocksv2.EKS , podID podidentityassociation.Identifier ) {
57
44
stackName := makeIRSAv2StackName (podID )
58
45
associationID := fmt .Sprintf ("%x" , sha1 .Sum ([]byte (stackName )))
59
46
mockListPodIdentityAssociations (eksAPI , podID , []ekstypes.PodIdentityAssociationSummary {
60
47
{
61
48
AssociationId : aws .String (associationID ),
62
49
},
63
50
}, nil )
64
- mockClientSet (clientSet )
65
51
eksAPI .On ("DeletePodIdentityAssociation" , mock .Anything , & eks.DeletePodIdentityAssociationInput {
66
52
ClusterName : aws .String (clusterName ),
67
53
AssociationId : aws .String (associationID ),
@@ -71,14 +57,12 @@ var _ = Describe("Pod Identity Deleter", func() {
71
57
72
58
DescribeTable ("delete pod identity association" , func (e deleteEntry ) {
73
59
provider := mockprovider .NewMockProvider ()
74
- clientSet := kubeclientfakes .NewSimpleClientset ()
75
60
var stackManager managerfakes.FakeStackManager
76
- e .mockCalls (& stackManager , clientSet , provider .MockEKS ())
61
+ e .mockCalls (& stackManager , provider .MockEKS ())
77
62
deleter := podidentityassociation.Deleter {
78
63
ClusterName : clusterName ,
79
64
StackDeleter : & stackManager ,
80
65
APIDeleter : provider .EKS (),
81
- ClientSet : clientSet ,
82
66
}
83
67
err := deleter .Delete (context .Background (), podidentityassociation .ToIdentifiers (e .podIdentityAssociations ))
84
68
@@ -96,13 +80,13 @@ var _ = Describe("Pod Identity Deleter", func() {
96
80
ServiceAccountName : "default" ,
97
81
},
98
82
},
99
- mockCalls : func (stackManager * managerfakes.FakeStackManager , fakeClientSet * kubeclientfakes. Clientset , eksAPI * mocksv2.EKS ) {
83
+ mockCalls : func (stackManager * managerfakes.FakeStackManager , eksAPI * mocksv2.EKS ) {
100
84
podID := podidentityassociation.Identifier {
101
85
Namespace : "default" ,
102
86
ServiceAccountName : "default" ,
103
87
}
104
88
mockListStackNames (stackManager , []podidentityassociation.Identifier {podID })
105
- mockCalls (stackManager , fakeClientSet , eksAPI , podID )
89
+ mockCalls (stackManager , eksAPI , podID )
106
90
},
107
91
108
92
expectedCalls : func (stackManager * managerfakes.FakeStackManager , eksAPI * mocksv2.EKS ) {
@@ -123,7 +107,7 @@ var _ = Describe("Pod Identity Deleter", func() {
123
107
ServiceAccountName : "aws-node" ,
124
108
},
125
109
},
126
- mockCalls : func (stackManager * managerfakes.FakeStackManager , clientSet * kubeclientfakes. Clientset , eksAPI * mocksv2.EKS ) {
110
+ mockCalls : func (stackManager * managerfakes.FakeStackManager , eksAPI * mocksv2.EKS ) {
127
111
podIDs := []podidentityassociation.Identifier {
128
112
{
129
113
Namespace : "default" ,
@@ -136,7 +120,7 @@ var _ = Describe("Pod Identity Deleter", func() {
136
120
}
137
121
mockListStackNamesWithIRSAv1 (stackManager , podIDs [:1 ], podIDs [1 :])
138
122
for _ , podID := range podIDs {
139
- mockCalls (stackManager , clientSet , eksAPI , podID )
123
+ mockCalls (stackManager , eksAPI , podID )
140
124
}
141
125
},
142
126
@@ -183,7 +167,7 @@ var _ = Describe("Pod Identity Deleter", func() {
183
167
ServiceAccountName : "coredns" ,
184
168
},
185
169
},
186
- mockCalls : func (stackManager * managerfakes.FakeStackManager , clientSet * kubeclientfakes. Clientset , eksAPI * mocksv2.EKS ) {
170
+ mockCalls : func (stackManager * managerfakes.FakeStackManager , eksAPI * mocksv2.EKS ) {
187
171
podIDs := []podidentityassociation.Identifier {
188
172
{
189
173
Namespace : "default" ,
@@ -200,7 +184,7 @@ var _ = Describe("Pod Identity Deleter", func() {
200
184
}
201
185
mockListStackNames (stackManager , podIDs )
202
186
for _ , podID := range podIDs {
203
- mockCalls (stackManager , clientSet , eksAPI , podID )
187
+ mockCalls (stackManager , eksAPI , podID )
204
188
}
205
189
mockListPodIdentityAssociations (eksAPI , podidentityassociation.Identifier {
206
190
Namespace : "kube-system" ,
@@ -223,7 +207,7 @@ var _ = Describe("Pod Identity Deleter", func() {
223
207
ServiceAccountName : "aws-node" ,
224
208
},
225
209
},
226
- mockCalls : func (stackManager * managerfakes.FakeStackManager , clientSet * kubeclientfakes. Clientset , eksAPI * mocksv2.EKS ) {
210
+ mockCalls : func (stackManager * managerfakes.FakeStackManager , eksAPI * mocksv2.EKS ) {
227
211
podID := podidentityassociation.Identifier {
228
212
Namespace : "kube-system" ,
229
213
ServiceAccountName : "aws-node" ,
@@ -252,7 +236,7 @@ var _ = Describe("Pod Identity Deleter", func() {
252
236
ServiceAccountName : "aws-node" ,
253
237
},
254
238
},
255
- mockCalls : func (stackManager * managerfakes.FakeStackManager , clientSet * kubeclientfakes. Clientset , eksAPI * mocksv2.EKS ) {
239
+ mockCalls : func (stackManager * managerfakes.FakeStackManager , eksAPI * mocksv2.EKS ) {
256
240
podIDs := []podidentityassociation.Identifier {
257
241
{
258
242
Namespace : "default" ,
@@ -279,7 +263,7 @@ var _ = Describe("Pod Identity Deleter", func() {
279
263
280
264
Entry ("delete IAM resources on cluster deletion" , deleteEntry {
281
265
podIdentityAssociations : []api.PodIdentityAssociation {},
282
- mockCalls : func (stackManager * managerfakes.FakeStackManager , clientSet * kubeclientfakes. Clientset , eksAPI * mocksv2.EKS ) {
266
+ mockCalls : func (stackManager * managerfakes.FakeStackManager , eksAPI * mocksv2.EKS ) {
283
267
podIDs := []podidentityassociation.Identifier {
284
268
{
285
269
Namespace : "default" ,
0 commit comments