package utils

import (
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestSplitDomainPort(t *testing.T) {
	domain, port, ok := SplitDomainPort("www.example.com:5612", 443)
	assert.True(t, ok, "Output should be true")
	assert.Equal(t, "www.example.com", domain)
	assert.Equal(t, int(5612), port)

	domain, port, ok = SplitDomainPort("example.com", 443)
	assert.True(t, ok, "Output should be true")
	assert.Equal(t, "example.com", domain)
	assert.Equal(t, int(443), port)
}

func TestDomainWithoutPort(t *testing.T) {
	domain := GetDomainWithoutPort("www.example.com:5612")
	assert.Equal(t, "www.example.com", domain)

	domain = GetDomainWithoutPort("example.com:443")
	assert.Equal(t, "example.com", domain)

	domain = GetDomainWithoutPort("www.example.com")
	assert.Equal(t, "www.example.com", domain)

	domain = GetDomainWithoutPort("example.com")
	assert.Equal(t, "example.com", domain)
}

func TestReplaceSubdomainWithWildcard(t *testing.T) {
	domain, ok := ReplaceSubdomainWithWildcard("www.example.com")
	assert.True(t, ok, "Output should be true")
	assert.Equal(t, "*.example.com", domain)

	domain, ok = ReplaceSubdomainWithWildcard("www.example.com:5612")
	assert.True(t, ok, "Output should be true")
	assert.Equal(t, "*.example.com:5612", domain)
}

func TestGetBaseDomain(t *testing.T) {
	domain, ok := GetParentDomain("www.example.com")
	assert.True(t, ok, "Output should be true")
	assert.Equal(t, "example.com", domain)

	domain, ok = GetParentDomain("www.example.com:5612")
	assert.True(t, ok, "Output should be true")
	assert.Equal(t, "example.com:5612", domain)
}

func TestGetTopFqdn(t *testing.T) {
	domain, ok := GetTopFqdn("www.example.com")
	assert.True(t, ok, "Output should be true")
	assert.Equal(t, "example.com", domain)

	domain, ok = GetTopFqdn("www.www.example.com")
	assert.True(t, ok, "Output should be true")
	assert.Equal(t, "example.com", domain)
}

func TestSplitHostPath(t *testing.T) {
	h, p := SplitHostPath("example.com/hello/world")
	assert.Equal(t, "example.com", h)
	assert.Equal(t, "/hello/world", p)

	h, p = SplitHostPath("example.com")
	assert.Equal(t, "example.com", h)
	assert.Equal(t, "/", p)
}

func TestSplitHostPathQuery(t *testing.T) {
	h, p, q := SplitHostPathQuery("example.com/hello/world")
	assert.Equal(t, "example.com", h)
	assert.Equal(t, "/hello/world", p)
	assert.Equal(t, "", q)

	h, p, q = SplitHostPathQuery("example.com")
	assert.Equal(t, "example.com", h)
	assert.Equal(t, "/", p)
	assert.Equal(t, "", q)

	h, p, q = SplitHostPathQuery("example.com/hello/world?a=b")
	assert.Equal(t, "example.com", h)
	assert.Equal(t, "/hello/world", p)
	assert.Equal(t, "a=b", q)

	h, p, q = SplitHostPathQuery("example.com?a=b")
	assert.Equal(t, "example.com", h)
	assert.Equal(t, "/", p)
	assert.Equal(t, "a=b", q)

	h, p, q = SplitHostPathQuery("example.com/?a=b")
	assert.Equal(t, "example.com", h)
	assert.Equal(t, "/", p)
	assert.Equal(t, "a=b", q)
}