package phonenumber

import (
	"testing"
)

type testCase struct {
	input     string
	expected  string
	expectErr bool
}

var numberTests = []testCase{
	{"(123) 456-7890", "1234567890", false},
	{"123.456.7890", "1234567890", false},
	{"1234567890", "1234567890", false},
	{"12345678901234567", "", true},
	{"21234567890", "", true},
	{"123456789", "", true},
}

func TestNumber(t *testing.T) {
	for _, test := range numberTests {
		actual, actualErr := Number(test.input)
		if actual != test.expected {
			t.Errorf("Number(%s): expected [%s], actual: [%s]", test.input, test.expected, actual)
		}
		// if we expect an error and there isn't one
		if test.expectErr && actualErr == nil {
			t.Errorf("Number(%s): expected an error, but error is nil", test.input)
		}
		// if we don't expect an error and there is one
		if !test.expectErr && actualErr != nil {
			t.Errorf("Number(%s): expected no error, but error is: %s", test.input, actualErr)
		}
	}
}

func BenchmarkNumber(b *testing.B) {
	b.StopTimer()
	for _, test := range numberTests {
		b.StartTimer()
		for i := 0; i < b.N; i++ {
			Number(test.input)
		}
		b.StopTimer()
	}
}

var areaCodeTests = []testCase{
	{"1234567890", "123", false},
	{"213.456.7890", "213", false},
	{"213.456.7890.2345", "", true},
	{"213.456", "", true},
}

func TestAreaCode(t *testing.T) {
	for _, test := range areaCodeTests {
		actual, actualErr := AreaCode(test.input)
		if actual != test.expected {
			t.Errorf("AreaCode(%s): expected [%s], actual: [%s]", test.input, test.expected, actual)
		}
		// if we expect an error and there isn't one
		if test.expectErr && actualErr == nil {
			t.Errorf("AreaCode(%s): expected an error, but error is nil", test.input)
		}
		// if we don't expect an error and there is one
		if !test.expectErr && actualErr != nil {
			t.Errorf("AreaCode(%s): expected no error, but error is: %s", test.input, actualErr)
		}
	}
}

func BenchmarkAreaCode(b *testing.B) {
	b.StopTimer()
	for _, test := range areaCodeTests {
		b.StartTimer()
		for i := 0; i < b.N; i++ {
			AreaCode(test.input)
		}
		b.StopTimer()
	}
}

var formatTests = []testCase{
	{"1234567890", "(123) 456-7890", false},
	{"11234567890", "(123) 456-7890", false},
	{"112345", "", true},
	{"11234590870986", "", true},
}

func TestFormat(t *testing.T) {
	for _, test := range formatTests {
		actual, actualErr := Format(test.input)
		if actual != test.expected {
			t.Errorf("Format(%s): expected [%s], actual: [%s]", test.input, test.expected, actual)
		}
		// if we expect an error and there isn't one
		if test.expectErr && actualErr == nil {
			t.Errorf("Format(%s): expected an error, but error is nil", test.input)
		}
		// if we don't expect an error and there is one
		if !test.expectErr && actualErr != nil {
			t.Errorf("Format(%s): expected no error, but error is: %s", test.input, actualErr)
		}
	}
}

func BenchmarkFormat(b *testing.B) {
	b.StopTimer()
	for _, test := range areaCodeTests {
		b.StartTimer()
		for i := 0; i < b.N; i++ {
			Format(test.input)
		}
		b.StopTimer()
	}
}