@ -25,13 +25,16 @@ import (
"github.com/prometheus/prometheus/discovery/targetgroup"
)
func makeNode ( name , address string , labels , annotations map [ string ] string ) * v1 . Node {
func makeNode ( name , address , providerID string , labels , annotations map [ string ] string ) * v1 . Node {
return & v1 . Node {
ObjectMeta : metav1 . ObjectMeta {
Name : name ,
Labels : labels ,
Annotations : annotations ,
} ,
Spec : v1 . NodeSpec {
ProviderID : providerID ,
} ,
Status : v1 . NodeStatus {
Addresses : [ ] v1 . NodeAddress {
{
@ -49,7 +52,7 @@ func makeNode(name, address string, labels, annotations map[string]string) *v1.N
}
func makeEnumeratedNode ( i int ) * v1 . Node {
return makeNode ( fmt . Sprintf ( "test%d" , i ) , "1.2.3.4" , map [ string ] string { } , map [ string ] string { } )
return makeNode ( fmt . Sprintf ( "test%d" , i ) , "1.2.3.4" , fmt . Sprintf ( "aws:///de-west-3a/i-%d" , i ) , map [ string ] string { } , map [ string ] string { } )
}
func TestNodeDiscoveryBeforeStart ( t * testing . T ) {
@ -61,6 +64,7 @@ func TestNodeDiscoveryBeforeStart(t *testing.T) {
obj := makeNode (
"test" ,
"1.2.3.4" ,
"aws:///nl-north-7b/i-03149834983492827" ,
map [ string ] string { "test-label" : "testvalue" } ,
map [ string ] string { "test-annotation" : "testannotationvalue" } ,
)
@ -78,6 +82,7 @@ func TestNodeDiscoveryBeforeStart(t *testing.T) {
} ,
Labels : model . LabelSet {
"__meta_kubernetes_node_name" : "test" ,
"__meta_kubernetes_node_provider_id" : "aws:///nl-north-7b/i-03149834983492827" ,
"__meta_kubernetes_node_label_test_label" : "testvalue" ,
"__meta_kubernetes_node_labelpresent_test_label" : "true" ,
"__meta_kubernetes_node_annotation_test_annotation" : "testannotationvalue" ,
@ -109,7 +114,8 @@ func TestNodeDiscoveryAdd(t *testing.T) {
} ,
} ,
Labels : model . LabelSet {
"__meta_kubernetes_node_name" : "test1" ,
"__meta_kubernetes_node_name" : "test1" ,
"__meta_kubernetes_node_provider_id" : "aws:///de-west-3a/i-1" ,
} ,
Source : "node/test1" ,
} ,
@ -146,6 +152,7 @@ func TestNodeDiscoveryUpdate(t *testing.T) {
obj2 := makeNode (
"test0" ,
"1.2.3.4" ,
"aws:///fr-south-1c/i-49508290343823952" ,
map [ string ] string { "Unschedulable" : "true" } ,
map [ string ] string { } ,
)
@ -165,6 +172,7 @@ func TestNodeDiscoveryUpdate(t *testing.T) {
"__meta_kubernetes_node_label_Unschedulable" : "true" ,
"__meta_kubernetes_node_labelpresent_Unschedulable" : "true" ,
"__meta_kubernetes_node_name" : "test0" ,
"__meta_kubernetes_node_provider_id" : "aws:///fr-south-1c/i-49508290343823952" ,
} ,
Source : "node/test0" ,
} ,