joyful/internal/virtualdevice/eventbuffer_test.go

88 lines
1.9 KiB
Go

package virtualdevice
import (
"errors"
"testing"
"github.com/holoplot/go-evdev"
"github.com/stretchr/testify/mock"
"github.com/stretchr/testify/suite"
)
type EventBufferTests struct {
suite.Suite
device *VirtualDeviceMock
writeOneCall *mock.Call
}
type VirtualDeviceMock struct {
mock.Mock
}
func (m *VirtualDeviceMock) WriteOne(event *evdev.InputEvent) error {
args := m.Called()
return args.Error(0)
}
func TestRunnerEventBufferTests(t *testing.T) {
suite.Run(t, new(EventBufferTests))
}
func (t *EventBufferTests) SetupTest() {
t.device = new(VirtualDeviceMock)
}
func (t *EventBufferTests) SetupSubTest() {
t.device = new(VirtualDeviceMock)
t.writeOneCall = t.device.On("WriteOne").Return(nil)
}
func (t *EventBufferTests) TearDownSubTest() {
t.writeOneCall.Unset()
}
func (t *EventBufferTests) TestNewEventBuffer() {
buffer := NewEventBuffer(t.device)
t.Equal(t.device, buffer.Device)
t.Len(buffer.events, 0)
}
func (t *EventBufferTests) TestEventBufferAddEvent() {
buffer := NewEventBuffer(t.device)
buffer.AddEvent(&evdev.InputEvent{})
buffer.AddEvent(&evdev.InputEvent{})
buffer.AddEvent(&evdev.InputEvent{})
t.Len(buffer.events, 3)
}
func (t *EventBufferTests) TestEventBufferSendEvents() {
t.Run("3 Events", func() {
buffer := NewEventBuffer(t.device)
buffer.AddEvent(&evdev.InputEvent{})
buffer.AddEvent(&evdev.InputEvent{})
buffer.AddEvent(&evdev.InputEvent{})
errs := buffer.SendEvents()
t.Len(errs, 0)
t.device.AssertNumberOfCalls(t.T(), "WriteOne", 4)
})
t.Run("No Events", func() {
buffer := NewEventBuffer(t.device)
errs := buffer.SendEvents()
t.Len(errs, 0)
t.device.AssertNumberOfCalls(t.T(), "WriteOne", 0)
})
t.Run("Bad Event", func() {
t.writeOneCall.Unset()
t.writeOneCall = t.device.On("WriteOne").Return(errors.New("Fail"))
buffer := NewEventBuffer(t.device)
buffer.AddEvent(&evdev.InputEvent{})
errs := buffer.SendEvents()
t.Len(errs, 2)
})
}