Issue #131

ADD:
  - database/impl/mysqlStore_test.go Some tests for channel updating methods

database/impl/mysqlStore_test.go

@@ -0,0 +1,94 @@
+package sqlStoreImpl
+
+import (
+	. "github.com/GrappigPanda/notorious/database"
+	"testing"
+	"time"
+)
+
+var MYSQLSTORE MySQLStore
+
+func TestInitMySQLStore(t *testing.T) {
+	MYSQLSTORE = InitMySQLStore()
+}
+
+func TestOpenConnection(t *testing.T) {
+	_, err := MYSQLSTORE.OpenConnection()
+	if err != nil {
+		t.Errorf("TestMySQLStore:TestOpenConnection() failed to open a connection with err %v", err)
+	}
+}
+
+func TestPeerUpdate(t *testing.T) {
+	expectedReturn := &PeerStats{
+		Downloaded: 6,
+		Uploaded:   21,
+		Ip:         "127.0.0.1",
+	}
+
+	newPeer := &PeerStats{
+		Downloaded: 1,
+		Uploaded:   1,
+		Ip:         "127.0.0.1",
+	}
+
+	MYSQLSTORE.dbPool.Save(&newPeer)
+
+	peerUpdate := PeerTrackerDelta{
+		Uploaded:   5,
+		Downloaded: 20,
+		IP:         "127.0.0.1",
+		Event:      PEERUPDATE,
+	}
+
+	MYSQLSTORE.UpdateConsumer <- peerUpdate
+	time.Sleep(1 * time.Second)
+
+	retval := &PeerStats{}
+	MYSQLSTORE.dbPool.First(&retval)
+
+	if retval.Downloaded != expectedReturn.Downloaded {
+		t.Fatalf("Expected %v, got %v",
+			expectedReturn.Downloaded,
+			retval.Downloaded)
+	}
+
+	if retval.Uploaded != expectedReturn.Uploaded {
+		t.Fatalf("Expected %v, got %v",
+			expectedReturn.Uploaded,
+			retval.Uploaded)
+	}
+
+	if retval.Ip != expectedReturn.Ip {
+		t.Fatalf("Expected %v, got %v",
+			expectedReturn.Ip,
+			retval.Ip)
+	}
+}
+
+func TestTrackerUpdate(t *testing.T) {
+	expectedReturn := &TrackerStats{
+		Downloaded: 10,
+		Uploaded:   50,
+	}
+
+	trackerUpdate := PeerTrackerDelta{
+		Uploaded:   10,
+		Downloaded: 50,
+		Event:      TRACKERUPDATE,
+	}
+
+	MYSQLSTORE.UpdateConsumer <- trackerUpdate
+	time.Sleep(1 * time.Second)
+
+	retval := &TrackerStats{}
+	MYSQLSTORE.dbPool.First(&retval)
+
+	if retval.Downloaded != expectedReturn.Downloaded {
+		t.Fatalf("Expected %v, got %v -- TestTrackerUpdate()", expectedReturn.Downloaded, retval.Downloaded)
+	}
+
+	if retval.Uploaded != expectedReturn.Uploaded {
+		t.Fatalf("Expected %v, got %v -- TestTrackerUpdate():Upload", expectedReturn.Uploaded, retval.Uploaded)
+	}
+}