Skip to content

Commit 3749be4

Browse files
committed
Migrate tests from gotest to gomega
Signed-off-by: Stefan Prodan <stefan.prodan@gmail.com>
1 parent 1a244f7 commit 3749be4

File tree

11 files changed

+151
-91
lines changed

11 files changed

+151
-91
lines changed

go.mod

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,6 @@ require (
6767
golang.org/x/oauth2 v0.30.0
6868
golang.org/x/sync v0.16.0
6969
google.golang.org/api v0.248.0
70-
gotest.tools v2.2.0+incompatible
7170
helm.sh/helm/v3 v3.18.6
7271
k8s.io/api v0.34.0
7372
k8s.io/apimachinery v0.34.0

go.sum

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1359,8 +1359,6 @@ gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
13591359
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
13601360
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
13611361
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
1362-
gotest.tools v2.2.0+incompatible h1:VsBPFP1AI068pPrMxtb/S8Zkgf9xEmTLJjfM+P5UIEo=
1363-
gotest.tools v2.2.0+incompatible/go.mod h1:DsYFclhRJ6vuDpmuTbkuFWG+y2sxOXAzmJt81HFBacw=
13641362
gotest.tools/v3 v3.5.2 h1:7koQfIKdy+I8UTetycgUqXWSDwpgv193Ka+qRsmBY8Q=
13651363
gotest.tools/v3 v3.5.2/go.mod h1:LtdLGcnqToBH83WByAAi/wiwSFCArdFIUV/xxN4pcjA=
13661364
helm.sh/helm/v3 v3.18.6 h1:S/2CqcYnNfLckkHLI0VgQbxgcDaU3N4A/46E3n9wSNY=

internal/bucket/azure/blob_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ import (
4545
func TestNewClientAndBucketExistsWithProxy(t *testing.T) {
4646
g := NewWithT(t)
4747

48-
proxyAddr, proxyPort := testproxy.New(t)
48+
proxyAddr, _ := testproxy.New(t)
4949

5050
// start mock bucket server
5151
bucketListener, bucketAddr, _ := testlistener.New(t)
@@ -91,7 +91,7 @@ func TestNewClientAndBucketExistsWithProxy(t *testing.T) {
9191
{
9292
name: "with incorrect proxy",
9393
endpoint: bucketEndpoint,
94-
proxyURL: &url.URL{Scheme: "http", Host: fmt.Sprintf("localhost:%d", proxyPort+1)},
94+
proxyURL: &url.URL{Scheme: "http", Host: fmt.Sprintf("localhost:%d", 1)},
9595
err: "connection refused",
9696
},
9797
}

internal/bucket/gcp/gcp_test.go

Lines changed: 40 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@ import (
3838
"google.golang.org/api/googleapi"
3939
"google.golang.org/api/option"
4040
raw "google.golang.org/api/storage/v1"
41-
"gotest.tools/assert"
4241
corev1 "k8s.io/api/core/v1"
4342
v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
4443

@@ -167,14 +166,17 @@ func TestNewClientWithSecretErr(t *testing.T) {
167166
bucket := createTestBucket()
168167
gcpClient, err := NewClient(context.Background(), bucket, WithSecret(secret.DeepCopy()))
169168
t.Log(err)
170-
assert.Error(t, err, "dialing: invalid character 'e' looking for beginning of value")
171-
assert.Assert(t, gcpClient == nil)
169+
g := NewWithT(t)
170+
g.Expect(err).To(HaveOccurred())
171+
g.Expect(err.Error()).To(Equal("dialing: invalid character 'e' looking for beginning of value"))
172+
g.Expect(gcpClient).To(BeNil())
172173
}
173174

174175
func TestNewClientWithProxyErr(t *testing.T) {
175176
_, envADCIsSet := os.LookupEnv(envADC)
176-
assert.Assert(t, !envADCIsSet)
177-
assert.Assert(t, !metadata.OnGCE())
177+
g := NewWithT(t)
178+
g.Expect(envADCIsSet).To(BeFalse())
179+
g.Expect(metadata.OnGCE()).To(BeFalse())
178180

179181
t.Run("with secret", func(t *testing.T) {
180182
g := NewWithT(t)
@@ -205,10 +207,11 @@ func TestProxy(t *testing.T) {
205207
proxyAddr, proxyPort := testproxy.New(t)
206208

207209
err := os.Setenv(envGCSHost, fmt.Sprintf("https://%s", host))
208-
assert.NilError(t, err)
210+
g := NewWithT(t)
211+
g.Expect(err).NotTo(HaveOccurred())
209212
defer func() {
210213
err := os.Unsetenv(envGCSHost)
211-
assert.NilError(t, err)
214+
g.Expect(err).NotTo(HaveOccurred())
212215
}()
213216

214217
tests := []struct {
@@ -242,15 +245,17 @@ func TestProxy(t *testing.T) {
242245
})
243246
bucket := createTestBucket()
244247
gcpClient, err := NewClient(context.Background(), bucket, opts...)
245-
assert.NilError(t, err)
246-
assert.Assert(t, gcpClient != nil)
248+
g := NewWithT(t)
249+
g.Expect(err).NotTo(HaveOccurred())
250+
g.Expect(gcpClient).NotTo(BeNil())
247251
gcpClient.Client.SetRetry(gcpstorage.WithMaxAttempts(1))
248252
exists, err := gcpClient.BucketExists(context.Background(), bucketName)
249253
if tt.err != "" {
250-
assert.ErrorContains(t, err, tt.err)
254+
g.Expect(err).To(HaveOccurred())
255+
g.Expect(err.Error()).To(ContainSubstring(tt.err))
251256
} else {
252-
assert.NilError(t, err)
253-
assert.Assert(t, exists)
257+
g.Expect(err).NotTo(HaveOccurred())
258+
g.Expect(exists).To(BeTrue())
254259
}
255260
})
256261
}
@@ -261,8 +266,9 @@ func TestBucketExists(t *testing.T) {
261266
Client: client,
262267
}
263268
exists, err := gcpClient.BucketExists(context.Background(), bucketName)
264-
assert.NilError(t, err)
265-
assert.Assert(t, exists)
269+
g := NewWithT(t)
270+
g.Expect(err).NotTo(HaveOccurred())
271+
g.Expect(exists).To(BeTrue())
266272
}
267273

268274
func TestBucketNotExists(t *testing.T) {
@@ -271,8 +277,9 @@ func TestBucketNotExists(t *testing.T) {
271277
Client: client,
272278
}
273279
exists, err := gcpClient.BucketExists(context.Background(), bucket)
274-
assert.NilError(t, err)
275-
assert.Assert(t, !exists)
280+
g := NewWithT(t)
281+
g.Expect(err).NotTo(HaveOccurred())
282+
g.Expect(exists).To(BeFalse())
276283
}
277284

278285
func TestVisitObjects(t *testing.T) {
@@ -286,9 +293,10 @@ func TestVisitObjects(t *testing.T) {
286293
etags = append(etags, etag)
287294
return nil
288295
})
289-
assert.NilError(t, err)
290-
assert.DeepEqual(t, keys, []string{objectName})
291-
assert.DeepEqual(t, etags, []string{objectEtag})
296+
g := NewWithT(t)
297+
g.Expect(err).NotTo(HaveOccurred())
298+
g.Expect(keys).To(Equal([]string{objectName}))
299+
g.Expect(etags).To(Equal([]string{objectEtag}))
292300
}
293301

294302
func TestVisitObjectsErr(t *testing.T) {
@@ -313,20 +321,23 @@ func TestVisitObjectsCallbackErr(t *testing.T) {
313321
err := gcpClient.VisitObjects(context.Background(), bucketName, "", func(key, etag string) error {
314322
return mockErr
315323
})
316-
assert.Error(t, err, mockErr.Error())
324+
g := NewWithT(t)
325+
g.Expect(err).To(HaveOccurred())
326+
g.Expect(err.Error()).To(Equal(mockErr.Error()))
317327
}
318328

319329
func TestFGetObject(t *testing.T) {
330+
g := NewWithT(t)
320331
tempDir := t.TempDir()
321332
gcpClient := &GCSClient{
322333
Client: client,
323334
}
324335
localPath := filepath.Join(tempDir, objectName)
325336
etag, err := gcpClient.FGetObject(context.Background(), bucketName, objectName, localPath)
326337
if err != io.EOF {
327-
assert.NilError(t, err)
338+
g.Expect(err).NotTo(HaveOccurred())
328339
}
329-
assert.Equal(t, etag, objectEtag)
340+
g.Expect(etag).To(Equal(objectEtag))
330341
}
331342

332343
func TestFGetObjectNotExists(t *testing.T) {
@@ -343,13 +354,15 @@ func TestFGetObjectNotExists(t *testing.T) {
343354
}
344355

345356
func TestFGetObjectDirectoryIsFileName(t *testing.T) {
357+
g := NewWithT(t)
346358
tempDir := t.TempDir()
347359
gcpClient := &GCSClient{
348360
Client: client,
349361
}
350362
_, err = gcpClient.FGetObject(context.Background(), bucketName, objectName, tempDir)
351363
if err != io.EOF {
352-
assert.Error(t, err, "filename is a directory")
364+
g.Expect(err).To(HaveOccurred())
365+
g.Expect(err.Error()).To(Equal("filename is a directory"))
353366
}
354367
}
355368

@@ -375,10 +388,12 @@ func TestValidateSecret(t *testing.T) {
375388
t.Run(tt.name, func(t *testing.T) {
376389
t.Parallel()
377390
err := ValidateSecret(tt.secret)
391+
g := NewWithT(t)
378392
if tt.error {
379-
assert.Error(t, err, fmt.Sprintf("invalid '%v' secret data: required fields 'serviceaccount'", tt.secret.Name))
393+
g.Expect(err).To(HaveOccurred())
394+
g.Expect(err.Error()).To(Equal(fmt.Sprintf("invalid '%v' secret data: required fields 'serviceaccount'", tt.secret.Name)))
380395
} else {
381-
assert.NilError(t, err)
396+
g.Expect(err).NotTo(HaveOccurred())
382397
}
383398
})
384399
}

0 commit comments

Comments
 (0)