diff options
author | Jaime Martinez <jmartinez@gitlab.com> | 2020-07-27 04:13:51 +0300 |
---|---|---|
committer | Jaime Martinez <jmartinez@gitlab.com> | 2020-08-04 09:44:10 +0300 |
commit | 3c58f5f641c8f8dd51358fd812cffeb6491e75cc (patch) | |
tree | a7b260e0c6528cd173833cafe41da50304c6d0a4 /internal/source/domains_test.go | |
parent | 194bf0568619be8df50a51ed5979a34d4f697b7d (diff) |
Use IsReady instead
Diffstat (limited to 'internal/source/domains_test.go')
-rw-r--r-- | internal/source/domains_test.go | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/internal/source/domains_test.go b/internal/source/domains_test.go index da6f3877..12c3220a 100644 --- a/internal/source/domains_test.go +++ b/internal/source/domains_test.go @@ -66,6 +66,7 @@ func TestGetDomain(t *testing.T) { newSource.On("GetDomain", testDomain). Return(&domain.Domain{Name: testDomain}, nil). Once() + newSource.On("IsReady").Return(true).Once() defer newSource.AssertExpectations(t) domains := newTestDomains(t, newSource) @@ -75,6 +76,8 @@ func TestGetDomain(t *testing.T) { t.Run("when requesting a non-test domain", func(t *testing.T) { newSource := NewMockSource() + newSource.On("IsReady").Return(true).Once() + defer newSource.AssertExpectations(t) domains := newTestDomains(t, newSource) @@ -114,6 +117,8 @@ func TestGetDomain(t *testing.T) { newSource.On("GetDomain", testDomain). Return(&domain.Domain{Name: testDomain}, nil). Once() + newSource.On("IsReady").Return(true).Once() + defer newSource.AssertExpectations(t) domains := newTestDomains(t, newSource) @@ -188,6 +193,7 @@ func TestGetDomainWithIncrementalrolloutOfGitLabSource(t *testing.T) { Once() } } + gitlabSource.On("IsReady").Return(true) defer gitlabSource.AssertExpectations(t) domains := newTestDomains(t, gitlabSource) |