diff options
Diffstat (limited to 'middleware/kubernetes/nametemplate/nametemplate_test.go')
-rw-r--r-- | middleware/kubernetes/nametemplate/nametemplate_test.go | 129 |
1 files changed, 129 insertions, 0 deletions
diff --git a/middleware/kubernetes/nametemplate/nametemplate_test.go b/middleware/kubernetes/nametemplate/nametemplate_test.go new file mode 100644 index 000000000..a866c4fb6 --- /dev/null +++ b/middleware/kubernetes/nametemplate/nametemplate_test.go @@ -0,0 +1,129 @@ +package nametemplate + +import ( + "fmt" + "strings" + "testing" +) + +const ( + zone = 0 + namespace = 1 + service = 2 +) + +// Map of format string :: expected locations of name symbols in the format. +// -1 value indicates that symbol does not exist in format. +var exampleTemplates = map[string][]int{ + "{service}.{namespace}.{zone}": []int{2, 1, 0}, // service symbol expected @ position 0, namespace @ 1, zone @ 2 + "{namespace}.{zone}": []int{1, 0, -1}, + "": []int{-1, -1, -1}, +} + +func TestSetTemplate(t *testing.T) { + fmt.Printf("\n") + for s, expectedValue := range exampleTemplates { + + n := new(NameTemplate) + n.SetTemplate(s) + + // check the indexes resulting from calling SetTemplate() against expectedValues + if expectedValue[zone] != -1 { + if n.Element["zone"] != expectedValue[zone] { + t.Errorf("Expected zone at index '%v', instead found at index '%v' for format string '%v'", expectedValue[zone], n.Element["zone"], s) + } + } + } +} + +func TestGetServiceFromSegmentArray(t *testing.T) { + var ( + n *NameTemplate + formatString string + queryString string + splitQuery []string + expectedService string + actualService string + ) + + // Case where template contains {service} + n = new(NameTemplate) + formatString = "{service}.{namespace}.{zone}" + n.SetTemplate(formatString) + + queryString = "myservice.mynamespace.coredns" + splitQuery = strings.Split(queryString, ".") + expectedService = "myservice" + actualService = n.GetServiceFromSegmentArray(splitQuery) + + if actualService != expectedService { + t.Errorf("Expected service name '%v', instead got service name '%v' for query string '%v' and format '%v'", expectedService, actualService, queryString, formatString) + } + + // Case where template does not contain {service} + n = new(NameTemplate) + formatString = "{namespace}.{zone}" + n.SetTemplate(formatString) + + queryString = "mynamespace.coredns" + splitQuery = strings.Split(queryString, ".") + expectedService = "" + actualService = n.GetServiceFromSegmentArray(splitQuery) + + if actualService != expectedService { + t.Errorf("Expected service name '%v', instead got service name '%v' for query string '%v' and format '%v'", expectedService, actualService, queryString, formatString) + } +} + +func TestGetZoneFromSegmentArray(t *testing.T) { + var ( + n *NameTemplate + formatString string + queryString string + splitQuery []string + expectedZone string + actualZone string + ) + + // Case where template contains {zone} + n = new(NameTemplate) + formatString = "{service}.{namespace}.{zone}" + n.SetTemplate(formatString) + + queryString = "myservice.mynamespace.coredns" + splitQuery = strings.Split(queryString, ".") + expectedZone = "coredns" + actualZone = n.GetZoneFromSegmentArray(splitQuery) + + if actualZone != expectedZone { + t.Errorf("Expected zone name '%v', instead got zone name '%v' for query string '%v' and format '%v'", expectedZone, actualZone, queryString, formatString) + } + + // Case where template does not contain {zone} + n = new(NameTemplate) + formatString = "{service}.{namespace}" + n.SetTemplate(formatString) + + queryString = "mynamespace.coredns" + splitQuery = strings.Split(queryString, ".") + expectedZone = "" + actualZone = n.GetZoneFromSegmentArray(splitQuery) + + if actualZone != expectedZone { + t.Errorf("Expected zone name '%v', instead got zone name '%v' for query string '%v' and format '%v'", expectedZone, actualZone, queryString, formatString) + } + + // Case where zone is multiple segments + n = new(NameTemplate) + formatString = "{service}.{namespace}.{zone}" + n.SetTemplate(formatString) + + queryString = "myservice.mynamespace.coredns.cluster.local" + splitQuery = strings.Split(queryString, ".") + expectedZone = "coredns.cluster.local" + actualZone = n.GetZoneFromSegmentArray(splitQuery) + + if actualZone != expectedZone { + t.Errorf("Expected zone name '%v', instead got zone name '%v' for query string '%v' and format '%v'", expectedZone, actualZone, queryString, formatString) + } +} |