EdgeCommon/pkg/dnsconfigs/routes_test.go

// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.

package dnsconfigs

import (
	"strings"
	"testing"
)

func TestRoutes(t *testing.T) {
	// 检查代号是否有空，或者代号是否重复

	var codeMap = map[string]bool{} // code => true

	for _, routes := range [][]*Route{
		AllDefaultChinaProvinceRoutes,
		AllDefaultISPRoutes,
		AllDefaultWorldRegionRoutes,
	} {
		for _, route := range routes {
			if len(route.Name) == 0 {
				t.Fatal("route name should not empty:", route)
			}
			if len(route.AliasNames) == 0 {
				t.Fatal("route alias names should not empty:", route)
			}
			if len(route.Code) == 0 || route.Code == "world:" {
				t.Fatal("route code should not empty:", route)
			}

			_, ok := codeMap[route.Code]
			if ok {
				t.Fatal("code duplicated:", route)
			}

			codeMap[route.Code] = true

			if strings.HasPrefix(route.Code, "world:sp:") || (strings.HasPrefix(route.Code, "world:") && route.Code != "world:UAR" && len(route.Code) != 8) {
				t.Log("no shorten code:", route)
			}
		}
	}
}

func TestFindDefaultRoute(t *testing.T) {
	t.Log(FindDefaultRoute("isp:china_unicom"))
	t.Log(FindDefaultRoute("china:province:beijing"))
	t.Log(FindDefaultRoute("world:CN"))
	t.Log(FindDefaultRoute("world:US"))
}

func TestRoutes_Markdown(t *testing.T) {
	var markdown = ""

	for index, routes := range [][]*Route{
		AllDefaultRoutes,
		AllDefaultChinaProvinceRoutes,
		AllDefaultISPRoutes,
		AllDefaultWorldRegionRoutes,
	} {
		switch index {
		case 0:
			markdown += "## 默认线路\n"
		case 1:
			markdown += "## 中国地区\n"
		case 2:
			markdown += "## 运营商\n"
		case 3:
			markdown += "## 全球地区\n"
		}

		markdown += "| 名称 | 代号 |\n"
		markdown += "|-----------|-----------|\n"
		for _, route := range routes {
			markdown += "| "  + route.Name + " | " + route.Code + " |\n"
		}
		markdown += "\n"
	}
	t.Log(markdown)
}

func BenchmarkFindDefaultRoute(b *testing.B) {
	for i := 0; i < b.N; i++ {
		_ = FindDefaultRoute("world:CN")
	}
}