|
|
@@ -33,8 +33,8 @@ func TestRegionValueFromMapField(t *testing.T) {
|
|
|
wantpid := strings.ToLower("/subscriptions/0bd50fdf-c923-4e1e-850c-196dd3dcc5d3/resourceGroups/MC_test_test_eastus/providers/Microsoft.Compute/virtualMachines/aks-agentpool-20139558-0")
|
|
|
providerIDWant := wantRegion + "," + wantpid
|
|
|
|
|
|
- n := &v1.Node{}
|
|
|
- n.Spec.ProviderID = "azure:///subscriptions/0bd50fdf-c923-4e1e-850c-196dd3dcc5d3/resourceGroups/MC_test_test_eastus/providers/Microsoft.Compute/virtualMachines/aks-agentpool-20139558-0"
|
|
|
+ n := &clustercache.Node{}
|
|
|
+ n.SpecProviderID = "azure:///subscriptions/0bd50fdf-c923-4e1e-850c-196dd3dcc5d3/resourceGroups/MC_test_test_eastus/providers/Microsoft.Compute/virtualMachines/aks-agentpool-20139558-0"
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels[v1.LabelTopologyRegion] = wantRegion
|
|
|
got := provider.NodeValueFromMapField(providerIDMap, n, true)
|
|
|
@@ -45,24 +45,24 @@ func TestRegionValueFromMapField(t *testing.T) {
|
|
|
}
|
|
|
func TestTransformedValueFromMapField(t *testing.T) {
|
|
|
providerIDWant := "i-05445591e0d182d42"
|
|
|
- n := &v1.Node{}
|
|
|
- n.Spec.ProviderID = "aws:///us-east-1a/i-05445591e0d182d42"
|
|
|
+ n := &clustercache.Node{}
|
|
|
+ n.SpecProviderID = "aws:///us-east-1a/i-05445591e0d182d42"
|
|
|
got := provider.NodeValueFromMapField(providerIDMap, n, false)
|
|
|
if got != providerIDWant {
|
|
|
t.Errorf("Assert on '%s' want '%s' got '%s'", providerIDMap, providerIDWant, got)
|
|
|
}
|
|
|
|
|
|
providerIDWant2 := strings.ToLower("/subscriptions/0bd50fdf-c923-4e1e-850c-196dd3dcc5d3/resourceGroups/MC_test_test_eastus/providers/Microsoft.Compute/virtualMachines/aks-agentpool-20139558-0")
|
|
|
- n2 := &v1.Node{}
|
|
|
- n2.Spec.ProviderID = "azure:///subscriptions/0bd50fdf-c923-4e1e-850c-196dd3dcc5d3/resourceGroups/MC_test_test_eastus/providers/Microsoft.Compute/virtualMachines/aks-agentpool-20139558-0"
|
|
|
+ n2 := &clustercache.Node{}
|
|
|
+ n2.SpecProviderID = "azure:///subscriptions/0bd50fdf-c923-4e1e-850c-196dd3dcc5d3/resourceGroups/MC_test_test_eastus/providers/Microsoft.Compute/virtualMachines/aks-agentpool-20139558-0"
|
|
|
got2 := provider.NodeValueFromMapField(providerIDMap, n2, false)
|
|
|
if got2 != providerIDWant2 {
|
|
|
t.Errorf("Assert on '%s' want '%s' got '%s'", providerIDMap, providerIDWant2, got2)
|
|
|
}
|
|
|
|
|
|
providerIDWant3 := strings.ToLower("/subscriptions/0bd50fdf-c923-4e1e-850c-196dd3dcc5d3/resourceGroups/mc_testspot_testspot_eastus/providers/Microsoft.Compute/virtualMachineScaleSets/aks-nodepool1-19213364-vmss/virtualMachines/0")
|
|
|
- n3 := &v1.Node{}
|
|
|
- n3.Spec.ProviderID = "azure:///subscriptions/0bd50fdf-c923-4e1e-850c-196dd3dcc5d3/resourceGroups/mc_testspot_testspot_eastus/providers/Microsoft.Compute/virtualMachineScaleSets/aks-nodepool1-19213364-vmss/virtualMachines/0"
|
|
|
+ n3 := &clustercache.Node{}
|
|
|
+ n3.SpecProviderID = "azure:///subscriptions/0bd50fdf-c923-4e1e-850c-196dd3dcc5d3/resourceGroups/mc_testspot_testspot_eastus/providers/Microsoft.Compute/virtualMachineScaleSets/aks-nodepool1-19213364-vmss/virtualMachines/0"
|
|
|
got3 := provider.NodeValueFromMapField(providerIDMap, n3, false)
|
|
|
if got3 != providerIDWant3 {
|
|
|
t.Errorf("Assert on '%s' want '%s' got '%s'", providerIDMap, providerIDWant3, got3)
|
|
|
@@ -74,8 +74,8 @@ func TestNodeValueFromMapField(t *testing.T) {
|
|
|
nameWant := "gke-standard-cluster-1-pool-1-91dc432d-cg69"
|
|
|
labelFooWant := "labelfoo"
|
|
|
|
|
|
- n := &v1.Node{}
|
|
|
- n.Spec.ProviderID = providerIDWant
|
|
|
+ n := &clustercache.Node{}
|
|
|
+ n.SpecProviderID = providerIDWant
|
|
|
n.Name = nameWant
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels["foo"] = labelFooWant
|
|
|
@@ -173,8 +173,8 @@ func TestNodePriceFromCSVWithGPU(t *testing.T) {
|
|
|
LocalConfigPath: "./",
|
|
|
})
|
|
|
|
|
|
- n := &v1.Node{}
|
|
|
- n.Spec.ProviderID = providerIDWant
|
|
|
+ n := &clustercache.Node{}
|
|
|
+ n.SpecProviderID = providerIDWant
|
|
|
n.Name = nameWant
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels["foo"] = labelFooWant
|
|
|
@@ -182,8 +182,8 @@ func TestNodePriceFromCSVWithGPU(t *testing.T) {
|
|
|
n.Status.Capacity = v1.ResourceList{"nvidia.com/gpu": *resource.NewScaledQuantity(2, 0)}
|
|
|
wantPrice := "1.633700"
|
|
|
|
|
|
- n2 := &v1.Node{}
|
|
|
- n2.Spec.ProviderID = providerIDWant
|
|
|
+ n2 := &clustercache.Node{}
|
|
|
+ n2.SpecProviderID = providerIDWant
|
|
|
n2.Name = nameWant
|
|
|
n2.Labels = make(map[string]string)
|
|
|
n2.Labels["foo"] = labelFooWant
|
|
|
@@ -244,7 +244,7 @@ func TestNodePriceFromCSVSpecialChar(t *testing.T) {
|
|
|
LocalConfigPath: "./",
|
|
|
})
|
|
|
|
|
|
- n := &v1.Node{}
|
|
|
+ n := &clustercache.Node{}
|
|
|
n.Name = nameWant
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels["<http://metadata.label.servers.com/label|metadata.label.servers.com/label>"] = nameWant
|
|
|
@@ -281,8 +281,8 @@ func TestNodePriceFromCSV(t *testing.T) {
|
|
|
LocalConfigPath: "./",
|
|
|
})
|
|
|
|
|
|
- n := &v1.Node{}
|
|
|
- n.Spec.ProviderID = providerIDWant
|
|
|
+ n := &clustercache.Node{}
|
|
|
+ n.SpecProviderID = providerIDWant
|
|
|
n.Name = nameWant
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels["foo"] = labelFooWant
|
|
|
@@ -309,8 +309,8 @@ func TestNodePriceFromCSV(t *testing.T) {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- unknownN := &v1.Node{}
|
|
|
- unknownN.Spec.ProviderID = providerIDWant
|
|
|
+ unknownN := &clustercache.Node{}
|
|
|
+ unknownN.SpecProviderID = providerIDWant
|
|
|
unknownN.Name = "unknownname"
|
|
|
unknownN.Labels = make(map[string]string)
|
|
|
unknownN.Labels["foo"] = labelFooWant
|
|
|
@@ -343,24 +343,24 @@ func TestNodePriceFromCSVWithRegion(t *testing.T) {
|
|
|
LocalConfigPath: "./",
|
|
|
})
|
|
|
|
|
|
- n := &v1.Node{}
|
|
|
- n.Spec.ProviderID = providerIDWant
|
|
|
+ n := &clustercache.Node{}
|
|
|
+ n.SpecProviderID = providerIDWant
|
|
|
n.Name = nameWant
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels["foo"] = labelFooWant
|
|
|
n.Labels[v1.LabelTopologyRegion] = "regionone"
|
|
|
wantPrice := "0.133700"
|
|
|
|
|
|
- n2 := &v1.Node{}
|
|
|
- n2.Spec.ProviderID = providerIDWant
|
|
|
+ n2 := &clustercache.Node{}
|
|
|
+ n2.SpecProviderID = providerIDWant
|
|
|
n2.Name = nameWant
|
|
|
n2.Labels = make(map[string]string)
|
|
|
n2.Labels["foo"] = labelFooWant
|
|
|
n2.Labels[v1.LabelTopologyRegion] = "regiontwo"
|
|
|
wantPrice2 := "0.133800"
|
|
|
|
|
|
- n3 := &v1.Node{}
|
|
|
- n3.Spec.ProviderID = providerIDWant
|
|
|
+ n3 := &clustercache.Node{}
|
|
|
+ n3.SpecProviderID = providerIDWant
|
|
|
n3.Name = nameWant
|
|
|
n3.Labels = make(map[string]string)
|
|
|
n3.Labels["foo"] = labelFooWant
|
|
|
@@ -411,8 +411,8 @@ func TestNodePriceFromCSVWithRegion(t *testing.T) {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- unknownN := &v1.Node{}
|
|
|
- unknownN.Spec.ProviderID = "fake providerID"
|
|
|
+ unknownN := &clustercache.Node{}
|
|
|
+ unknownN.SpecProviderID = "fake providerID"
|
|
|
unknownN.Name = "unknownname"
|
|
|
unknownN.Labels = make(map[string]string)
|
|
|
unknownN.Labels[v1.LabelTopologyRegion] = "fakeregion"
|
|
|
@@ -437,11 +437,11 @@ func TestNodePriceFromCSVWithRegion(t *testing.T) {
|
|
|
}
|
|
|
|
|
|
type FakeCache struct {
|
|
|
- nodes []*v1.Node
|
|
|
+ nodes []*clustercache.Node
|
|
|
clustercache.ClusterCache
|
|
|
}
|
|
|
|
|
|
-func (f FakeCache) GetAllNodes() []*v1.Node {
|
|
|
+func (f FakeCache) GetAllNodes() []*clustercache.Node {
|
|
|
return f.nodes
|
|
|
}
|
|
|
|
|
|
@@ -449,7 +449,7 @@ func (f FakeCache) GetAllDaemonSets() []*appsv1.DaemonSet {
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
-func NewFakeNodeCache(nodes []*v1.Node) FakeCache {
|
|
|
+func NewFakeNodeCache(nodes []*clustercache.Node) FakeCache {
|
|
|
return FakeCache{
|
|
|
nodes: nodes,
|
|
|
}
|
|
|
@@ -473,14 +473,14 @@ func TestNodePriceFromCSVWithBadConfig(t *testing.T) {
|
|
|
}
|
|
|
c.DownloadPricingData()
|
|
|
|
|
|
- n := &v1.Node{}
|
|
|
- n.Spec.ProviderID = "fake"
|
|
|
+ n := &clustercache.Node{}
|
|
|
+ n.SpecProviderID = "fake"
|
|
|
n.Name = "nameWant"
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels["foo"] = "labelFooWant"
|
|
|
n.Labels[v1.LabelTopologyRegion] = "regionone"
|
|
|
|
|
|
- fc := NewFakeNodeCache([]*v1.Node{n})
|
|
|
+ fc := NewFakeNodeCache([]*clustercache.Node{n})
|
|
|
fm := FakeClusterMap{}
|
|
|
d, _ := time.ParseDuration("1m")
|
|
|
|
|
|
@@ -507,15 +507,15 @@ func TestSourceMatchesFromCSV(t *testing.T) {
|
|
|
}
|
|
|
c.DownloadPricingData()
|
|
|
|
|
|
- n := &v1.Node{}
|
|
|
- n.Spec.ProviderID = "fake"
|
|
|
+ n := &clustercache.Node{}
|
|
|
+ n.SpecProviderID = "fake"
|
|
|
n.Name = "nameWant"
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels["foo"] = "labelFooWant"
|
|
|
n.Labels[v1.LabelTopologyRegion] = "regionone"
|
|
|
|
|
|
- n2 := &v1.Node{}
|
|
|
- n2.Spec.ProviderID = "azure:///subscriptions/123a7sd-asd-1234-578a9-123abcdef/resourceGroups/case_12_STaGe_TeSt7/providers/Microsoft.Compute/virtualMachineScaleSets/vmss-agent-worker0-12stagetest7-ezggnore/virtualMachines/7"
|
|
|
+ n2 := &clustercache.Node{}
|
|
|
+ n2.SpecProviderID = "azure:///subscriptions/123a7sd-asd-1234-578a9-123abcdef/resourceGroups/case_12_STaGe_TeSt7/providers/Microsoft.Compute/virtualMachineScaleSets/vmss-agent-worker0-12stagetest7-ezggnore/virtualMachines/7"
|
|
|
n2.Labels = make(map[string]string)
|
|
|
n2.Labels[v1.LabelTopologyRegion] = "eastus2"
|
|
|
n2.Labels["foo"] = "labelFooWant"
|
|
|
@@ -532,14 +532,14 @@ func TestSourceMatchesFromCSV(t *testing.T) {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- n3 := &v1.Node{}
|
|
|
- n3.Spec.ProviderID = "fake"
|
|
|
+ n3 := &clustercache.Node{}
|
|
|
+ n3.SpecProviderID = "fake"
|
|
|
n3.Name = "nameWant"
|
|
|
n3.Labels = make(map[string]string)
|
|
|
n3.Labels[v1.LabelTopologyRegion] = "eastus2"
|
|
|
n3.Labels[v1.LabelInstanceTypeStable] = "Standard_F32s_v2"
|
|
|
|
|
|
- fc := NewFakeNodeCache([]*v1.Node{n, n2, n3})
|
|
|
+ fc := NewFakeNodeCache([]*clustercache.Node{n, n2, n3})
|
|
|
fm := FakeClusterMap{}
|
|
|
d, _ := time.ParseDuration("1m")
|
|
|
|
|
|
@@ -568,8 +568,8 @@ func TestSourceMatchesFromCSV(t *testing.T) {
|
|
|
}
|
|
|
|
|
|
func TestNodePriceFromCSVWithCase(t *testing.T) {
|
|
|
- n := &v1.Node{}
|
|
|
- n.Spec.ProviderID = "azure:///subscriptions/123a7sd-asd-1234-578a9-123abcdef/resourceGroups/case_12_STaGe_TeSt7/providers/Microsoft.Compute/virtualMachineScaleSets/vmss-agent-worker0-12stagetest7-ezggnore/virtualMachines/7"
|
|
|
+ n := &clustercache.Node{}
|
|
|
+ n.SpecProviderID = "azure:///subscriptions/123a7sd-asd-1234-578a9-123abcdef/resourceGroups/case_12_STaGe_TeSt7/providers/Microsoft.Compute/virtualMachineScaleSets/vmss-agent-worker0-12stagetest7-ezggnore/virtualMachines/7"
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels[v1.LabelTopologyRegion] = "eastus2"
|
|
|
wantPrice := "0.13370357"
|
|
|
@@ -608,7 +608,7 @@ func TestNodePriceFromCSVMixed(t *testing.T) {
|
|
|
LocalConfigPath: "./",
|
|
|
})
|
|
|
|
|
|
- n := &v1.Node{}
|
|
|
+ n := &clustercache.Node{}
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels["TestClusterUsage"] = labelFooWant
|
|
|
n.Labels["nvidia.com/gpu_type"] = "a100-ondemand"
|
|
|
@@ -616,7 +616,7 @@ func TestNodePriceFromCSVMixed(t *testing.T) {
|
|
|
wantPrice := "1.904110"
|
|
|
|
|
|
labelFooWant2 := "Reserved"
|
|
|
- n2 := &v1.Node{}
|
|
|
+ n2 := &clustercache.Node{}
|
|
|
n2.Labels = make(map[string]string)
|
|
|
n2.Labels["TestClusterUsage"] = labelFooWant2
|
|
|
n2.Labels["nvidia.com/gpu_type"] = "a100-reserved"
|
|
|
@@ -659,8 +659,8 @@ func TestNodePriceFromCSVMixed(t *testing.T) {
|
|
|
}
|
|
|
|
|
|
func TestNodePriceFromCSVByClass(t *testing.T) {
|
|
|
- n := &v1.Node{}
|
|
|
- n.Spec.ProviderID = "fakeproviderid"
|
|
|
+ n := &clustercache.Node{}
|
|
|
+ n.SpecProviderID = "fakeproviderid"
|
|
|
n.Labels = make(map[string]string)
|
|
|
n.Labels[v1.LabelTopologyRegion] = "eastus2"
|
|
|
n.Labels[v1.LabelInstanceTypeStable] = "Standard_F32s_v2"
|
|
|
@@ -691,8 +691,8 @@ func TestNodePriceFromCSVByClass(t *testing.T) {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
- n2 := &v1.Node{}
|
|
|
- n2.Spec.ProviderID = "fakeproviderid"
|
|
|
+ n2 := &clustercache.Node{}
|
|
|
+ n2.SpecProviderID = "fakeproviderid"
|
|
|
n2.Labels = make(map[string]string)
|
|
|
n2.Labels[v1.LabelTopologyRegion] = "fakeregion"
|
|
|
n2.Labels[v1.LabelInstanceTypeStable] = "Standard_F32s_v2"
|