normalize_test.go

package nostr

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

type urlTest struct {
	url, expected string
}

var urlTests = []urlTest{
	{"", ""},
	{"wss://x.com/y", "wss://x.com/y"},
	{"wss://x.com/y/", "wss://x.com/y"},
	{"http://x.com/y", "ws://x.com/y"},
	{NormalizeURL("http://x.com/y"), "ws://x.com/y"},
	{NormalizeURL("wss://x.com"), "wss://x.com"},
	{NormalizeURL("wss://x.com/"), "wss://x.com"},
	{NormalizeURL(NormalizeURL(NormalizeURL("wss://x.com/"))), "wss://x.com"},
	{"wss://x.com", "wss://x.com"},
	{"wss://x.com/", "wss://x.com"},
	{"x.com////", "wss://x.com"},
	{"x.com/?x=23", "wss://x.com?x=23"},
	{"localhost:4036", "ws://localhost:4036"},
	{"localhost:4036/relay", "ws://localhost:4036/relay"},
	{"localhostmagnanimus.com", "wss://localhostmagnanimus.com"},
	{NormalizeURL("localhost:4036/relay"), "ws://localhost:4036/relay"},
}

func TestNormalizeURL(t *testing.T) {
	for _, test := range urlTests {
		output := NormalizeURL(test.url)
		assert.Equal(t, test.expected, output)
	}
}