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_init(self):
instanceMeta = InstanceMetadata(id=1,
aws_id=2,
ncommands=10,
nuploads=10,
ndownloads=10,
lastmessage="message",
lastupdate="2017",
ncommandsFinished=0,
nuploadsFinished=0,
ndownloadsFinished=0)
self.assertEquals(instanceMeta.Get("Id"), 1)
self.assertEquals(instanceMeta.Get("AWS_Instance_Id"), 2)
self.assertEquals(instanceMeta.Get("CommandCount"), 10)
self.assertEquals(instanceMeta.Get("UploadCount"), 10)
self.assertEquals(instanceMeta.Get("DownloadCount"), 10)
self.assertEquals(instanceMeta.Get("LastMessage"), "message")
self.assertEquals(instanceMeta.Get("LastUpdate"), "2017")
self.assertEquals(instanceMeta.Get("NCommandsFinished"), 0)
self.assertEquals(instanceMeta.Get("NUploadsFinished"), 0)
self.assertEquals(instanceMeta.Get("NDownloadsFinished"), 0)
def test_Get_raises_error_on_bad_key(self):
instanceMeta = InstanceMetadata(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
self.assertRaises(KeyError, lambda: instanceMeta.Get("missing"))