test: remote

sundowndev · Jul 23, 2022 · dff93c5 · dff93c5
1 parent b4b6918
commit dff93c5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/remote/remote_test.go b/lib/remote/remote_test.go
@@ -26,12 +26,12 @@ func TestRemoteLibrarySuccessScan(t *testing.T) {
 
 	fakeScanner := &mocks.Scanner{}
 	fakeScanner.On("ShouldRun").Return(true).Once()
-	fakeScanner.On("Identifier").Return("fake").Once()
+	fakeScanner.On("Identifier").Return("fake").Times(2)
 	fakeScanner.On("Scan", num).Return(fakeScannerResponse{Valid: true}, nil).Once()
 
 	fakeScanner2 := &mocks.Scanner{}
 	fakeScanner2.On("ShouldRun").Return(true).Once()
-	fakeScanner2.On("Identifier").Return("fake2").Once()
+	fakeScanner2.On("Identifier").Return("fake2").Times(2)
 	fakeScanner2.On("Scan", num).Return(fakeScannerResponse{Valid: false}, nil).Once()
 
 	lib := NewLibrary(filter.NewEngine())
@@ -57,7 +57,7 @@ func TestRemoteLibraryFailedScan(t *testing.T) {
 
 	fakeScanner := &mocks.Scanner{}
 	fakeScanner.On("ShouldRun").Return(true).Once()
-	fakeScanner.On("Identifier").Return("fake").Once()
+	fakeScanner.On("Identifier").Return("fake").Times(2)
 	fakeScanner.On("Scan", num).Return(nil, dummyError).Once()
 
 	lib := NewLibrary(filter.NewEngine())