11package cachedcarthage
22
33import (
4- "github.com/bitrise-io/go-utils/env"
5- "github.com/stretchr/testify/mock"
6- "io"
7-
84 "github.com/bitrise-io/go-steputils/stepconf"
95 "github.com/bitrise-io/go-utils/command"
6+ "github.com/stretchr/testify/mock"
107)
118
12- type CommandBlueprint struct {
13- Command string
14- Arguments []string
15- }
16-
179// MockCommandBuilder is an autogenerated mock type for the CommandBuilder type
1810type MockCommandBuilder struct {
1911 mock.Mock
@@ -38,9 +30,9 @@ func (m *MockCommandBuilder) Append(args ...string) CommandBuilder {
3830}
3931
4032// Command provides a mock function with given fields:
41- func (m * MockCommandBuilder ) Command (stdout io. Writer , stderr io. Writer ) command.Command {
42- args := m .Called (stdout , stderr )
43- return args .Get (0 ).(command.Command )
33+ func (m * MockCommandBuilder ) Command () * command.Model {
34+ args := m .Called ()
35+ return args .Get (0 ).(* command.Model )
4436}
4537
4638func (m * MockCommandBuilder ) GivenAddGitHubTokenSucceeds () * MockCommandBuilder {
@@ -58,27 +50,17 @@ func (m *MockCommandBuilder) GivenAppendSucceeds() *MockCommandBuilder {
5850 return m
5951}
6052
61- func (m * MockCommandBuilder ) GivenCommandReturned (blueprint CommandBlueprint ) * MockCommandBuilder {
62- command := command .NewFactory (env .NewRepository ()).Create (blueprint .Command , blueprint .Arguments , nil )
63-
64- m .On ("Command" , mock .Anything , mock .Anything ).Return (command )
53+ func (m * MockCommandBuilder ) GivenCommandReturned (command * command.Model ) * MockCommandBuilder {
54+ m .On ("Command" ).Return (command )
6555 return m
6656}
6757
68- func (m * MockCommandBuilder ) GivenCommandsReturned (commandBlueprints []CommandBlueprint ) * MockCommandBuilder {
58+ func (m * MockCommandBuilder ) GivenCommandsReturned (commands []* command. Model ) * MockCommandBuilder {
6959 index := 0
70-
71- call := m .On ("Command" , mock .Anything , mock .Anything )
60+ call := m .On ("Command" )
7261 call .RunFn = func (args mock.Arguments ) {
73- blueprint := commandBlueprints [index ]
74- command := command .NewFactory (env .NewRepository ()).Create (blueprint .Command , blueprint .Arguments , & command.Opts {
75- Stdout : args [0 ].(io.Writer ),
76- Stderr : args [1 ].(io.Writer ),
77- })
78- call .ReturnArguments = mock.Arguments {command , nil }
79-
62+ call .ReturnArguments = mock.Arguments {commands [index ], nil }
8063 index ++
8164 }
82-
8365 return m
8466}
0 commit comments