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)
}