119 lines
3.2 KiB
Go
119 lines
3.2 KiB
Go
package i2c
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/hybridgroup/gobot"
|
|
"github.com/hybridgroup/gobot/gobottest"
|
|
)
|
|
|
|
var _ gobot.Driver = (*AdafruitMotorHatDriver)(nil)
|
|
|
|
// --------- HELPERS
|
|
func initTestAdafruitMotorHatDriver() (driver *AdafruitMotorHatDriver) {
|
|
driver, _ = initTestAdafruitMotorHatDriverWithStubbedAdaptor()
|
|
return
|
|
}
|
|
|
|
func initTestAdafruitMotorHatDriverWithStubbedAdaptor() (*AdafruitMotorHatDriver, *i2cTestAdaptor) {
|
|
adaptor := newI2cTestAdaptor()
|
|
return NewAdafruitMotorHatDriver(adaptor), adaptor
|
|
}
|
|
|
|
// --------- TESTS
|
|
func TestNewAdafruitMotorHatDriver(t *testing.T) {
|
|
var adafruit interface{} = NewAdafruitMotorHatDriver(newI2cTestAdaptor())
|
|
_, ok := adafruit.(*AdafruitMotorHatDriver)
|
|
if !ok {
|
|
t.Errorf("AdafruitMotorHatDriver() should have returned a *AdafruitMotorHatDriver")
|
|
}
|
|
|
|
a := NewAdafruitMotorHatDriver(newI2cTestAdaptor())
|
|
gobottest.Assert(t, a.Name(), "AdafruitMotorHat")
|
|
}
|
|
|
|
// Methods
|
|
func TestAdafruitMotorHatDriverStart(t *testing.T) {
|
|
ada, adaptor := initTestAdafruitMotorHatDriverWithStubbedAdaptor()
|
|
|
|
gobottest.Assert(t, ada.Start(), nil)
|
|
|
|
adaptor.i2cStartImpl = func() error {
|
|
return errors.New("start error")
|
|
}
|
|
err := ada.Start()
|
|
gobottest.Assert(t, err, errors.New("start error"))
|
|
|
|
}
|
|
|
|
func TestAdafruitMotorHatDriverHalt(t *testing.T) {
|
|
ada, _ := initTestAdafruitMotorHatDriverWithStubbedAdaptor()
|
|
|
|
gobottest.Assert(t, ada.Halt(), nil)
|
|
}
|
|
func TestSetHatAddresses(t *testing.T) {
|
|
ada, _ := initTestAdafruitMotorHatDriverWithStubbedAdaptor()
|
|
|
|
motorHatAddr := 0x61
|
|
servoHatAddr := 0x41
|
|
gobottest.Assert(t, ada.SetMotorHatAddress(motorHatAddr), nil)
|
|
gobottest.Assert(t, ada.SetServoHatAddress(servoHatAddr), nil)
|
|
}
|
|
|
|
func TestAdafruitMotorHatDriverSetServoMotorFreq(t *testing.T) {
|
|
ada, _ := initTestAdafruitMotorHatDriverWithStubbedAdaptor()
|
|
|
|
freq := 60.0
|
|
err := ada.SetServoMotorFreq(freq)
|
|
gobottest.Assert(t, err, nil)
|
|
}
|
|
|
|
func TestAdafruitMotorHatDriverSetServoMotorPulse(t *testing.T) {
|
|
ada, _ := initTestAdafruitMotorHatDriverWithStubbedAdaptor()
|
|
|
|
var channel byte = 7
|
|
var on int32 = 1234
|
|
var off int32 = 4321
|
|
err := ada.SetServoMotorPulse(channel, on, off)
|
|
gobottest.Assert(t, err, nil)
|
|
}
|
|
|
|
func TestAdafruitMotorHatDriverSetDCMotorSpeed(t *testing.T) {
|
|
ada, _ := initTestAdafruitMotorHatDriverWithStubbedAdaptor()
|
|
|
|
dcMotor := 1
|
|
var speed int32 = 255
|
|
err := ada.SetDCMotorSpeed(dcMotor, speed)
|
|
gobottest.Assert(t, err, nil)
|
|
}
|
|
|
|
func TestAdafruitMotorHatDriverRunDCMotor(t *testing.T) {
|
|
ada, _ := initTestAdafruitMotorHatDriverWithStubbedAdaptor()
|
|
|
|
dcMotor := 1
|
|
// NOTE: not using the direction constant to prevent importing
|
|
// the i2c package
|
|
err := ada.RunDCMotor(dcMotor, 1)
|
|
gobottest.Assert(t, err, nil)
|
|
}
|
|
|
|
func TestAdafruitMotorHatDriverSetStepperMotorSpeed(t *testing.T) {
|
|
ada, _ := initTestAdafruitMotorHatDriverWithStubbedAdaptor()
|
|
|
|
stepperMotor := 1
|
|
rpm := 30
|
|
gobottest.Assert(t, ada.SetStepperMotorSpeed(stepperMotor, rpm), nil)
|
|
}
|
|
|
|
func TestAdafruitMotorHatDriverStepperStep(t *testing.T) {
|
|
ada, _ := initTestAdafruitMotorHatDriverWithStubbedAdaptor()
|
|
|
|
// NOTE: not using the direction and style constants to prevent importing
|
|
// the i2c package
|
|
stepperMotor := 0
|
|
steps := 50
|
|
err := ada.Step(stepperMotor, steps, 1, 3)
|
|
gobottest.Assert(t, err, nil)
|
|
}
|