From 2449a352ef291ff8aec32025b8b965b97ffef805 Mon Sep 17 00:00:00 2001 From: Victor Vieux Date: Tue, 5 Jan 2016 10:31:47 -0800 Subject: [PATCH] add unit test Signed-off-by: Victor Vieux --- cluster/config_test.go | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/cluster/config_test.go b/cluster/config_test.go index 82a9827f12..92cd7041e2 100644 --- a/cluster/config_test.go +++ b/cluster/config_test.go @@ -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())