Clean up tests

wavelets · Dec 31, 2013 · d6d9e89 · d6d9e89
1 parent 7da1b80
commit d6d9e89
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 33 deletions.
diff --git a/device_test.go b/device_test.go
@@ -12,10 +12,7 @@ var _ = Describe("Device", func() {
 	)
 
 	BeforeEach(func() {
-		someRobot = Robot{
-			Connections: []Connection{newTestAdaptor("Connection 1")},
-			Devices:     []Device{newTestDriver("Device 1")},
-		}
+		someRobot = newTestRobot("")
 	})
 
 	Context("when valid", func() {

diff --git a/master_test.go b/master_test.go
@@ -8,31 +8,19 @@ import (
 var _ = Describe("Master", func() {
 
 	var (
-		myMaster Master
+		myMaster *Master
 	)
 
 	BeforeEach(func() {
-		myMaster = Master{
-			Robots: []Robot{
-				Robot{
-					Name:        "Robot 1",
-					Connections: []Connection{newTestAdaptor("Connection 1")},
-					Devices:     []Device{newTestDriver("Device 1")},
-				},
-				Robot{
-					Name:        "Robot 2",
-					Connections: []Connection{newTestAdaptor("Connection 2")},
-					Devices:     []Device{newTestDriver("Device 2")},
-				},
-				Robot{
-					Name:        "Robot 3",
-					Connections: []Connection{newTestAdaptor("Connection 3")},
-					Devices:     []Device{newTestDriver("Device 3")},
-				},
-			},
+		myMaster = GobotMaster()
+		myMaster.Robots = []Robot{
+			newTestRobot("Robot 1"),
+			newTestRobot("Robot 2"),
+			newTestRobot("Robot 3"),
+		}
+		for s := range myMaster.Robots {
+			myMaster.Robots[s].startRobot()
 		}
-		myMaster.Robots[0].initDevices()
-		myMaster.Robots[0].initConnections()
 	})
 
 	Context("when valid", func() {

diff --git a/robot_test.go b/robot_test.go
@@ -12,14 +12,7 @@ var _ = Describe("Robot", func() {
 	)
 
 	BeforeEach(func() {
-		someRobot = Robot{
-			Connections: []Connection{newTestAdaptor("Connection 1"), newTestAdaptor("Connection 2"), newTestAdaptor("Connection 3")},
-			Devices:     []Device{newTestDriver("Device 1"), newTestDriver("Device 2"), newTestDriver("Device 3")},
-			Commands: map[string]interface{}{
-				"Command1": func() {},
-				"Command2": func() {},
-			},
-		}
+		someRobot = newTestRobot("")
 	})
 
 	Context("when valid", func() {

diff --git a/test_helper.go b/test_helper.go
@@ -30,3 +30,15 @@ func newTestAdaptor(name string) *testAdaptor {
 	a.Name = name
 	return a
 }
+
+func newTestRobot(name string) Robot {
+	return Robot{
+		Name:        name,
+		Connections: []Connection{newTestAdaptor("Connection 1"), newTestAdaptor("Connection 2"), newTestAdaptor("Connection 3")},
+		Devices:     []Device{newTestDriver("Device 1"), newTestDriver("Device 2"), newTestDriver("Device 3")},
+		Commands: map[string]interface{}{
+			"Command1": func() {},
+			"Command2": func() {},
+		},
+	}
+}