38 lines
852 B
Go
38 lines
852 B
Go
package gobot
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func initTestGobot() *Gobot {
|
|
log.SetOutput(&NullReadWriteCloser{})
|
|
g := NewGobot()
|
|
g.trap = func(c chan os.Signal) {
|
|
c <- os.Interrupt
|
|
}
|
|
g.Robots = []*Robot{
|
|
NewTestRobot("Robot 1"),
|
|
NewTestRobot("Robot 2"),
|
|
NewTestRobot("Robot 3"),
|
|
}
|
|
return g
|
|
}
|
|
|
|
func TestGobotStart(t *testing.T) {
|
|
g := initTestGobot()
|
|
g.Start()
|
|
}
|
|
|
|
func TestGobotRobot(t *testing.T) {
|
|
g := initTestGobot()
|
|
Expect(t, g.Robot("Robot 1").Name, "Robot 1")
|
|
Expect(t, g.Robot("Robot 4"), (*Robot)(nil))
|
|
Expect(t, g.Robot("Robot 1").Device("Device 4"), (*device)(nil))
|
|
Expect(t, g.Robot("Robot 1").Device("Device 1").Name, "Device 1")
|
|
Expect(t, len(g.Robot("Robot 1").Devices()), 3)
|
|
Expect(t, g.Robot("Robot 1").Connection("Connection 4"), (*connection)(nil))
|
|
Expect(t, len(g.Robot("Robot 1").Connections()), 3)
|
|
}
|