added machine wake test

This commit is contained in:
Prashant Gupta 2019-04-30 19:45:09 -07:00
parent 642e2f3924
commit 90d53f3e4c
1 changed files with 17 additions and 4 deletions

View File

@ -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() {