hybridgroup.gobot/drivers/gpio/relay_driver_test.go

package gpio

import (
	"strings"
	"testing"

	"gobot.io/x/gobot"
	"gobot.io/x/gobot/gobottest"
)

var _ gobot.Driver = (*RelayDriver)(nil)

// Helper to return low/high value for testing
func (l *RelayDriver) High() bool { return l.high }

func initTestRelayDriver() *RelayDriver {
	a := newGpioTestAdaptor()
	a.testAdaptorDigitalWrite = func(string, byte) (err error) {
		return nil
	}
	a.testAdaptorPwmWrite = func(string, byte) (err error) {
		return nil
	}
	return NewRelayDriver(a, "1")
}

func TestRelayDriverDefaultName(t *testing.T) {
	g := initTestRelayDriver()
	gobottest.Refute(t, g.Connection(), nil)
	gobottest.Assert(t, strings.HasPrefix(g.Name(), "Relay"), true)
}

func TestRelayDriverSetName(t *testing.T) {
	g := initTestRelayDriver()
	g.SetName("mybot")
	gobottest.Assert(t, g.Name(), "mybot")
}

func TestRelayDriverStart(t *testing.T) {
	d := initTestRelayDriver()
	gobottest.Assert(t, d.Start(), nil)
}

func TestRelayDriverHalt(t *testing.T) {
	d := initTestRelayDriver()
	gobottest.Assert(t, d.Halt(), nil)
}

func TestRelayDriverToggle(t *testing.T) {
	d := initTestRelayDriver()
	d.Off()
	d.Toggle()
	gobottest.Assert(t, d.State(), true)
	d.Toggle()
	gobottest.Assert(t, d.State(), false)
}

func TestRelayDriverToggleInverted(t *testing.T) {
	d := initTestRelayDriver()
	d.Inverted = true
	d.Off()
	gobottest.Assert(t, d.State(), false)
	d.Toggle()
	gobottest.Assert(t, d.State(), true)
	d.Toggle()
	gobottest.Assert(t, d.State(), false)
}

func TestRelayDriverCommands(t *testing.T) {
	d := initTestRelayDriver()
	gobottest.Assert(t, d.Command("Off")(nil), nil)
	gobottest.Assert(t, d.State(), false)

	gobottest.Assert(t, d.Command("On")(nil), nil)
	gobottest.Assert(t, d.State(), true)

	gobottest.Assert(t, d.Command("Toggle")(nil), nil)
	gobottest.Assert(t, d.State(), false)
}

func TestRelayDriverCommandsInverted(t *testing.T) {
	d := initTestRelayDriver()
	d.Inverted = true

	gobottest.Assert(t, d.Command("Off")(nil), nil)
	gobottest.Assert(t, d.High(), true)
	gobottest.Assert(t, d.State(), false)

	gobottest.Assert(t, d.Command("On")(nil), nil)
	gobottest.Assert(t, d.High(), false)
	gobottest.Assert(t, d.State(), true)

	gobottest.Assert(t, d.Command("Toggle")(nil), nil)
	gobottest.Assert(t, d.High(), true)
	gobottest.Assert(t, d.State(), false)
}