@@ -78,6 +78,20 @@ func TestInstanceIdParsedFromProviderIdCorrectly(t *testing.T) {
78
78
}
79
79
}
80
80
81
+ func TestTaskIdParsedFromProviderIdCorrectly (t * testing.T ) {
82
+ clientSet := setupKubernetesClient (t , nodeName , createFargateNode ())
83
+ k8sMp := kubernetesApiMetadataProvider {api : clientSet }
84
+
85
+ metadata , err := k8sMp .getMetadata ()
86
+
87
+ if err != nil {
88
+ t .Fatalf ("Error occurred when parsing instance ID, %v" , err )
89
+ }
90
+ if metadata .GetInstanceID () != taskId {
91
+ t .Fatalf ("Instance ID not extracted correctly, expected %s, got %s" , taskId , metadata .GetInstanceID ())
92
+ }
93
+ }
94
+
81
95
func TestRegionAndZoneExtractedCorrectlyFromLabels (t * testing.T ) {
82
96
clientSet := setupKubernetesClient (t , nodeName , createDefaultNode ())
83
97
k8sMp := kubernetesApiMetadataProvider {api : clientSet }
@@ -123,3 +137,7 @@ func createNode(nodeName string, nodeRegion string, nodeZone string, providerId
123
137
func createDefaultNode () * v1.Node {
124
138
return createNode (nodeName , nodeRegion , nodeZone , fmt .Sprintf ("aws:///%s/%s" , nodeZone , instanceId ))
125
139
}
140
+
141
+ func createFargateNode () * v1.Node {
142
+ return createNode (nodeName , nodeRegion , nodeZone , fmt .Sprintf ("aws:///%s/1234567890-%s/%s" , nodeZone , taskId , nodeName ))
143
+ }
0 commit comments