diff options
author | Jaime Martinez <jmartinez@gitlab.com> | 2021-07-08 05:00:58 +0300 |
---|---|---|
committer | Jaime Martinez <jmartinez@gitlab.com> | 2021-07-08 08:19:06 +0300 |
commit | 15645988b383de8adcca24c49357b347843e994c (patch) | |
tree | 932da45947989231ba07a99c507a2c13c13af55d /internal/source/domains_test.go | |
parent | c81339af0bd218ccb956f53d0f7bbabef835a5a4 (diff) |
Update test mocks
Diffstat (limited to 'internal/source/domains_test.go')
-rw-r--r-- | internal/source/domains_test.go | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/internal/source/domains_test.go b/internal/source/domains_test.go index 7c351718..8618df38 100644 --- a/internal/source/domains_test.go +++ b/internal/source/domains_test.go @@ -1,6 +1,7 @@ package source import ( + "context" "testing" "time" @@ -113,7 +114,7 @@ func TestGetDomain(t *testing.T) { domains := newTestDomains(t, newSource, sourceGitlab) - domain, err := domains.GetDomain(testDomain) + domain, err := domains.GetDomain(context.Background(), testDomain) require.NoError(t, err) require.NotNil(t, domain) }) @@ -130,7 +131,7 @@ func TestGetDomain(t *testing.T) { domains := newTestDomains(t, newSource, sourceAuto) - domain, err := domains.GetDomain(testDomain) + domain, err := domains.GetDomain(context.Background(), testDomain) require.NoError(t, err) require.NotNil(t, domain) }) @@ -145,7 +146,7 @@ func TestGetDomain(t *testing.T) { domains := newTestDomains(t, newSource, sourceGitlab) - domain, err := domains.GetDomain("does-not-exist.test.io") + domain, err := domains.GetDomain(context.Background(), "does-not-exist.test.io") require.NoError(t, err) require.Nil(t, domain) }) |