fix tests

Signed-off-by: aiordache <anca.iordache@docker.com>
This commit is contained in:
aiordache 2020-07-17 16:08:36 +02:00
parent fb63373a9b
commit cfa7406d73
2 changed files with 14 additions and 14 deletions

View File

@ -34,16 +34,16 @@ import (
type ContextSuiteTest struct {
suite.Suite
mockUserSelector *MockUserSelector
mockUserPrompt *mockUserPrompt
mockResourceGroupHeper *MockResourceGroupHelper
contextCreateHelper contextCreateACIHelper
}
func (suite *ContextSuiteTest) BeforeTest(suiteName, testName string) {
suite.mockUserSelector = &MockUserSelector{}
suite.mockUserPrompt = &mockUserPrompt{}
suite.mockResourceGroupHeper = &MockResourceGroupHelper{}
suite.contextCreateHelper = contextCreateACIHelper{
suite.mockUserSelector,
suite.mockUserPrompt,
suite.mockResourceGroupHeper,
}
}
@ -79,7 +79,7 @@ func (suite *ContextSuiteTest) TestCreateNewResourceGroup() {
suite.mockResourceGroupHeper.On("GetGroup", ctx, "1234", "myResourceGroup").Return(group("myResourceGroup", "eastus"), nil)
selectOptions := []string{"create a new resource group", "group1 (eastus)", "group2 (westeurope)"}
suite.mockUserSelector.On("Select", "Select a resource group", selectOptions).Return(0, nil)
suite.mockUserPrompt.On("Select", "Select a resource group", selectOptions).Return(0, nil)
suite.mockResourceGroupHeper.On("CreateOrUpdate", ctx, "1234", mock.AnythingOfType("string"), mock.AnythingOfType("resources.Group")).Return(group("newResourceGroup", "eastus"), nil)
suite.mockResourceGroupHeper.On("ListGroups", ctx, "1234").Return([]resources.Group{
group("group1", "eastus"),
@ -97,7 +97,7 @@ func (suite *ContextSuiteTest) TestSelectExistingResourceGroup() {
ctx := context.TODO()
opts := options("1234", "")
selectOptions := []string{"create a new resource group", "group1 (eastus)", "group2 (westeurope)"}
suite.mockUserSelector.On("Select", "Select a resource group", selectOptions).Return(2, nil)
suite.mockUserPrompt.On("Select", "Select a resource group", selectOptions).Return(2, nil)
suite.mockResourceGroupHeper.On("ListGroups", ctx, "1234").Return([]resources.Group{
group("group1", "eastus"),
group("group2", "westeurope"),
@ -116,7 +116,7 @@ func (suite *ContextSuiteTest) TestSelectSingleSubscriptionIdAndExistingResource
suite.mockResourceGroupHeper.On("GetSubscriptionIDs", ctx).Return([]subscription.Model{subModel("123456", "Subscription1")}, nil)
selectOptions := []string{"create a new resource group", "group1 (eastus)", "group2 (westeurope)"}
suite.mockUserSelector.On("userSelect", "Select a resource group", selectOptions).Return(2, nil)
suite.mockUserPrompt.On("Select", "Select a resource group", selectOptions).Return(2, nil)
suite.mockResourceGroupHeper.On("ListGroups", ctx, "123456").Return([]resources.Group{
group("group1", "eastus"),
group("group2", "westeurope"),
@ -138,9 +138,9 @@ func (suite *ContextSuiteTest) TestSelectSubscriptionIdAndExistingResourceGroup(
suite.mockResourceGroupHeper.On("GetSubscriptionIDs", ctx).Return([]subscription.Model{sub1, sub2}, nil)
selectOptions := []string{"Subscription1 (1234)", "Subscription2 (5678)"}
suite.mockUserSelector.On("Select", "Select a subscription ID", selectOptions).Return(1, nil)
suite.mockUserPrompt.On("Select", "Select a subscription ID", selectOptions).Return(1, nil)
selectOptions = []string{"create a new resource group", "group1 (eastus)", "group2 (westeurope)"}
suite.mockUserSelector.On("Select", "Select a resource group", selectOptions).Return(2, nil)
suite.mockUserPrompt.On("Select", "Select a resource group", selectOptions).Return(2, nil)
suite.mockResourceGroupHeper.On("ListGroups", ctx, "5678").Return([]resources.Group{
group("group1", "eastus"),
group("group2", "westeurope"),
@ -188,25 +188,25 @@ func TestContextSuite(t *testing.T) {
suite.Run(t, new(ContextSuiteTest))
}
type MockUserSelector struct {
type mockUserPrompt struct {
mock.Mock
}
func (s *MockUserSelector) Select(message string, options []string) (int, error) {
func (s *mockUserPrompt) Select(message string, options []string) (int, error) {
args := s.Called(message, options)
return args.Int(0), args.Error(1)
}
func (s *MockUserSelector) Confirm(message string, defaultValue bool) (bool, error) {
func (s *mockUserPrompt) Confirm(message string, defaultValue bool) (bool, error) {
args := s.Called(message, options)
return args.Bool(0), args.Error(1)
}
func (s *MockUserSelector) Input(message string, defaultValue string) (string, error) {
func (s *mockUserPrompt) Input(message string, defaultValue string) (string, error) {
args := s.Called(message, options)
return args.String(0), args.Error(1)
}
func (s *MockUserSelector) Password(message string) (string, error) {
func (s *mockUserPrompt) Password(message string) (string, error) {
args := s.Called(message, options)
return args.String(0), args.Error(1)
}

View File

@ -15,7 +15,7 @@ type MobyExecSuite struct {
func (sut *MobyExecSuite) TestDelegateContextTypeToMoby() {
Expect(mustDelegateToMoby("moby")).To(BeTrue())
Expect(mustDelegateToMoby("aws")).To(BeTrue())
Expect(mustDelegateToMoby("aws")).To(BeFalse())
Expect(mustDelegateToMoby("aci")).To(BeFalse())
}