Feature/ring process tests (#8)

* Adding tests for ring process and setup
* Adding ring message tests
* Added test for lone ring process, updated ring as a result

Author: LostLaser

Diff for: server/ring/message/election_test.go

@@ -0,0 +1,72 @@
+package message
+
+import (
+	"testing"
+
+	"github.com/LostLaser/election/server"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestNewElection(t *testing.T) {
+	id := server.GenerateUniqueID()
+	e := NewElection(id)
+
+	assert.True(t, e.Exists(id), "Message not added to default notification list")
+}
+
+func TestAddNotified(t *testing.T) {
+	id := server.GenerateUniqueID()
+	addID := server.GenerateUniqueID()
+	e := NewElection(id)
+	e.AddNotified(addID)
+
+	found := false
+	for _, v := range e.notified {
+		if v == addID {
+			found = true
+			break
+		}
+	}
+	assert.True(t, found, "Message not added to notification list")
+}
+
+func TestNotifiedCount(t *testing.T) {
+	id := server.GenerateUniqueID()
+	addID := server.GenerateUniqueID()
+	e := NewElection(id)
+	e.AddNotified(addID)
+
+	assert.Equal(t, 2, e.NotifiedCount(), "Message not added to notification list")
+}
+
+func TestGetHighest(t *testing.T) {
+	id := server.GenerateUniqueID()
+	addID := server.GenerateUniqueID()
+	e := NewElection(id)
+	e.AddNotified(addID)
+
+	highest := ""
+	if id > addID {
+		highest = id
+	} else {
+		highest = addID
+	}
+
+	assert.Equal(t, highest, e.GetHighest(), "Highest id not determined correctly")
+}
+
+func TestExistsOk(t *testing.T) {
+	id := server.GenerateUniqueID()
+	addID := server.GenerateUniqueID()
+	e := NewElection(id)
+	e.AddNotified(addID)
+
+	assert.True(t, e.Exists(addID), "Existant ID was not found")
+}
+
+func TestExistsNotFound(t *testing.T) {
+	id := server.GenerateUniqueID()
+	e := NewElection(id)
+
+	assert.False(t, e.Exists(server.GenerateUniqueID()), "Non-existant ID was found")
+}

Diff for: server/ring/message/notify.go

@@ -6,7 +6,7 @@ type Notify struct {
 	visited []string
 }
 
-// NewNotify creates a new instance of an Notify message
+// NewNotify creates a new instance of a Notify message
 func NewNotify(id string) Notify {
 	e := Notify{}
 	e.visited = make([]string, 0)

Diff for: server/ring/message/notify_test.go

@@ -0,0 +1,40 @@
+package message
+
+import (
+	"testing"
+
+	"github.com/LostLaser/election/server"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestNewNotify(t *testing.T) {
+	id := server.GenerateUniqueID()
+	n := NewNotify(id)
+	assert.Equal(t, id, n.Master, "Base master ID not equal to input ID")
+}
+
+func TestAddVisited(t *testing.T) {
+	id := server.GenerateUniqueID()
+	addID := server.GenerateUniqueID()
+	n := NewNotify(id)
+	n.AddVisited(addID)
+
+	assert.Equal(t, addID, n.visited[0], "ID not added to notify object")
+}
+
+func TestVisitedFound(t *testing.T) {
+	id := server.GenerateUniqueID()
+	addID := server.GenerateUniqueID()
+	n := NewNotify(id)
+	n.AddVisited(addID)
+
+	assert.True(t, n.Visited(addID), "ID not found in visited list")
+}
+
+func TestVisitedNotFound(t *testing.T) {
+	id := server.GenerateUniqueID()
+	addID := server.GenerateUniqueID()
+	n := NewNotify(id)
+
+	assert.False(t, n.Visited(addID), "ID found in visited list")
+}

Diff for: server/ring/process.go

@@ -78,6 +78,8 @@ func (r *Process) electionResponder() {
 func (r *Process) startElection() {
 	p := r.getNeighbor()
 	if p == nil {
+		// only server in cluster
+		r.SetMaster(r.ID)
 		return
 	}
 	r.Emitter.Write(communication.NewEvent(r.ID, r.getNeighbor().ID, communication.StartNewElection))

Diff for: server/ring/process_test.go

@@ -0,0 +1,98 @@
+package ring
+
+import (
+	"regexp"
+	"testing"
+	"time"
+
+	"github.com/LostLaser/election/server"
+	"github.com/LostLaser/election/server/communication"
+	"github.com/stretchr/testify/assert"
+)
+
+var pollingTime = time.Millisecond * 250
+var settleTime = pollingTime + time.Millisecond*100
+
+func TestNew(t *testing.T) {
+	r := New(communication.New(10), pollingTime)
+
+	assert.False(t, r.IsUp())
+	assert.Regexp(t, regexp.MustCompile("^[a-zA-Z0-9]{8}-[a-zA-Z0-9]{4}-[a-zA-Z0-9]{4}-[a-zA-Z0-9]{4}-[a-zA-Z0-9]{12}$"), r.ID)
+	assert.Equal(t, r.HeartbeatPause, pollingTime)
+}
+
+func TestBootNoLinkedServers(t *testing.T) {
+	expectedState := server.Running
+	r := New(communication.New(10), pollingTime)
+
+	go r.Boot()
+	time.Sleep(settleTime)
+
+	assert.Equal(t, expectedState, r.State)
+	assert.Equal(t, r.ID, r.Master, "Lone server not elected as master")
+}
+
+func TestBootLinkedServers(t *testing.T) {
+	r := New(communication.New(10), pollingTime)
+	r2 := New(communication.New(10), pollingTime)
+
+	r.linkedServers = append(r.linkedServers, r2)
+	r2.linkedServers = append(r2.linkedServers, r)
+	go r.Boot()
+	go r2.Boot()
+	time.Sleep(settleTime)
+
+	masterID := ""
+	if r.ID > r2.ID {
+		masterID = r.ID
+	} else if r.ID < r2.ID {
+		masterID = r2.ID
+	} else {
+		assert.FailNow(t, "UUIDs are equivalent. Consider different ID gen solution or buy a lottery ticket")
+	}
+	assert.Equal(t, masterID, r.Master, "Server with greatest ID not elected master")
+	assert.Equal(t, masterID, r2.Master, "Server with greatest ID not elected master")
+}
+
+func TestStoppedMaster(t *testing.T) {
+	r1 := New(communication.New(10), pollingTime)
+	r2 := New(communication.New(10), pollingTime)
+	r3 := New(communication.New(10), pollingTime)
+
+	r1.linkedServers = append(r1.linkedServers, r2, r3)
+	r2.linkedServers = append(r2.linkedServers, r3, r1)
+	r3.linkedServers = append(r3.linkedServers, r1, r2)
+	servers := map[string]*Process{r1.ID: r1, r2.ID: r2, r3.ID: r3}
+	go r1.Boot()
+	go r2.Boot()
+	go r3.Boot()
+	time.Sleep(settleTime)
+
+	// set highest ID as expected master
+	expectedMaster := ""
+	for _, s := range servers {
+		if s.ID > expectedMaster {
+			expectedMaster = s.ID
+		}
+	}
+	for _, s := range servers {
+		assert.Equal(t, expectedMaster, s.Master, "Master not equal to max ID")
+	}
+
+	servers[expectedMaster].Stop()
+	time.Sleep(settleTime)
+
+	// set highest active ID as expected master
+	expectedMasterAfter := ""
+	for _, s := range servers {
+		if s.ID > expectedMasterAfter && s.IsUp() {
+			expectedMasterAfter = s.ID
+		}
+	}
+	for _, s := range servers {
+		if s.IsUp() {
+			assert.Equal(t, expectedMasterAfter, s.Master, "Master not equal to max ID")
+		}
+	}
+
+}

Diff for: server/ring/setup_test.go

@@ -0,0 +1,16 @@
+package ring
+
+import (
+	"testing"
+
+	"github.com/LostLaser/election/server/communication"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSetup(t *testing.T) {
+	e := communication.New(10)
+	count := 3
+	s := Setup{}
+	smap := s.Setup(count, e, pollingTime)
+	assert.Equal(t, count, len(smap))
+}