Skip to content

Commit

Permalink
Added mock interfaces for unit testing
Browse files Browse the repository at this point in the history
  • Loading branch information
Ed Cranford committed Mar 6, 2018
1 parent 9dc8645 commit ee0f573
Show file tree
Hide file tree
Showing 3 changed files with 120 additions and 42 deletions.
48 changes: 40 additions & 8 deletions cli/api/apimocks/API.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,38 @@ func (_m *API) AddHost(_a0 api.HostConfig) (*host.Host, []byte, error) {
return r0, r1, r2
}

// AddHostPrivate provides a mock function with given fields: _a0
func (_m *API) AddHostPrivate(_a0 api.HostConfig) (*host.Host, []byte, error) {
ret := _m.Called(_a0)

var r0 *host.Host
if rf, ok := ret.Get(0).(func(api.HostConfig) *host.Host); ok {
r0 = rf(_a0)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*host.Host)
}
}

var r1 []byte
if rf, ok := ret.Get(1).(func(api.HostConfig) []byte); ok {
r1 = rf(_a0)
} else {
if ret.Get(1) != nil {
r1 = ret.Get(1).([]byte)
}
}

var r2 error
if rf, ok := ret.Get(2).(func(api.HostConfig) error); ok {
r2 = rf(_a0)
} else {
r2 = ret.Error(2)
}

return r0, r1, r2
}

// AddPublicEndpointPort provides a mock function with given fields: serviceid, endpointName, portAddr, usetls, protocol, isEnabled, restart
func (_m *API) AddPublicEndpointPort(serviceid string, endpointName string, portAddr string, usetls bool, protocol string, isEnabled bool, restart bool) (*servicedefinition.Port, error) {
ret := _m.Called(serviceid, endpointName, portAddr, usetls, protocol, isEnabled, restart)
Expand Down Expand Up @@ -233,16 +265,16 @@ func (_m *API) RemoveIP(args []string) error {

// SetIP provides a mock function with given fields: _a0
func (_m *API) SetIP(_a0 api.IPConfig) error {
ret := _m.Called(_a0)
ret := _m.Called(_a0)

var r0 error
if rf, ok := ret.Get(0).(func(api.IPConfig) error); ok {
r0 = rf(_a0)
} else {
r0 = ret.Error(0)
}
var r0 error
if rf, ok := ret.Get(0).(func(api.IPConfig) error); ok {
r0 = rf(_a0)
} else {
r0 = ret.Error(0)
}

return r0
return r0
}

// AttachServiceInstance provides a mock function with given fields: serviceID, instanceID, command, args
Expand Down
63 changes: 43 additions & 20 deletions facade/mocks/FacadeInterface.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,29 @@ func (_m *FacadeInterface) AddHost(ctx datastore.Context, entity *host.Host) ([]
return r0, r1
}

// AddHostPrivate provides a mock function with given fields: ctx, entity
func (_m *FacadeInterface) AddHostPrivate(ctx datastore.Context, entity *host.Host) ([]byte, error) {
ret := _m.Called(ctx, entity)

var r0 []byte
if rf, ok := ret.Get(0).(func(datastore.Context, *host.Host) []byte); ok {
r0 = rf(ctx, entity)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]byte)
}
}

var r1 error
if rf, ok := ret.Get(1).(func(datastore.Context, *host.Host) error); ok {
r1 = rf(ctx, entity)
} else {
r1 = ret.Error(1)
}

return r0, r1
}

// AddPublicEndpointPort provides a mock function with given fields: ctx, serviceid, endpointName, portAddr, usetls, protocol, isEnabled, restart
func (_m *FacadeInterface) AddPublicEndpointPort(ctx datastore.Context, serviceid string, endpointName string, portAddr string, usetls bool, protocol string, isEnabled bool, restart bool) (*servicedefinition.Port, error) {
ret := _m.Called(ctx, serviceid, endpointName, portAddr, usetls, protocol, isEnabled, restart)
Expand Down Expand Up @@ -189,24 +212,24 @@ func (_m *FacadeInterface) RemoveIPs(ctx datastore.Context, args []string) error
if rf, ok := ret.Get(0).(func(datastore.Context, []string) error); ok {
r0 = rf(ctx, args)
} else {
r0 = ret.Error(0)
r0 = ret.Error(0)
}

return r0
}

// SetIPs provides a mock function with given fields: ctx, assignmentRequest
func (_m *FacadeInterface) SetIPs(ctx datastore.Context, assignmentRequest addressassignment.AssignmentRequest) error {
ret := _m.Called(ctx, assignmentRequest)
ret := _m.Called(ctx, assignmentRequest)

var r0 error
if rf, ok := ret.Get(0).(func(datastore.Context, addressassignment.AssignmentRequest) error); ok {
r0 = rf(ctx, assignmentRequest)
} else {
r0 = ret.Error(0)
}
var r0 error
if rf, ok := ret.Get(0).(func(datastore.Context, addressassignment.AssignmentRequest) error); ok {
r0 = rf(ctx, assignmentRequest)
} else {
r0 = ret.Error(0)
}

return r0
return r0
}

// ClearEmergencyStopFlag provides a mock function with given fields: ctx, serviceID
Expand Down Expand Up @@ -1498,7 +1521,7 @@ func (_m *FacadeInterface) SetHostExpiration(ctx datastore.Context, hostID strin
}

