fix: adds additional spot scheduling label (#26)

castai · Jun 8, 2021 · 8f1df5e · 8f1df5e
1 parent d3c759a
commit 8f1df5e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/internal/services/controller/controller.go b/internal/services/controller/controller.go
@@ -147,7 +147,7 @@ func (c *Controller) nodeAddHandler(
 	}
 
 	if spot {
-		node.Labels[labels.Spot] = "true"
+		node.Labels[labels.FakeSpot] = "true"
 	}
 
 	genericHandler(log, c.queue, reflect.TypeOf(&corev1.Node{}), event, node)

diff --git a/internal/services/controller/controller_test.go b/internal/services/controller/controller_test.go
@@ -36,7 +36,7 @@ func Test(t *testing.T) {
 
 	node := &v1.Node{ObjectMeta: metav1.ObjectMeta{Name: "node1", Labels: map[string]string{}}}
 	expectedNode := node.DeepCopy()
-	expectedNode.Labels[labels.Spot] = "true"
+	expectedNode.Labels[labels.FakeSpot] = "true"
 	nodeData, err := encode(expectedNode)
 	require.NoError(t, err)
 

diff --git a/pkg/labels/labels.go b/pkg/labels/labels.go
@@ -1,5 +1,6 @@
 package labels
 
 const (
-	Spot = "scheduling.cast.ai/spot"
+	Spot     = "scheduling.cast.ai/spot"
+	FakeSpot = "scheduling.cast.ai/fake-spot"
 )