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