add unit test

Signed-off-by: Victor Vieux <vieux@docker.com>
This commit is contained in:
Victor Vieux 2016-01-05 10:31:47 -08:00
parent 5daaecdaa1
commit 2449a352ef
1 changed files with 14 additions and 0 deletions

View File

@ -90,6 +90,20 @@ func TestAddAffinity(t *testing.T) {
assert.Len(t, config.Affinities(), 1)
}
func TestRemoveAffinity(t *testing.T) {
config := BuildContainerConfig(dockerclient.ContainerConfig{})
assert.Empty(t, config.Affinities())
config.AddAffinity("image==~testimage1")
config.AddAffinity("image==~testimage2")
assert.Len(t, config.Affinities(), 2)
config.RemoveAffinity("image==~testimage1")
assert.Len(t, config.Affinities(), 1)
assert.Equal(t, config.Affinities()[0], "image==~testimage2")
}
func TestHaveNodeConstraint(t *testing.T) {
config := BuildContainerConfig(dockerclient.ContainerConfig{})
assert.False(t, config.HaveNodeConstraint())