package routing

import (
	"net/http"
	"testing"

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

func TestClient(t *testing.T) {
	hub := NewHub(nil)
	client := &Client{
		hub:     hub,
		send:    make(chan []byte, 256),
		Auth:    Auth{UID: "UIDABC001", Role: "admin"},
		pubSub:  []string{},
		privSub: []string{},
	}

	assert.Equal(t, "UIDABC001", client.GetAuth().UID)
	assert.Equal(t, []string{}, client.GetSubscriptions())

	client.SubscribePublic("a.x")
	assert.Equal(t, []string{"a.x"}, client.GetSubscriptions())
	assert.Equal(t, []string{"a.x"}, client.pubSub)
	assert.Equal(t, []string{}, client.privSub)

	client.SubscribePublic("a.y")
	assert.Equal(t, []string{"a.x", "a.y"}, client.GetSubscriptions())
	assert.Equal(t, []string{"a.x", "a.y"}, client.pubSub)
	assert.Equal(t, []string{}, client.privSub)

	client.UnsubscribePublic("a.y")
	assert.Equal(t, []string{"a.x"}, client.GetSubscriptions())
	assert.Equal(t, []string{"a.x"}, client.pubSub)
	assert.Equal(t, []string{}, client.privSub)

	client.SubscribePrivate("b")
	assert.Equal(t, []string{"a.x", "b"}, client.GetSubscriptions())
	assert.Equal(t, []string{"a.x"}, client.pubSub)
	assert.Equal(t, []string{"b"}, client.privSub)

	client.SubscribePrivate("c")
	assert.Equal(t, []string{"a.x", "b", "c"}, client.GetSubscriptions())
	assert.Equal(t, []string{"a.x"}, client.pubSub)
	assert.Equal(t, []string{"b", "c"}, client.privSub)

	client.UnsubscribePrivate("b")
	assert.Equal(t, []string{"a.x", "c"}, client.GetSubscriptions())
	assert.Equal(t, []string{"a.x"}, client.pubSub)
	assert.Equal(t, []string{"c"}, client.privSub)

	client.UnsubscribePrivate("c")
	assert.Equal(t, []string{"a.x"}, client.GetSubscriptions())
	assert.Equal(t, []string{"a.x"}, client.pubSub)
	assert.Equal(t, []string{}, client.privSub)

	client.UnsubscribePublic("a.x")
	assert.Equal(t, []string{}, client.GetSubscriptions())
	assert.Equal(t, []string{}, client.pubSub)
	assert.Equal(t, []string{}, client.privSub)
}

func TestParseStreamsFromURI(t *testing.T) {
	assert.Equal(t, []string{}, parseStreamsFromURI("/?"))
	assert.Equal(t, []string{}, parseStreamsFromURI(""))
	assert.Equal(t, []string{"aaa", "bbb"}, parseStreamsFromURI("/?stream=aaa&stream=bbb"))
	assert.Equal(t, []string{"aaa", "bbb"}, parseStreamsFromURI("/?stream=aaa,bbb"))
	assert.Equal(t, []string{"aaa", "bbb"}, parseStreamsFromURI("/public/?stream=aaa,bbb"))
}

func TestCheckSameOriginEmpty(t *testing.T) {
	var checkSameOriginTests = []struct {
		ok bool
		r  *http.Request
	}{
		{false, &http.Request{Host: "example.org", Header: map[string][]string{"Origin": {"https://other.org"}}}},
		{true, &http.Request{Host: "example.org", Header: map[string][]string{"Origin": {"https://example.org"}}}},
		{true, &http.Request{Host: "Example.org", Header: map[string][]string{"Origin": {"https://example.org"}}}},
	}

	for _, tt := range checkSameOriginTests {
		ok := checkSameOrigin("")(tt.r)
		if tt.ok != ok {
			t.Errorf("checkSameOrigin(%+v) returned %v, want %v", tt.r, ok, tt.ok)
		}
	}
}

func TestCheckSameOriginDomainsSetup(t *testing.T) {
	var checkSameOriginTests = []struct {
		ok bool
		r  *http.Request
	}{
		{false, &http.Request{Host: "example.org", Header: map[string][]string{"Origin": {"https://other.org"}}}},
		{true, &http.Request{Host: "whatever.org", Header: map[string][]string{"Origin": {"https://example.org"}}}},
		{true, &http.Request{Host: "whatever.org", Header: map[string][]string{"Origin": {"https://Example.org"}}}},
		{true, &http.Request{Host: "whatever.org", Header: map[string][]string{"Origin": {"https://example.com"}}}},
		{true, &http.Request{Host: "whatever.org", Header: map[string][]string{"Origin": {"https://Example.com"}}}},
		{true, &http.Request{Host: "whatever.org", Header: map[string][]string{"Origin": {}}}},
	}

	checker := checkSameOrigin("example.org,example.com")
	for _, tt := range checkSameOriginTests {
		ok := checker(tt.r)
		if tt.ok != ok {
			t.Errorf("checkSameOrigin(%+v) returned %v, want %v", tt.r, ok, tt.ok)
		}
	}

	checker = checkSameOrigin("example.org, example.com")
	for _, tt := range checkSameOriginTests {
		ok := checker(tt.r)
		if tt.ok != ok {
			t.Errorf("checkSameOrigin(%+v) returned %v, want %v", tt.r, ok, tt.ok)
		}
	}

	checker = checkSameOrigin("https://example.org,https://example.com")
	for _, tt := range checkSameOriginTests {
		ok := checker(tt.r)
		if tt.ok != ok {
			t.Errorf("checkSameOrigin(%+v) returned %v, want %v", tt.r, ok, tt.ok)
		}
	}

	checker = checkSameOrigin("https://example.org, https://example.com")
	for _, tt := range checkSameOriginTests {
		ok := checker(tt.r)
		if tt.ok != ok {
			t.Errorf("checkSameOrigin(%+v) returned %v, want %v", tt.r, ok, tt.ok)
		}
	}
}

func TestCheckSameOriginBadConfiguration(t *testing.T) {
	assert.Panics(t, func() { checkSameOrigin("https://ex ample.org") })
	assert.Panics(t, func() { checkSameOrigin("https://ex:ample.org") })
}