From 3ce9bad2a3651f5175081d74332c33b23af2e544 Mon Sep 17 00:00:00 2001 From: Julien Duchesne Date: Sun, 21 Jul 2019 16:44:51 -0400 Subject: [PATCH] Test DeleteListOfCredentials --- sync/target_test.go | 26 ++++++++++++++++++-------- 1 file changed, 18 insertions(+), 8 deletions(-) diff --git a/sync/target_test.go b/sync/target_test.go index ad9c557..d70e554 100644 --- a/sync/target_test.go +++ b/sync/target_test.go @@ -8,31 +8,41 @@ import ( "github.com/golang/mock/gomock" ) -func setSourceMock(t *testing.T, config *Configuration) *credentials.MockSource { +func setSourceMock(t *testing.T, config *Configuration) (*gomock.Controller, *credentials.MockSource) { ctrl := gomock.NewController(t) source := credentials.NewMockSource(ctrl) sourceCollection := credentials.NewMockSourceCollection(ctrl) sourceCollection.EXPECT().AllSources().Return([]credentials.Source{source}).AnyTimes() config.SetSources(sourceCollection) - return source + return ctrl, source } -func setTargetMock(t *testing.T, config *Configuration) *targets.MockTarget { +func setTargetMock(t *testing.T, config *Configuration) (*gomock.Controller, *targets.MockTarget) { ctrl := gomock.NewController(t) target := targets.NewMockTarget(ctrl) targetCollection := targets.NewMockTargetCollection(ctrl) targetCollection.EXPECT().AllTargets().Return([]targets.Target{target}).AnyTimes() config.SetTargets(targetCollection) - return target + return ctrl, target } func TestDeleteListOfCredentials(t *testing.T) { - config := &Configuration{} + config := &Configuration{ + CredentialsToDelete: []string{"test1", "test-not-exist"}, + } setSourceMock(t, config) - target := setTargetMock(t, config) - target.EXPECT().GetExistingCredentials().Return([]string{"test123"}).AnyTimes() + targetController, target := setTargetMock(t, config) + defer targetController.Finish() + + // The existing creds is only `test1` + target.EXPECT().GetExistingCredentials().Return([]string{"test1"}).AnyTimes() + // Does not assert on GetName + target.EXPECT().GetName().Return("").AnyTimes() + // Asserts that DeleteCredentials is called with `test1` + target.EXPECT().DeleteCredentials("test1").Return(nil) + + config.DeleteListOfCredentials(target) - target.EXPECT().DeleteCredentials("test123").Return(nil) }