Skip to content

Commit 40f8633

Browse files
committed
add t.helper() for test lint
Signed-off-by: richie <mychenforeverl@gmail.com>
1 parent 261fc87 commit 40f8633

File tree

2 files changed

+16
-12
lines changed

2 files changed

+16
-12
lines changed

internal/controller/nginx/agent/command_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,14 +72,15 @@ func createFakeK8sClient(initObjs ...runtime.Object) (client.Client, error) {
7272
}
7373

7474
func createGrpcContext(t *testing.T) context.Context {
75+
t.Helper()
7576
return grpcContext.NewGrpcContext(t.Context(), grpcContext.GrpcInfo{
7677
IPAddress: "127.0.0.1",
7778
})
7879
}
7980

8081
func createGrpcContextWithCancel(t *testing.T) (context.Context, context.CancelFunc) {
82+
t.Helper()
8183
ctx, cancel := context.WithCancel(t.Context())
82-
8384
return grpcContext.NewGrpcContext(ctx, grpcContext.GrpcInfo{
8485
IPAddress: "127.0.0.1",
8586
}), cancel

internal/controller/provisioner/provisioner_test.go

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,8 @@ func createScheme() *runtime.Scheme {
5151
return scheme
5252
}
5353

54-
func expectResourcesToExist(g *WithT, t *testing.T, k8sClient client.Client, nsName types.NamespacedName, plus bool) {
54+
func expectResourcesToExist(t *testing.T, g *WithT, k8sClient client.Client, nsName types.NamespacedName, plus bool) {
55+
t.Helper()
5556
g.Expect(k8sClient.Get(t.Context(), nsName, &appsv1.Deployment{})).To(Succeed())
5657

5758
g.Expect(k8sClient.Get(t.Context(), nsName, &corev1.Service{})).To(Succeed())
@@ -105,7 +106,8 @@ func expectResourcesToExist(g *WithT, t *testing.T, k8sClient client.Client, nsN
105106
g.Expect(k8sClient.Get(t.Context(), dockerSecret, &corev1.Secret{})).To(Succeed())
106107
}
107108

108-
func expectResourcesToNotExist(g *WithT, t *testing.T, k8sClient client.Client, nsName types.NamespacedName) {
109+
func expectResourcesToNotExist(t *testing.T, g *WithT, k8sClient client.Client, nsName types.NamespacedName) {
110+
t.Helper()
109111
g.Expect(k8sClient.Get(t.Context(), nsName, &appsv1.Deployment{})).ToNot(Succeed())
110112

111113
g.Expect(k8sClient.Get(t.Context(), nsName, &corev1.Service{})).ToNot(Succeed())
@@ -269,7 +271,7 @@ func TestEnable(t *testing.T) {
269271
provisioner.Enable(t.Context())
270272
g.Expect(provisioner.isLeader()).To(BeTrue())
271273
g.Expect(provisioner.resourcesToDeleteOnStartup).To(BeEmpty())
272-
expectResourcesToNotExist(g, t, fakeClient, types.NamespacedName{Name: "gw-nginx", Namespace: "default"})
274+
expectResourcesToNotExist(t, g, fakeClient, types.NamespacedName{Name: "gw-nginx", Namespace: "default"})
273275
}
274276

275277
func TestRegisterGateway(t *testing.T) {
@@ -323,11 +325,11 @@ func TestRegisterGateway(t *testing.T) {
323325
provisioner, fakeClient, deploymentStore := defaultNginxProvisioner(objects...)
324326

325327
g.Expect(provisioner.RegisterGateway(t.Context(), gateway, "gw-nginx")).To(Succeed())
326-
expectResourcesToExist(g, t, fakeClient, types.NamespacedName{Name: "gw-nginx", Namespace: "default"}, true) // plus
328+
expectResourcesToExist(t, g, fakeClient, types.NamespacedName{Name: "gw-nginx", Namespace: "default"}, true) // plus
327329

328330
// Call again, no updates so nothing should happen
329331
g.Expect(provisioner.RegisterGateway(t.Context(), gateway, "gw-nginx")).To(Succeed())
330-
expectResourcesToExist(g, t, fakeClient, types.NamespacedName{Name: "gw-nginx", Namespace: "default"}, true) // plus
332+
expectResourcesToExist(t, g, fakeClient, types.NamespacedName{Name: "gw-nginx", Namespace: "default"}, true) // plus
331333

332334
// Now set the Gateway to invalid, and expect a deprovision to occur
333335
invalid := &graph.Gateway{
@@ -340,7 +342,7 @@ func TestRegisterGateway(t *testing.T) {
340342
Valid: false,
341343
}
342344
g.Expect(provisioner.RegisterGateway(t.Context(), invalid, "gw-nginx")).To(Succeed())
343-
expectResourcesToNotExist(g, t, fakeClient, types.NamespacedName{Name: "gw-nginx", Namespace: "default"})
345+
expectResourcesToNotExist(t, g, fakeClient, types.NamespacedName{Name: "gw-nginx", Namespace: "default"})
344346

345347
resources := provisioner.store.getNginxResourcesForGateway(types.NamespacedName{Name: "gw", Namespace: "default"})
346348
g.Expect(resources).To(BeNil())
@@ -479,16 +481,16 @@ func TestNonLeaderProvisioner(t *testing.T) {
479481
nsName := types.NamespacedName{Name: "gw-nginx", Namespace: "default"}
480482

481483
g.Expect(provisioner.RegisterGateway(t.Context(), nil, "gw-nginx")).To(Succeed())
482-
expectResourcesToNotExist(g, t, fakeClient, nsName)
484+
expectResourcesToNotExist(t, g, fakeClient, nsName)
483485

484486
g.Expect(provisioner.provisionNginx(t.Context(), "gw-nginx", nil, nil)).To(Succeed())
485-
expectResourcesToNotExist(g, t, fakeClient, nsName)
487+
expectResourcesToNotExist(t, g, fakeClient, nsName)
486488

487489
g.Expect(provisioner.reprovisionNginx(t.Context(), "gw-nginx", nil, nil)).To(Succeed())
488-
expectResourcesToNotExist(g, t, fakeClient, nsName)
490+
expectResourcesToNotExist(t, g, fakeClient, nsName)
489491

490492
g.Expect(provisioner.deprovisionNginx(t.Context(), nsName)).To(Succeed())
491-
expectResourcesToNotExist(g, t, fakeClient, nsName)
493+
expectResourcesToNotExist(t, g, fakeClient, nsName)
492494
g.Expect(deploymentStore.RemoveCallCount()).To(Equal(1))
493495
}
494496

@@ -524,7 +526,8 @@ func TestProvisionerRestartsDeployment(t *testing.T) {
524526
provisioner.cfg.NginxDockerSecretNames = nil
525527

526528
g.Expect(provisioner.RegisterGateway(t.Context(), gateway, "gw-nginx")).To(Succeed())
527-
expectResourcesToExist(g, t, fakeClient, types.NamespacedName{Name: "gw-nginx", Namespace: "default"}, false) // not plus
529+
// not plus
530+
expectResourcesToExist(t, g, fakeClient, types.NamespacedName{Name: "gw-nginx", Namespace: "default"}, false)
528531

529532
// update agent config
530533
updatedConfig := &graph.Gateway{

0 commit comments

Comments
 (0)