From 90d53f3e4c05f7f71e071b8f8def4abae33537fd Mon Sep 17 00:00:00 2001 From: Prashant Gupta Date: Tue, 30 Apr 2019 19:45:09 -0700 Subject: [PATCH] added machine wake test --- pkg/mousemover/mouseMover_test.go | 21 +++++++++++++++++---- 1 file changed, 17 insertions(+), 4 deletions(-) diff --git a/pkg/mousemover/mouseMover_test.go b/pkg/mousemover/mouseMover_test.go index 6606ca1..1e4112f 100644 --- a/pkg/mousemover/mouseMover_test.go +++ b/pkg/mousemover/mouseMover_test.go @@ -57,7 +57,7 @@ func (suite *TestMover) TestSingleton() { assert.True(t, mouseMover2.state.isRunning(), "instance should have started") } -func (suite *TestMover) TestSystemSleep() { +func (suite *TestMover) TestSystemSleepAndWake() { t := suite.T() mouseMover := GetInstance() @@ -69,15 +69,28 @@ func (suite *TestMover) TestSystemSleep() { //fake a machine-sleep activity machineSleepActivityMap := make(map[activity.Type][]time.Time) - var timeArray []time.Time - timeArray = append(timeArray, time.Now()) - machineSleepActivityMap[activity.MachineSleep] = timeArray + var sleepTimeArray []time.Time + sleepTimeArray = append(sleepTimeArray, time.Now()) + machineSleepActivityMap[activity.MachineSleep] = sleepTimeArray heartbeatCh <- &tracker.Heartbeat{ WasAnyActivity: true, ActivityMap: machineSleepActivityMap, } time.Sleep(time.Millisecond * 500) //wait for it to be registered assert.True(t, mouseMover.state.isSystemSleeping(), "machine should be sleeping now") + + //fake a machine-wake activity + machineWakeActivityMap := make(map[activity.Type][]time.Time) + var wakeTimeArray []time.Time + wakeTimeArray = append(wakeTimeArray, time.Now()) + machineWakeActivityMap[activity.MachineWake] = wakeTimeArray + heartbeatCh <- &tracker.Heartbeat{ + WasAnyActivity: true, + ActivityMap: machineWakeActivityMap, + } + + time.Sleep(time.Millisecond * 500) //wait for it to be registered + assert.False(t, mouseMover.state.isSystemSleeping(), "machine should be awake now") } func (suite *TestMover) TestMouseMoveSuccess() {