package nostr

import (
	"strconv"
	"testing"

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

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/?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"},
	{NormalizeURL("nostr:askjd"), "nostr://askjd"},
}

func TestNormalizeURL(t *testing.T) {
	for i, test := range urlTests {
		t.Run(strconv.Itoa(i)+" { "+test.url+" -> "+test.expected+" }", func(t *testing.T) {
			output := NormalizeURL(test.url)
			require.Equal(t, test.expected, output)
		})
	}
}