rio/internal/server/tls_test.go

package server

import "testing"

const (
	pagesDomain         = "pages.local"
	pagesDomainWildcard = "*.pages.local"
)

func equals(a, b []string) bool {
	if len(a) != len(b) {
		return false
	}

	for i, _ := range a {
		if a[i] != b[i] {
			return false
		}
	}

	return true
}

func TestDomainListBare(t *testing.T) {
	expect := []string{pagesDomain, pagesDomainWildcard}
	res := buildDomainList(pagesDomain, pagesDomain)
	if !equals(res, expect) {
		t.Fatalf("%v != %v", res, expect)
	}
}

func TestDomainListSubdomain(t *testing.T) {
	expect := []string{pagesDomain, pagesDomainWildcard}
	res := buildDomainList("user."+pagesDomain, pagesDomain)
	if !equals(res, expect) {
		t.Fatalf("%v != %v", res, expect)
	}
}

func TestDomainListCNAME(t *testing.T) {
	expect := []string{"testdomain.example"}
	res := buildDomainList("testdomain.example", pagesDomain)
	if !equals(res, expect) {
		t.Fatalf("%v != %v", res, expect)
	}
}

func TestDomainKeyBare(t *testing.T) {
	res := getDomainKey(pagesDomain, pagesDomain)
	if res != pagesDomainWildcard {
		t.Fatalf("%s != %s", res, pagesDomainWildcard)
	}
}

func TestDomainKeySubdomain(t *testing.T) {
	res := getDomainKey("user."+pagesDomain, pagesDomain)
	if res != pagesDomainWildcard {
		t.Fatalf("%s != %s", res, pagesDomainWildcard)
	}
}

func TestDomainKeyCNAME(t *testing.T) {
	res := getDomainKey("testdomain.example", pagesDomain)
	if res != "testdomain.example" {
		t.Fatalf("%s != %s", res, "testdomain.example")
	}
}