From 493481efe8f8ec3753e7ba31e3b6474696eee3ff Mon Sep 17 00:00:00 2001 From: Marcin Wielgus Date: Wed, 12 Oct 2016 17:19:47 +0200 Subject: [PATCH] Add non default replica preferences test to federation depoyment tests --- .../deployment/deploymentcontroller_test.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/federation/pkg/federation-controller/deployment/deploymentcontroller_test.go b/federation/pkg/federation-controller/deployment/deploymentcontroller_test.go index 19488ae805c..e035655bb43 100644 --- a/federation/pkg/federation-controller/deployment/deploymentcontroller_test.go +++ b/federation/pkg/federation-controller/deployment/deploymentcontroller_test.go @@ -147,6 +147,18 @@ func TestDeploymentController(t *testing.T) { clusterWatch.Add(cluster2) assert.NoError(t, CheckObjectFromChan(cluster1UpdateChan, checkDeployment(dep1, *dep1.Spec.Replicas/2))) assert.NoError(t, CheckObjectFromChan(cluster2CreateChan, checkDeployment(dep1, *dep1.Spec.Replicas/2))) + + // Add new deployment with non-default replica placement preferences. + dep2 := newDeploymentWithReplicas("deployment2", 9) + dep2.Annotations = make(map[string]string) + dep2.Annotations[FedDeploymentPreferencesAnnotation] = `{"rebalance": true, + "clusters": { + "cluster1": {"weight": 2}, + "cluster2": {"weight": 1} + }}` + deploymentsWatch.Add(dep2) + assert.NoError(t, CheckObjectFromChan(cluster1CreateChan, checkDeployment(dep2, 6))) + assert.NoError(t, CheckObjectFromChan(cluster2CreateChan, checkDeployment(dep2, 3))) } func GetDeploymentFromChan(c chan runtime.Object) *extensionsv1.Deployment {