Initial commit

pkg/routing/client_test.go

@@ -0,0 +1,139 @@
+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") })
+}