diff --git a/builder/docker/driver_mock.go b/builder/docker/driver_mock.go index 4cca3325b..4ffc69299 100644 --- a/builder/docker/driver_mock.go +++ b/builder/docker/driver_mock.go @@ -28,6 +28,10 @@ type MockDriver struct { IPAddressResult string IPAddressErr error + KillCalled bool + KillID string + KillError error + LoginCalled bool LoginUsername string LoginPassword string @@ -160,6 +164,12 @@ func (d *MockDriver) StartContainer(config *ContainerConfig) (string, error) { return d.StartID, d.StartError } +func (d *MockDriver) KillContainer(id string) error { + d.KillCalled = true + d.KillID = id + return d.KillError +} + func (d *MockDriver) StopContainer(id string) error { d.StopCalled = true d.StopID = id