//StartService provides a mock function with given fields: ctx, ScheduleServiceRequest
func (_m *FacadeInterface) StartService(ctx datastore.Context, request dao.ScheduleServiceRequest) (int, error) {
func (_m *FacadeInterface) StartService(ctx datastore.Context, request dao.ScheduleServiceRequest) (int, error) {

ret := _m.Called(ctx, request)

Expand All @@ -1520,7 +1543,7 @@ func (_m *FacadeInterface) StartService(ctx datastore.Context, request dao.Sche
}

//StopService provides a mock function with given fields: ctx, ScheduleServiceRequest
func (_m *FacadeInterface) StopService(ctx datastore.Context, request dao.ScheduleServiceRequest) (int, error) {
func (_m *FacadeInterface) StopService(ctx datastore.Context, request dao.ScheduleServiceRequest) (int, error) {

ret := _m.Called(ctx, request)

Expand All @@ -1540,8 +1563,9 @@ func (_m *FacadeInterface) StopService(ctx datastore.Context, request dao.Sched

return r0, r1
}

//RestartService provides a mock function with given fields: ctx, ScheduleServiceRequest
func (_m *FacadeInterface) RestartService(ctx datastore.Context, request dao.ScheduleServiceRequest) (int, error) {
func (_m *FacadeInterface) RestartService(ctx datastore.Context, request dao.ScheduleServiceRequest) (int, error) {

ret := _m.Called(ctx, request)

Expand All @@ -1562,7 +1586,6 @@ func (_m *FacadeInterface) RestartService(ctx datastore.Context, request dao.Sc
return r0, r1
}


// SyncServiceRegistry provides a mock function with given fields: ctx, svc
func (_m *FacadeInterface) SyncServiceRegistry(ctx datastore.Context, svc *service.Service) error {
ret := _m.Called(ctx, svc)
Expand Down Expand Up @@ -1745,13 +1768,13 @@ func (_m *FacadeInterface) GetServiceNamePath(ctx datastore.Context, serviceID s
}

var r1 string
if rf, ok := ret.Get(0).(func(datastore.Context, string) string); ok {
r1 = rf(ctx, serviceID)
} else {
if ret.Get(0) != nil {
r1 = ret.Get(0).(string)
}
}
if rf, ok := ret.Get(0).(func(datastore.Context, string) string); ok {
r1 = rf(ctx, serviceID)
} else {
if ret.Get(0) != nil {
r1 = ret.Get(0).(string)
}
}

var r2 error
if rf, ok := ret.Get(1).(func(datastore.Context, string) error); ok {
Expand Down
51 changes: 37 additions & 14 deletions rpc/master/mocks/ClientInterface.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,29 @@ func (_m *ClientInterface) AddHost(h host.Host) ([]byte, error) {
return r0, r1
}

// AddHostPrivate provides a mock function with given fields: h
func (_m *ClientInterface) AddHostPrivate(h host.Host) ([]byte, error) {
ret := _m.Called(h)

var r0 []byte
if rf, ok := ret.Get(0).(func(host.Host) []byte); ok {
r0 = rf(h)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]byte)
}
}

var r1 error
if rf, ok := ret.Get(1).(func(host.Host) error); ok {
r1 = rf(h)
} else {
r1 = ret.Error(1)
}

return r0, r1
}

// AddPublicEndpointPort provides a mock function with given fields: serviceid, endpointName, portAddr, usetls, protocol, isEnabled, restart
func (_m *ClientInterface) AddPublicEndpointPort(serviceid string, endpointName string, portAddr string, usetls bool, protocol string, isEnabled bool, restart bool) (*servicedefinition.Port, error) {
ret := _m.Called(serviceid, endpointName, portAddr, usetls, protocol, isEnabled, restart)
Expand Down Expand Up @@ -421,11 +444,11 @@ func (_m *ClientInterface) GetEvaluatedService(serviceID string, instanceID int)
}

var r2 string
if rf, ok := ret.Get(1).(func(string, int) string); ok {
r2 = rf(serviceID, instanceID)
} else {
r2 = ret.Get(1).(string)
}
if rf, ok := ret.Get(1).(func(string, int) string); ok {
r2 = rf(serviceID, instanceID)
} else {
r2 = ret.Get(1).(string)
}

var r3 error
if rf, ok := ret.Get(2).(func(string, int) error); ok {
Expand Down Expand Up @@ -1161,7 +1184,7 @@ func (_m *ClientInterface) WaitService(serviceIDs []string, state service.Desire

func (_m *ClientInterface) RemoveIPs(args []string) error {
ret := _m.Called(args)
var r0 error
var r0 error
if rf, ok := ret.Get(0).(func([]string) error); ok {
r0 = rf(args)
} else {
Expand All @@ -1171,16 +1194,16 @@ func (_m *ClientInterface) RemoveIPs(args []string) error {
}

func (_m *ClientInterface) SetIPs(assignmentRequest addressassignment.AssignmentRequest) error {
ret := _m.Called(assignmentRequest)
ret := _m.Called(assignmentRequest)

var r0 error
if rf, ok := ret.Get(0).(func(addressassignment.AssignmentRequest) error); ok {
r0 = rf(assignmentRequest)
} else {
r0 = ret.Error(0)
}
var r0 error
if rf, ok := ret.Get(0).(func(addressassignment.AssignmentRequest) error); ok {
r0 = rf(assignmentRequest)
} else {
r0 = ret.Error(0)
}

return r0
return r0
}

var _ master.ClientInterface = (*ClientInterface)(nil)

0 comments on commit ee0f573

Please sign in to comment.