2014-04-26 18:07:04 -07:00
|
|
|
package sphero
|
2014-04-26 03:11:51 -07:00
|
|
|
|
|
|
|
import (
|
2014-12-17 14:32:12 -08:00
|
|
|
"errors"
|
2014-12-23 07:00:23 -08:00
|
|
|
"io"
|
2014-06-13 14:26:18 -07:00
|
|
|
"testing"
|
2014-11-28 18:21:02 -08:00
|
|
|
|
2016-08-27 11:56:01 +02:00
|
|
|
"github.com/hybridgroup/gobot"
|
2016-02-22 00:21:24 -05:00
|
|
|
"github.com/hybridgroup/gobot/gobottest"
|
2014-04-26 03:11:51 -07:00
|
|
|
)
|
|
|
|
|
2016-08-27 11:56:01 +02:00
|
|
|
var _ gobot.Adaptor = (*SpheroAdaptor)(nil)
|
|
|
|
|
2014-12-17 14:32:12 -08:00
|
|
|
type nullReadWriteCloser struct{}
|
|
|
|
|
|
|
|
var testAdaptorRead = func(p []byte) (int, error) {
|
|
|
|
return len(p), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (nullReadWriteCloser) Write(p []byte) (int, error) {
|
|
|
|
return testAdaptorRead(p)
|
|
|
|
}
|
|
|
|
|
|
|
|
var testAdaptorWrite = func(b []byte) (int, error) {
|
|
|
|
return len(b), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (nullReadWriteCloser) Read(b []byte) (int, error) {
|
|
|
|
return testAdaptorWrite(b)
|
|
|
|
}
|
|
|
|
|
|
|
|
var testAdaptorClose = func() error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (nullReadWriteCloser) Close() error {
|
|
|
|
return testAdaptorClose()
|
|
|
|
}
|
|
|
|
|
2014-06-13 16:01:39 -07:00
|
|
|
func initTestSpheroAdaptor() *SpheroAdaptor {
|
|
|
|
a := NewSpheroAdaptor("bot", "/dev/null")
|
2014-12-23 07:00:23 -08:00
|
|
|
a.connect = func(string) (io.ReadWriteCloser, error) {
|
|
|
|
return &nullReadWriteCloser{}, nil
|
|
|
|
}
|
2014-06-13 16:01:39 -07:00
|
|
|
return a
|
2014-06-13 14:26:18 -07:00
|
|
|
}
|
2014-04-26 03:11:51 -07:00
|
|
|
|
2014-12-17 14:32:12 -08:00
|
|
|
func TestSpheroAdaptor(t *testing.T) {
|
|
|
|
a := initTestSpheroAdaptor()
|
2016-02-22 00:21:24 -05:00
|
|
|
gobottest.Assert(t, a.Name(), "bot")
|
|
|
|
gobottest.Assert(t, a.Port(), "/dev/null")
|
2014-12-17 14:32:12 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestSpheroAdaptorReconnect(t *testing.T) {
|
|
|
|
a := initTestSpheroAdaptor()
|
|
|
|
a.Connect()
|
2016-02-22 00:21:24 -05:00
|
|
|
gobottest.Assert(t, a.connected, true)
|
2014-12-17 14:32:12 -08:00
|
|
|
a.Reconnect()
|
2016-02-22 00:21:24 -05:00
|
|
|
gobottest.Assert(t, a.connected, true)
|
2014-12-17 14:32:12 -08:00
|
|
|
a.Disconnect()
|
2016-02-22 00:21:24 -05:00
|
|
|
gobottest.Assert(t, a.connected, false)
|
2014-12-17 14:32:12 -08:00
|
|
|
a.Reconnect()
|
2016-02-22 00:21:24 -05:00
|
|
|
gobottest.Assert(t, a.connected, true)
|
2014-12-17 14:32:12 -08:00
|
|
|
}
|
|
|
|
|
2014-06-13 16:01:39 -07:00
|
|
|
func TestSpheroAdaptorFinalize(t *testing.T) {
|
|
|
|
a := initTestSpheroAdaptor()
|
2014-12-23 07:00:23 -08:00
|
|
|
a.Connect()
|
2016-02-22 00:21:24 -05:00
|
|
|
gobottest.Assert(t, len(a.Finalize()), 0)
|
2014-12-17 14:32:12 -08:00
|
|
|
|
|
|
|
testAdaptorClose = func() error {
|
|
|
|
return errors.New("close error")
|
|
|
|
}
|
|
|
|
|
2014-12-23 07:00:23 -08:00
|
|
|
a.connected = true
|
2016-02-22 00:21:24 -05:00
|
|
|
gobottest.Assert(t, a.Finalize()[0], errors.New("close error"))
|
2014-06-13 14:26:18 -07:00
|
|
|
}
|
2014-12-17 14:32:12 -08:00
|
|
|
|
2014-06-13 16:01:39 -07:00
|
|
|
func TestSpheroAdaptorConnect(t *testing.T) {
|
|
|
|
a := initTestSpheroAdaptor()
|
2016-02-22 00:21:24 -05:00
|
|
|
gobottest.Assert(t, len(a.Connect()), 0)
|
2014-12-17 14:32:12 -08:00
|
|
|
|
2014-12-23 07:00:23 -08:00
|
|
|
a.connect = func(string) (io.ReadWriteCloser, error) {
|
|
|
|
return nil, errors.New("connect error")
|
2014-12-17 14:32:12 -08:00
|
|
|
}
|
|
|
|
|
2016-02-22 00:21:24 -05:00
|
|
|
gobottest.Assert(t, a.Connect()[0], errors.New("connect error"))
|
2014-06-13 14:26:18 -07:00
|
|
|
}
|