def test_IncrementUploadsFinished(self): instanceMeta = InstanceMetadata(id=1, aws_id=2, ncommands=10, nuploads=10, ndownloads=10, lastmessage="message", lastupdate="empty", ncommandsFinished=0, nuploadsFinished=0, ndownloadsFinished=0) self.assertEqual(instanceMeta.Get("NUploadsFinished"), 0) self.assertEqual(instanceMeta.Get("LastUpdate"), "empty") instanceMeta.IncrementUploadsFinished() self.assertEqual(instanceMeta.Get("NUploadsFinished"), 1) newTime = instanceMeta.Get("LastUpdate") self.assertTrue(TimeHelper.GetTimeElapsed(newTime) < 1.0)
def test_AllTasksFinished(self): instanceMeta = InstanceMetadata(id=1, aws_id=2, ncommands=10, nuploads=10, ndownloads=10, lastmessage="message", lastupdate=" ", ncommandsFinished=0, nuploadsFinished=0, ndownloadsFinished=0) for x in range(0, 10): self.assertFalse(instanceMeta.AllTasksFinished()) instanceMeta.IncrementUploadsFinished() instanceMeta.IncrementCommandFinished() instanceMeta.IncrementDownloadFinished() self.assertTrue(instanceMeta.AllTasksFinished())