134 lines
2.6 KiB
Go
134 lines
2.6 KiB
Go
|
package control
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
docker "github.com/fsouza/go-dockerclient"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func testServer() *server {
|
||
|
|
return &server{
|
||
|
|
UUID: "testuuid-something-something",
|
||
|
|
service: &service{
|
||
|
|
DockerImage: "alpine:latest",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNewDockerEnvironment(t *testing.T) {
|
||
|
|
env, err := createTestDockerEnv(nil)
|
||
|
|
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.NotNil(t, env)
|
||
|
|
assert.NotNil(t, env.client)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNewDockerEnvironmentExisting(t *testing.T) {
|
||
|
|
eenv, _ := createTestDockerEnv(nil)
|
||
|
|
eenv.Create()
|
||
|
|
|
||
|
|
env, err := createTestDockerEnv(eenv.server)
|
||
|
|
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.NotNil(t, env)
|
||
|
|
assert.NotNil(t, env.container)
|
||
|
|
|
||
|
|
eenv.Destroy()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCreateDockerEnvironment(t *testing.T) {
|
||
|
|
env, _ := createTestDockerEnv(nil)
|
||
|
|
|
||
|
|
err := env.Create()
|
||
|
|
|
||
|
|
a := assert.New(t)
|
||
|
|
a.Nil(err)
|
||
|
|
a.NotNil(env.container)
|
||
|
|
a.Equal(env.container.Name, "ptdl_testuuid")
|
||
|
|
|
||
|
|
if err := env.client.RemoveContainer(docker.RemoveContainerOptions{
|
||
|
|
ID: env.container.ID,
|
||
|
|
}); err != nil {
|
||
|
|
fmt.Println(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDestroyDockerEnvironment(t *testing.T) {
|
||
|
|
env, _ := createTestDockerEnv(nil)
|
||
|
|
env.Create()
|
||
|
|
|
||
|
|
err := env.Destroy()
|
||
|
|
|
||
|
|
_, ierr := env.client.InspectContainer(env.container.ID)
|
||
|
|
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.IsType(t, ierr, &docker.NoSuchContainer{})
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStartDockerEnvironment(t *testing.T) {
|
||
|
|
env, _ := createTestDockerEnv(nil)
|
||
|
|
env.Create()
|
||
|
|
err := env.Start()
|
||
|
|
|
||
|
|
i, ierr := env.client.InspectContainer(env.container.ID)
|
||
|
|
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.Nil(t, ierr)
|
||
|
|
assert.True(t, i.State.Running)
|
||
|
|
|
||
|
|
env.client.KillContainer(docker.KillContainerOptions{
|
||
|
|
ID: env.container.ID,
|
||
|
|
})
|
||
|
|
env.Destroy()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStopDockerEnvironment(t *testing.T) {
|
||
|
|
env, _ := createTestDockerEnv(nil)
|
||
|
|
env.Create()
|
||
|
|
env.Start()
|
||
|
|
err := env.Stop()
|
||
|
|
|
||
|
|
i, ierr := env.client.InspectContainer(env.container.ID)
|
||
|
|
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.Nil(t, ierr)
|
||
|
|
assert.False(t, i.State.Running)
|
||
|
|
|
||
|
|
env.client.KillContainer(docker.KillContainerOptions{
|
||
|
|
ID: env.container.ID,
|
||
|
|
})
|
||
|
|
env.Destroy()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestKillDockerEnvironment(t *testing.T) {
|
||
|
|
env, _ := createTestDockerEnv(nil)
|
||
|
|
env.Create()
|
||
|
|
env.Start()
|
||
|
|
err := env.Kill()
|
||
|
|
|
||
|
|
i, ierr := env.client.InspectContainer(env.container.ID)
|
||
|
|
|
||
|
|
assert.Nil(t, err)
|
||
|
|
assert.Nil(t, ierr)
|
||
|
|
assert.False(t, i.State.Running)
|
||
|
|
|
||
|
|
env.client.KillContainer(docker.KillContainerOptions{
|
||
|
|
ID: env.container.ID,
|
||
|
|
})
|
||
|
|
env.Destroy()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestExecDockerEnvironment(t *testing.T) {
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func createTestDockerEnv(s *server) (*dockerEnvironment, error) {
|
||
|
|
if s == nil {
|
||
|
|
s = testServer()
|
||
|
|
}
|
||
|
|
env, err := NewDockerEnvironment(s)
|
||
|
|
return env.(*dockerEnvironment), err
|
||
|
|
}
|