161 lines
4.0 KiB
Go
161 lines
4.0 KiB
Go
package docker
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
"time"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/client"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type Mock struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *Mock) GetClient() *client.Client {
|
|
args := m.Called()
|
|
return args.Get(0).(*client.Client)
|
|
}
|
|
|
|
func (m *Mock) NetworkCreate(id string, opts types.NetworkCreate) error {
|
|
args := m.Called(id, opts)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *Mock) NetworkConnect(container, network, ip string) (string, error) {
|
|
args := m.Called(container, network, ip)
|
|
return args.String(0), args.Error(1)
|
|
}
|
|
|
|
func (m *Mock) NetworkInspect(id string) (types.NetworkResource, error) {
|
|
args := m.Called(id)
|
|
return args.Get(0).(types.NetworkResource), args.Error(1)
|
|
}
|
|
|
|
func (m *Mock) DaemonInfo() (types.Info, error) {
|
|
args := m.Called()
|
|
return args.Get(0).(types.Info), args.Error(1)
|
|
}
|
|
|
|
func (m *Mock) DaemonHost() string {
|
|
args := m.Called()
|
|
return args.String(0)
|
|
}
|
|
|
|
func (m *Mock) GetSwarmPorts() ([]string, []uint16, error) {
|
|
args := m.Called()
|
|
return args.Get(0).([]string), args.Get(1).([]uint16), args.Error(2)
|
|
}
|
|
|
|
func (m *Mock) GetPorts() ([]uint16, error) {
|
|
args := m.Called()
|
|
return args.Get(0).([]uint16), args.Error(1)
|
|
}
|
|
func (m *Mock) ContainerStats(name string) (io.ReadCloser, error) {
|
|
args := m.Called(name)
|
|
return args.Get(0).(io.ReadCloser), args.Error(1)
|
|
}
|
|
func (m *Mock) ContainerResize(name string, rows, cols uint) error {
|
|
args := m.Called(name, rows, cols)
|
|
return args.Error(0)
|
|
}
|
|
func (m *Mock) ContainerRename(old, new string) error {
|
|
args := m.Called(old, new)
|
|
return args.Error(0)
|
|
}
|
|
func (m *Mock) CreateAttachConnection(name string) (net.Conn, error) {
|
|
args := m.Called(name)
|
|
return args.Get(0).(net.Conn), args.Error(1)
|
|
}
|
|
func (m *Mock) CopyToContainer(containerName, destination, fileName string, content io.Reader) error {
|
|
args := m.Called(containerName, destination, fileName, content)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *Mock) CopyFromContainer(containerName, filePath string) (io.Reader, error) {
|
|
args := m.Called(containerName, filePath)
|
|
return args.Get(0).(io.Reader), args.Error(1)
|
|
}
|
|
func (m *Mock) ContainerDelete(id string) error {
|
|
args := m.Called(id)
|
|
return args.Error(0)
|
|
}
|
|
func (m *Mock) ContainerCreate(opts CreateContainerOpts) error {
|
|
args := m.Called(opts)
|
|
return args.Error(0)
|
|
}
|
|
func (m *Mock) ContainerIPs(id string) (map[string]string, error) {
|
|
args := m.Called(id)
|
|
return args.Get(0).(map[string]string), args.Error(1)
|
|
}
|
|
|
|
func (m *Mock) ExecAttach(instanceName string, command []string, out io.Writer) (int, error) {
|
|
args := m.Called(instanceName, command, out)
|
|
return args.Int(0), args.Error(1)
|
|
}
|
|
func (m *Mock) NetworkDisconnect(containerId, networkId string) error {
|
|
args := m.Called(containerId, networkId)
|
|
return args.Error(0)
|
|
}
|
|
func (m *Mock) NetworkDelete(id string) error {
|
|
args := m.Called(id)
|
|
return args.Error(0)
|
|
}
|
|
func (m *Mock) Exec(instanceName string, command []string) (int, error) {
|
|
args := m.Called(instanceName, command)
|
|
return args.Int(0), args.Error(1)
|
|
}
|
|
func (m *Mock) SwarmInit(advertiseAddr string) (*SwarmTokens, error) {
|
|
args := m.Called(advertiseAddr)
|
|
return args.Get(0).(*SwarmTokens), args.Error(1)
|
|
}
|
|
func (m *Mock) SwarmJoin(addr, token string) error {
|
|
args := m.Called(addr, token)
|
|
return args.Error(0)
|
|
}
|
|
func (m *Mock) ConfigCreate(name string, labels map[string]string, data []byte) error {
|
|
args := m.Called(name, labels, data)
|
|
return args.Error(0)
|
|
}
|
|
func (m *Mock) ConfigDelete(name string) error {
|
|
args := m.Called(name)
|
|
return args.Error(0)
|
|
}
|
|
|
|
type MockConn struct {
|
|
}
|
|
|
|
func (m *MockConn) Read(b []byte) (n int, err error) {
|
|
return len(b), nil
|
|
}
|
|
|
|
func (m *MockConn) Write(b []byte) (n int, err error) {
|
|
return len(b), nil
|
|
}
|
|
|
|
func (m *MockConn) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (m *MockConn) LocalAddr() net.Addr {
|
|
return &net.IPAddr{}
|
|
}
|
|
|
|
func (m *MockConn) RemoteAddr() net.Addr {
|
|
return &net.IPAddr{}
|
|
}
|
|
|
|
func (m *MockConn) SetDeadline(t time.Time) error {
|
|
return nil
|
|
}
|
|
|
|
func (m *MockConn) SetReadDeadline(t time.Time) error {
|
|
return nil
|
|
}
|
|
|
|
func (m *MockConn) SetWriteDeadline(t time.Time) error {
|
|
return nil
|
|
}
|