34 lines
675 B
Go
34 lines
675 B
Go
package bebop
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/hybridgroup/gobot/gobottest"
|
|
)
|
|
|
|
func initTestBebopAdaptor() *BebopAdaptor {
|
|
a := NewBebopAdaptor("bot")
|
|
a.connect = func(b *BebopAdaptor) (err error) {
|
|
b.drone = &testDrone{}
|
|
return nil
|
|
}
|
|
return a
|
|
}
|
|
|
|
func TestBebopAdaptorConnect(t *testing.T) {
|
|
a := initTestBebopAdaptor()
|
|
gobottest.Assert(t, len(a.Connect()), 0)
|
|
|
|
a.connect = func(a *BebopAdaptor) error {
|
|
return errors.New("connection error")
|
|
}
|
|
gobottest.Assert(t, a.Connect()[0], errors.New("connection error"))
|
|
}
|
|
|
|
func TestBebopAdaptorFinalize(t *testing.T) {
|
|
a := initTestBebopAdaptor()
|
|
a.Connect()
|
|
gobottest.Assert(t, len(a.Finalize()), 0)
|
|
}
|