From 8a2012e821cc2437695fc3e4d0611500ffff371c Mon Sep 17 00:00:00 2001
From: Dylan Scott <4922401+dylanrhysscott@users.noreply.github.com>
Date: Tue, 25 Feb 2025 19:05:32 +0000
Subject: [PATCH] CON-11904 Ensure taints are correctly returned via node
 template endpoint (#795)

---
 kubernetes.go      | 1 +
 kubernetes_test.go | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/kubernetes.go b/kubernetes.go
index 4a1530a..b7dceb1 100644
--- a/kubernetes.go
+++ b/kubernetes.go
@@ -446,6 +446,7 @@ type KubernetesNodePoolTemplateResponse struct {
 // KubernetesNodePool represents the node pool template data for a given pool.
 type KubernetesNodePoolTemplate struct {
 	Labels      map[string]string            `json:"labels,omitempty"`
+	Taints      []string                     `json:"taints,omitempty"`
 	Capacity    *KubernetesNodePoolResources `json:"capacity,omitempty"`
 	Allocatable *KubernetesNodePoolResources `json:"allocatable,omitempty"`
 }
diff --git a/kubernetes_test.go b/kubernetes_test.go
index b59d7a2..cfb7b0a 100644
--- a/kubernetes_test.go
+++ b/kubernetes_test.go
@@ -1337,6 +1337,7 @@ func TestKubernetesClusters_GetNodePoolTemplate(t *testing.T) {
 		Name:        "pool-a",
 		Slug:        "s-1vcpu-2gb",
 		Template: &KubernetesNodePoolTemplate{
+			Taints: []string{"some-key=some-value:NoSchedule"},
 			Labels: map[string]string{
 				"some-label": "some-value",
 			},
@@ -1360,6 +1361,7 @@ func TestKubernetesClusters_GetNodePoolTemplate(t *testing.T) {
       "labels": {
          "some-label": "some-value"
       },
+      "taints": ["some-key=some-value:NoSchedule"],
       "capacity": {
          "cpu": 1,
          "memory": "2048Mi",