hybridgroup.gobot/test_helper.go

92 lines
2.0 KiB
Go

package gobot
import (
"fmt"
)
type testStruct struct {
i int
f float64
}
func (me *testStruct) Hello(name string, message string) string {
return fmt.Sprintf("Hello %v! %v", name, message)
}
type null struct{}
func (null) Write(p []byte) (int, error) {
return len(p), nil
}
type testDriver struct {
Driver
Adaptor *testAdaptor
}
func (me *testDriver) Init() bool { return true }
func (me *testDriver) Start() bool { return true }
func (me *testDriver) Halt() bool { return true }
func (me *testDriver) TestDriverCommand(params map[string]interface{}) string {
name := params["name"].(string)
return fmt.Sprintf("hello %v", name)
}
type testAdaptor struct {
Adaptor
}
func (me *testAdaptor) Finalize() bool { return true }
func (me *testAdaptor) Connect() bool { return true }
func newTestDriver(name string, adaptor *testAdaptor) *testDriver {
d := new(testDriver)
d.Name = name
d.Adaptor = adaptor
d.Commands = []string{
"TestDriverCommand",
"DriverCommand",
}
return d
}
func newTestAdaptor(name string) *testAdaptor {
a := new(testAdaptor)
a.Name = name
a.Params = map[string]interface{}{
"param1": "1",
"param2": 2,
}
return a
}
func robotTestFunction(params map[string]interface{}) string {
message := params["message"].(string)
robotname := params["robotname"].(string)
return fmt.Sprintf("hey %v, %v", robotname, message)
}
func NewTestRobot(name string) *Robot {
return newTestRobot(name)
}
func newTestRobot(name string) *Robot {
adaptor1 := newTestAdaptor("Connection 1")
adaptor2 := newTestAdaptor("Connection 2")
adaptor3 := newTestAdaptor("Connection 3")
driver1 := newTestDriver("Device 1", adaptor1)
driver2 := newTestDriver("Device 2", adaptor2)
driver3 := newTestDriver("Device 3", adaptor3)
work := func() {}
//Commands := map[string]interface{}{
// "robotTestFunction": robotTestFunction,
//}
return NewRobot(name, []Connection{adaptor1, adaptor2, adaptor3}, []Device{driver1, driver2, driver3}, work)
}
func newTestStruct() *testStruct {
s := new(testStruct)
s.i = 10
s.f = 0.2
return s
}