128 lines
3.4 KiB
Go
128 lines
3.4 KiB
Go
package digispark
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/hybridgroup/gobot"
|
|
)
|
|
|
|
type mock struct {
|
|
locationA uint8
|
|
locationB uint8
|
|
pwmChannelA uint8
|
|
pwmChannelB uint8
|
|
pwmPrescalerValue uint
|
|
pin uint8
|
|
mode uint8
|
|
state uint8
|
|
}
|
|
|
|
func (l *mock) digitalWrite(pin uint8, state uint8) error {
|
|
l.pin = pin
|
|
l.state = state
|
|
return l.error()
|
|
}
|
|
func (l *mock) pinMode(pin uint8, mode uint8) error {
|
|
l.pin = pin
|
|
l.mode = mode
|
|
return l.error()
|
|
}
|
|
|
|
var pwmInitErrorFunc = func() error { return nil }
|
|
|
|
func (l *mock) pwmInit() error { return pwmInitErrorFunc() }
|
|
func (l *mock) pwmStop() error { return l.error() }
|
|
func (l *mock) pwmUpdateCompare(channelA uint8, channelB uint8) error {
|
|
l.pwmChannelA = channelA
|
|
l.pwmChannelB = channelB
|
|
return l.error()
|
|
}
|
|
func (l *mock) pwmUpdatePrescaler(value uint) error {
|
|
l.pwmPrescalerValue = value
|
|
return l.error()
|
|
}
|
|
func (l *mock) servoInit() error { return l.error() }
|
|
func (l *mock) servoUpdateLocation(locationA uint8, locationB uint8) error {
|
|
l.locationA = locationA
|
|
l.locationB = locationB
|
|
return l.error()
|
|
}
|
|
|
|
var errorFunc = func() error { return nil }
|
|
|
|
func (l *mock) error() error { return errorFunc() }
|
|
|
|
func initTestDigisparkAdaptor() *DigisparkAdaptor {
|
|
a := NewDigisparkAdaptor("bot")
|
|
a.connect = func(a *DigisparkAdaptor) (err error) { return nil }
|
|
a.littleWire = new(mock)
|
|
errorFunc = func() error { return nil }
|
|
pwmInitErrorFunc = func() error { return nil }
|
|
return a
|
|
}
|
|
|
|
func TestDigisparkAdaptor(t *testing.T) {
|
|
a := NewDigisparkAdaptor("bot")
|
|
gobot.Assert(t, a.Name(), "bot")
|
|
}
|
|
|
|
func TestDigisparkAdaptorConnect(t *testing.T) {
|
|
a := NewDigisparkAdaptor("bot")
|
|
gobot.Assert(t, a.Connect()[0], ErrConnection)
|
|
|
|
a = initTestDigisparkAdaptor()
|
|
gobot.Assert(t, len(a.Connect()), 0)
|
|
}
|
|
|
|
func TestDigisparkAdaptorFinalize(t *testing.T) {
|
|
a := initTestDigisparkAdaptor()
|
|
gobot.Assert(t, len(a.Finalize()), 0)
|
|
}
|
|
|
|
func TestDigisparkAdaptorDigitalWrite(t *testing.T) {
|
|
a := initTestDigisparkAdaptor()
|
|
err := a.DigitalWrite("0", uint8(1))
|
|
gobot.Assert(t, err, nil)
|
|
gobot.Assert(t, a.littleWire.(*mock).pin, uint8(0))
|
|
gobot.Assert(t, a.littleWire.(*mock).state, uint8(1))
|
|
|
|
err = a.DigitalWrite("?", uint8(1))
|
|
gobot.Refute(t, err, nil)
|
|
|
|
errorFunc = func() error { return errors.New("pin mode error") }
|
|
err = a.DigitalWrite("0", uint8(1))
|
|
gobot.Assert(t, err, errors.New("pin mode error"))
|
|
}
|
|
|
|
func TestDigisparkAdaptorServoWrite(t *testing.T) {
|
|
a := initTestDigisparkAdaptor()
|
|
err := a.ServoWrite("2", uint8(80))
|
|
gobot.Assert(t, err, nil)
|
|
gobot.Assert(t, a.littleWire.(*mock).locationA, uint8(80))
|
|
gobot.Assert(t, a.littleWire.(*mock).locationB, uint8(80))
|
|
|
|
a = initTestDigisparkAdaptor()
|
|
errorFunc = func() error { return errors.New("servo error") }
|
|
err = a.ServoWrite("2", uint8(80))
|
|
gobot.Assert(t, err, errors.New("servo error"))
|
|
}
|
|
|
|
func TestDigisparkAdaptorPwmWrite(t *testing.T) {
|
|
a := initTestDigisparkAdaptor()
|
|
err := a.PwmWrite("1", uint8(100))
|
|
gobot.Assert(t, err, nil)
|
|
gobot.Assert(t, a.littleWire.(*mock).pwmChannelA, uint8(100))
|
|
gobot.Assert(t, a.littleWire.(*mock).pwmChannelB, uint8(100))
|
|
|
|
a = initTestDigisparkAdaptor()
|
|
pwmInitErrorFunc = func() error { return errors.New("pwminit error") }
|
|
err = a.PwmWrite("1", uint8(100))
|
|
gobot.Assert(t, err, errors.New("pwminit error"))
|
|
|
|
a = initTestDigisparkAdaptor()
|
|
errorFunc = func() error { return errors.New("pwm error") }
|
|
err = a.PwmWrite("1", uint8(100))
|
|
gobot.Assert(t, err, errors.New("pwm error"))
|
|
}
|