AWS: Add tests for awsTagging.hasClusterTag
Among other things these tests verify the fix for issue #64230.
This commit is contained in:
		| @@ -114,3 +114,68 @@ func TestFindClusterID(t *testing.T) { | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestHasClusterTag(t *testing.T) { | ||||
| 	awsServices := NewFakeAWSServices(TestClusterId) | ||||
| 	c, err := newAWSCloud(CloudConfig{}, awsServices) | ||||
| 	if err != nil { | ||||
| 		t.Errorf("Error building aws cloud: %v", err) | ||||
| 		return | ||||
| 	} | ||||
| 	grid := []struct { | ||||
| 		Tags     map[string]string | ||||
| 		Expected bool | ||||
| 	}{ | ||||
| 		{ | ||||
| 			Tags: map[string]string{}, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Tags: map[string]string{ | ||||
| 				TagNameKubernetesClusterLegacy: TestClusterId, | ||||
| 			}, | ||||
| 			Expected: true, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Tags: map[string]string{ | ||||
| 				TagNameKubernetesClusterLegacy: "a", | ||||
| 			}, | ||||
| 			Expected: false, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Tags: map[string]string{ | ||||
| 				TagNameKubernetesClusterPrefix + TestClusterId: "owned", | ||||
| 			}, | ||||
| 			Expected: true, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Tags: map[string]string{ | ||||
| 				TagNameKubernetesClusterPrefix + TestClusterId: "", | ||||
| 			}, | ||||
| 			Expected: true, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Tags: map[string]string{ | ||||
| 				TagNameKubernetesClusterLegacy:                 "a", | ||||
| 				TagNameKubernetesClusterPrefix + TestClusterId: "shared", | ||||
| 			}, | ||||
| 			Expected: true, | ||||
| 		}, | ||||
| 		{ | ||||
| 			Tags: map[string]string{ | ||||
| 				TagNameKubernetesClusterPrefix + TestClusterId: "shared", | ||||
| 				TagNameKubernetesClusterPrefix + "b":           "shared", | ||||
| 			}, | ||||
| 			Expected: true, | ||||
| 		}, | ||||
| 	} | ||||
| 	for _, g := range grid { | ||||
| 		var ec2Tags []*ec2.Tag | ||||
| 		for k, v := range g.Tags { | ||||
| 			ec2Tags = append(ec2Tags, &ec2.Tag{Key: aws.String(k), Value: aws.String(v)}) | ||||
| 		} | ||||
| 		result := c.tagging.hasClusterTag(ec2Tags) | ||||
| 		if result != g.Expected { | ||||
| 			t.Errorf("Unexpected result for tags %v: %t", g.Tags, result) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
		Reference in New Issue
	
	Block a user
	 Arto Jantunen
					Arto Jantunen