Add missing method to Daemon mocks

moncho · Apr 24, 2017 · 73e02f4 · 73e02f4
1 parent 0c808ec
commit 73e02f4
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/docker/types.go → docker/api.go b/docker/types.go → docker/api.go
diff --git a/mocks/docker_daemon.go b/mocks/docker_daemon.go
@@ -246,6 +246,11 @@ func (_m *DockerDaemonMock) RemoveNetwork(id string) error {
 	return nil
 }
 
+//Services mock
+func (_m *DockerDaemonMock) Services() ([]swarm.Service, error) {
+	return nil, nil
+}
+
 // StopContainer provides a mock function with given fields: id
 func (_m *DockerDaemonMock) StopContainer(id string) error {
 	return nil

diff --git a/mocks/docker_swarmDaemon.go b/mocks/docker_swarmDaemon.go
@@ -35,6 +35,12 @@ func (_m *SwarmDockerDaemon) Nodes() ([]swarm.Node, error) {
 	return []swarm.Node{swarm.Node{ID: "1"}}, nil
 }
 
+//Services returns a list of services with 1 element
+func (_m *SwarmDockerDaemon) Services() ([]swarm.Service, error) {
+	return []swarm.Service{
+		swarm.Service{ID: "ServiceID"}}, nil
+}
+
 //Tasks mock
 func (_m *SwarmDockerDaemon) Tasks(nodeID string) ([]swarm.Task, error) {
 	return []swarm.Task{swarm.Task{NodeID: nodeID}}, nil