Adds socket testing

access_key.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"encoding/json"
+
 	"github.com/garyburd/redigo/redis"
 )
 

main.go

@@ -1,16 +1,18 @@
 package main
 
 import (
-	"github.com/garyburd/redigo/redis"
-	"github.com/pote/redisurl"
-	"golang.org/x/net/websocket"
 	"log"
-	"lua"
 	"net/http"
 	"os"
 	"runtime"
 	"strconv"
 	"strings"
+
+	"github.com/garyburd/redigo/redis"
+	"github.com/pote/redisurl"
+	"golang.org/x/net/websocket"
+
+	"lua"
 )
 
 var RedisPool *redis.Pool = SetupRedis()

main_test.go

@@ -1,12 +1,13 @@
 package main
 
 import (
-	"code.google.com/p/go-uuid/uuid"
 	"fmt"
-	"golang.org/x/net/websocket"
 	"os"
 	"testing"
 	"time"
+
+	"golang.org/x/net/websocket"
+	"code.google.com/p/go-uuid/uuid"
 )
 
 func createTestAccessKey() (*AccessKey, error) {

socket.go

@@ -1,11 +1,12 @@
 package main
 
 import (
-	"code.google.com/p/go-uuid/uuid"
 	"encoding/json"
+	"log"
+
+	"code.google.com/p/go-uuid/uuid"
 	"github.com/garyburd/redigo/redis"
 	"golang.org/x/net/websocket"
-	"log"
 )
 
 type Socket struct {

socket_test.go

@@ -0,0 +1,87 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"testing"
+	"time"
+
+	"golang.org/x/net/websocket"
+)
+
+func createMatchingAccessKeys() (*AccessKey, *AccessKey) {
+	ak1 := &AccessKey{
+		Read: []string{"test-channel"},
+		Write: []string{"test-channel"},
+		Token: "blah",
+	}
+
+	ak1.Save()
+
+	ak2 := &AccessKey{
+		Read: []string{"test-channel"},
+		Write: []string{"test-channel"},
+		Token: "bleh",
+	}
+
+	ak2.Save()
+
+	return ak1, ak2
+}
+
+func TestBasicPubSub (t *testing.T) {
+	ak1, ak2 := createMatchingAccessKeys()
+	go main(); time.Sleep(time.Second) // Give it a second, will you?
+
+	url1 := fmt.Sprintf("ws://localhost:%v/%v", os.Getenv("PORT"), ak1.Token)
+	url2 := fmt.Sprintf("ws://localhost:%v/%v", os.Getenv("PORT"), ak2.Token)
+
+	ws1, err := websocket.Dial(url1, "", "http://localhost"); if err != nil {
+		t.Fatal(err)
+	}
+
+	ws2, err := websocket.Dial(url2, "", "http://localhost"); if err != nil {
+		t.Fatal(err)
+	}
+
+	channel := make(chan Message, 1)
+
+	go func() {
+		receivedMsg := Message{}
+		err := websocket.JSON.Receive(ws2, &receivedMsg); if err != nil {
+			t.Fatal(err)
+		}
+
+		channel <- receivedMsg
+	}()
+
+
+	originalMessage := &Message{
+		Channel: "test-channel",
+		Data: "hey there!",
+		Event: "message",
+	}
+
+	go func() {
+		time.Sleep(time.Second * 1)
+		websocket.JSON.Send(ws1, &originalMessage)
+	}()
+
+	select {
+	case receivedMessage := <- channel:
+		if originalMessage.Channel != receivedMessage.Channel {
+			t.Error("Received message channel is incorrect")
+		}
+
+		if originalMessage.Data != receivedMessage.Data {
+			t.Error("Received message Data is incorrect")
+		}
+
+		if originalMessage.Event != receivedMessage.Event {
+			t.Error("Received message Event is incorrect")
+		}
+
+	case <- time.After(time.Second * 3):
+		t.Error("timout reached, no message received")
+	}
+}