package spi import ( "errors" "image" "sync" "testing" "gobot.io/x/gobot/v2" "gobot.io/x/gobot/v2/gobottest" ) // this ensures that the implementation is based on spi.Driver, which implements the gobot.Driver // and tests all implementations, so no further tests needed here for gobot.Driver interface var _ gobot.Driver = (*SSD1306Driver)(nil) func initTestSSDDriver() *SSD1306Driver { return NewSSD1306Driver(newGpioTestAdaptor()) } func TestDriverSSDStart(t *testing.T) { d := initTestSSDDriver() gobottest.Assert(t, d.Start(), nil) } func TestDriverSSDHalt(t *testing.T) { d := initTestSSDDriver() _ = d.Start() gobottest.Assert(t, d.Halt(), nil) } func TestDriverSSDDisplay(t *testing.T) { d := initTestSSDDriver() _ = d.Start() gobottest.Assert(t, d.Display(), nil) } func TestSSD1306DriverShowImage(t *testing.T) { d := initTestSSDDriver() _ = d.Start() img := image.NewRGBA(image.Rect(0, 0, 640, 480)) gobottest.Assert(t, d.ShowImage(img), errors.New("Image must match the display width and height")) img = image.NewRGBA(image.Rect(0, 0, 128, 64)) gobottest.Assert(t, d.ShowImage(img), nil) } type gpioTestAdaptor struct { name string port string mtx sync.Mutex Connector testAdaptorDigitalWrite func() (err error) testAdaptorServoWrite func() (err error) testAdaptorPwmWrite func() (err error) testAdaptorAnalogRead func() (val int, err error) testAdaptorDigitalRead func() (val int, err error) } func (t *gpioTestAdaptor) ServoWrite(string, byte) (err error) { t.mtx.Lock() defer t.mtx.Unlock() return t.testAdaptorServoWrite() } func (t *gpioTestAdaptor) PwmWrite(string, byte) (err error) { t.mtx.Lock() defer t.mtx.Unlock() return t.testAdaptorPwmWrite() } func (t *gpioTestAdaptor) AnalogRead(string) (val int, err error) { t.mtx.Lock() defer t.mtx.Unlock() return t.testAdaptorAnalogRead() } func (t *gpioTestAdaptor) DigitalRead(string) (val int, err error) { t.mtx.Lock() defer t.mtx.Unlock() return t.testAdaptorDigitalRead() } func (t *gpioTestAdaptor) DigitalWrite(string, byte) (err error) { t.mtx.Lock() defer t.mtx.Unlock() return t.testAdaptorDigitalWrite() } func (t *gpioTestAdaptor) Connect() (err error) { return } func (t *gpioTestAdaptor) Finalize() (err error) { return } func (t *gpioTestAdaptor) Name() string { return t.name } func (t *gpioTestAdaptor) SetName(n string) { t.name = n } func (t *gpioTestAdaptor) Port() string { return t.port } func newGpioTestAdaptor() *gpioTestAdaptor { a := newSpiTestAdaptor() return &gpioTestAdaptor{ port: "/dev/null", testAdaptorDigitalWrite: func() (err error) { return nil }, testAdaptorServoWrite: func() (err error) { return nil }, testAdaptorPwmWrite: func() (err error) { return nil }, testAdaptorAnalogRead: func() (val int, err error) { return 99, nil }, testAdaptorDigitalRead: func() (val int, err error) { return 1, nil }, Connector: a, } }