// Copyright 2011 Miek Gieben. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package dns

import (
	"testing"
)

func TestCompareDomainName(t *testing.T) {
	s1 := "www.miek.nl."
	s2 := "miek.nl."
	s3 := "www.bla.nl."
	s4 := "nl.www.bla."
	s5 := "nl"

	if CompareDomainName(s1, s2) != 2 {
		t.Logf("%s with %s should be %d", s1, s2, 2)
		t.Fail()
	}
	if CompareDomainName(s1, s3) != 1 {
		t.Logf("%s with %s should be %d", s1, s3, 1)
		t.Fail()
	}
	if CompareDomainName(s3, s4) != 0 {
		t.Logf("%s with %s should be %d", s3, s4, 0)
		t.Fail()
	}
	if CompareDomainName(s1, s5) != 1 {
		t.Logf("%s with %s should be %d", s1, s5, 1)
		t.Fail()
	}
	if CompareDomainName(s1, ".") != 0 {
		t.Logf("%s with %s should be %d", s1, s5, 0)
		t.Fail()
	}
	if CompareDomainName(".", ".") != 0 {
		t.Logf("%s with %s should be %d", ".", ".", 0)
		t.Fail()
	}
}

func TestSplit(t *testing.T) {
	splitter := map[string]int{
		"www.miek.nl.":   3,
		"www.miek.nl":    3,
		`www\.miek.nl.`:  2,
		`www\\.miek.nl.`: 3,
		".":              0,
		"nl.":            1,
		"nl":             1,
		"com.":           1,
		".com.":          2,
	}
	for s, i := range splitter {
		if x := len(Split(s)); x != i {
			t.Logf("Labels should be %d, got %d: %s %v\n", i, x, s, Split(s))
			t.Fail()
		} else {
			t.Logf("%s %v\n", s, Split(s))
		}
	}
}

func TestCountLabel(t *testing.T) {
	labels := map[string]int{
		"miek.nl":       2,
		".":             0,
		"www.miek.nl.":  3,
		"www.miek.nl":   3,
		"www..miek.nl":  4,
		`www\.miek.nl`:  2,
		`www\\.miek.nl`: 3,
	}
	for owner, lab := range labels {
		if l := CountLabel(owner); l != lab {
			t.Logf("%s should have %d labels, got %d\n", owner, lab, l)
			t.Fail()
		}
	}
}

func BenchmarkSplitLabels(b *testing.B) {
	for i := 0; i < b.N; i++ {
		Split("www.example.com")
	}
}

func BenchmarkLenLabels(b *testing.B) {
	for i := 0; i < b.N; i++ {
		CountLabel("www.example.com")
	}
}

func BenchmarkCompareLabels(b *testing.B) {
	for i := 0; i < b.N; i++ {
		CompareDomainName("www.example.com", "aa.example.com")
	}
}

func BenchmarkIsSubDomain(b *testing.B) {
	for i := 0; i < b.N; i++ {
		IsSubDomain("www.example.com", "aa.example.com")
		IsSubDomain("example.com", "aa.example.com")
		IsSubDomain("miek.nl", "aa.example.com")
	}
}