Abstract TopologyManager Policy Merge() tests into their own function

This commit is contained in:
Kevin Klues 2019-11-04 15:11:52 +01:00
parent dee22d1fbc
commit d7d7bfcda0

View File

@ -104,7 +104,7 @@ func TestGetAffinity(t *testing.T) {
} }
} }
func TestCalculateAffinity(t *testing.T) { func TestPolicyMerge(t *testing.T) {
numaNodes := []int{0, 1} numaNodes := []int{0, 1}
tcases := []struct { tcases := []struct {
@ -745,11 +745,13 @@ func TestCalculateAffinity(t *testing.T) {
} }
for _, tc := range tcases { for _, tc := range tcases {
mngr := manager{ var providersHints []map[string][]TopologyHint
policy: tc.policy, for _, provider := range tc.hp {
hintProviders: tc.hp, hints := provider.GetTopologyHints(v1.Pod{}, v1.Container{})
providersHints = append(providersHints, hints)
} }
actual, _ := mngr.calculateAffinity(v1.Pod{}, v1.Container{})
actual, _ := tc.policy.Merge(providersHints)
if !actual.NUMANodeAffinity.IsEqual(tc.expected.NUMANodeAffinity) { if !actual.NUMANodeAffinity.IsEqual(tc.expected.NUMANodeAffinity) {
t.Errorf("Expected NUMANodeAffinity in result to be %v, got %v", tc.expected.NUMANodeAffinity, actual.NUMANodeAffinity) t.Errorf("Expected NUMANodeAffinity in result to be %v, got %v", tc.expected.NUMANodeAffinity, actual.NUMANodeAffinity)
} }