46 lines
1.1 KiB
Go
46 lines
1.1 KiB
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.AddRobot(NewTestRobot("Robot1"))
|
|
g.AddRobot(NewTestRobot("Robot2"))
|
|
g.AddRobot(NewTestRobot("Robot3"))
|
|
return g
|
|
}
|
|
|
|
func TestGobotStart(t *testing.T) {
|
|
g := initTestGobot()
|
|
g.Start()
|
|
}
|
|
|
|
func TestGobotRobot(t *testing.T) {
|
|
g := initTestGobot()
|
|
Assert(t, g.Robot("Robot1").Name, "Robot1")
|
|
Assert(t, g.Robot("Robot4"), (*Robot)(nil))
|
|
Assert(t, g.Robot("Robot1").Device("Device4"), (Device)(nil))
|
|
Assert(t, g.Robot("Robot1").Device("Device1").Name(), "Device1")
|
|
Assert(t, g.Robot("Robot1").Devices().Len(), 3)
|
|
Assert(t, g.Robot("Robot1").Connection("Connection4"), (Connection)(nil))
|
|
Assert(t, g.Robot("Robot1").Connections().Len(), 3)
|
|
}
|
|
|
|
func TestGobotToJSON(t *testing.T) {
|
|
g := initTestGobot()
|
|
g.AddCommand("test_function", func(params map[string]interface{}) interface{} {
|
|
return nil
|
|
})
|
|
json := NewJSONGobot(g)
|
|
Assert(t, len(json.Robots), g.Robots().Len())
|
|
Assert(t, len(json.Commands), len(g.Commands()))
|
|
}
|