func TestLog(t *testing.T) {
t.Parallel()
workerHost := "worker"
targetContainer := 1
mockGetter := new(testutils.Getter)
mockGetter.On("Client", mock.Anything).Return(&clientMock.Client{}, nil)
mockGetter.On("ContainerClient", mock.Anything, mock.Anything).Return(
&clientMock.Client{
ContainerReturn: []db.Container{
{
StitchID: targetContainer,
DockerID: "foo",
},
},
HostReturn: workerHost,
}, nil)
mockSSHClient := new(testutils.MockSSHClient)
logsCmd := Log{
privateKey: "key",
targetContainer: targetContainer,
shouldTail: true,
showTimestamps: true,
sinceTimestamp: "2006-01-02T15:04:05",
SSHClient: mockSSHClient,
clientGetter: mockGetter,
common: &commonFlags{
host: api.DefaultSocket,
},
}
mockSSHClient.On("Connect", workerHost, "key").Return(nil)
mockSSHClient.On("Run", "docker logs --since=2006-01-02T15:04:05 --timestamps "+
"--follow foo").Return(nil)
mockSSHClient.On("Disconnect").Return(nil)
logsCmd.Run()
mockSSHClient.AssertExpectations(t)
}
func TestExecNoPTY(t *testing.T) {
isTerminal = func() bool {
return false
}
workerHost := "worker"
targetContainer := 1
mockGetter := new(testutils.Getter)
mockGetter.On("Client", mock.Anything).Return(&clientMock.Client{}, nil)
mockGetter.On("ContainerClient", mock.Anything, mock.Anything).Return(
&clientMock.Client{
ContainerReturn: []db.Container{
{
StitchID: targetContainer,
DockerID: "foo",
},
},
HostReturn: workerHost,
}, nil)
mockSSHClient := new(testutils.MockSSHClient)
execCmd := Exec{
command: "cat /etc/hosts",
targetContainer: targetContainer,
SSHClient: mockSSHClient,
clientGetter: mockGetter,
common: &commonFlags{
host: api.DefaultSocket,
},
}
mockSSHClient.On("Connect", workerHost, "").Return(nil)
mockSSHClient.On("Run", "docker exec foo cat /etc/hosts").Return(nil)
mockSSHClient.On("Disconnect").Return(nil)
execCmd.Run()
mockSSHClient.AssertExpectations(t)
}