mirror of
https://github.com/1f349/violet.git
synced 2024-11-22 03:11:44 +00:00
63 lines
1.8 KiB
Go
63 lines
1.8 KiB
Go
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)
|
|
}
|