class TestGetVmDetails(TestCase):
def setUp(self):
self.vmDetailsProvider = VmDetailsProvider()
def test_get_vm_details(self):
shell = AWSShell()
requests_json = encode({'items': [{'deployedAppJson': { 'name': 'something', 'vmdetails': {'uid': '514'}}}]})
shell.get_vm_details(Mock(), Mock(), requests_json)
def test_get_vm_details_name_exist(self):
shell = AWSShell()
requests_json = encode({'items': [{'deployedAppJson': { 'name': 'something', 'vmdetails': {'uid': '514'}}}]})
cancellation_context = Mock()
cancellation_context.is_cancelled = False
res = shell.get_vm_details(Mock(), cancellation_context, requests_json)
res_list = jsonpickle.decode(res)
self.assertTrue(res_list[0][u'appName'] == 'something')
def test_get_volume_when_empty(self):
instance = Mock()
instance.volumes.all = lambda: []
volume = self.vmDetailsProvider._get_volume(instance)
self.assertTrue(volume is None)
def test_get_volume(self):
volume1 = 'hi'
instance = Mock()
instance.volumes.all = lambda: [volume1]
volume = self.vmDetailsProvider._get_volume(instance)
self.assertTrue(volume==volume1)
def test_get_volume_size(self):
volume = Mock()
volume.size = 10
size = self.vmDetailsProvider._get_volume_size(volume)
self.assertTrue(size == '10 GiB')
volume = None
size = self.vmDetailsProvider._get_volume_size(volume)
self.assertTrue(size is None)
def test_get_volume_type(self):
volume = Mock()
volume.volume_type = 'hi'
volume_type = self.vmDetailsProvider._get_volume_type(volume)
self.assertTrue(volume_type == 'hi')
volume = None
volume_type = self.vmDetailsProvider._get_volume_type(volume)
self.assertTrue(volume_type is None)
def test_get_volume_id(self):
volume = Mock()
volume.volume_id = 'hi'
volume_id = self.vmDetailsProvider._get_volume_id(volume)
self.assertTrue(volume_id == 'hi')
volume = None
volume_id = self.vmDetailsProvider._get_volume_id(volume)
self.assertTrue(volume_id is None)
class TestGetVmDetails(TestCase):
def setUp(self):
self.vmDetailsProvider = VmDetailsProvider()
def test_get_vm_details(self):
shell = AWSShell()
requests_json = encode({
"items": [{
"deployedAppJson": {
"name": "something",
"vmdetails": {
"uid": "514"
},
}
}]
})
shell.get_vm_details(Mock(), Mock(), requests_json)
def test_get_vm_details_name_exist(self):
shell = AWSShell()
requests_json = encode({
"items": [{
"deployedAppJson": {
"name": "something",
"vmdetails": {
"uid": "514"
},
}
}]
})
cancellation_context = Mock()
cancellation_context.is_cancelled = False
res = shell.get_vm_details(Mock(), cancellation_context, requests_json)
res_list = jsonpickle.decode(res)
self.assertTrue(res_list[0]["appName"] == "something")
def test_get_volume_when_empty(self):
instance = Mock()
instance.volumes.all = lambda: []
volume = self.vmDetailsProvider._get_volume(instance)
self.assertTrue(volume is None)
def test_get_volume(self):
volume1 = "hi"
instance = Mock()
instance.volumes.all = lambda: [volume1]
volume = self.vmDetailsProvider._get_volume(instance)
self.assertTrue(volume == volume1)
def test_get_volume_size(self):
volume = Mock()
volume.size = 10
size = self.vmDetailsProvider._get_volume_size(volume)
self.assertTrue(size == "10 GiB")
volume = None
size = self.vmDetailsProvider._get_volume_size(volume)
self.assertTrue(size is None)
def test_get_volume_type(self):
volume = Mock()
volume.volume_type = "hi"
volume_type = self.vmDetailsProvider._get_volume_type(volume)
self.assertTrue(volume_type == "hi")
volume = None
volume_type = self.vmDetailsProvider._get_volume_type(volume)
self.assertTrue(volume_type is None)
def test_get_volume_id(self):
volume = Mock()
volume.volume_id = "hi"
volume_id = self.vmDetailsProvider._get_volume_id(volume)
self.assertTrue(volume_id == "hi")
volume = None
volume_id = self.vmDetailsProvider._get_volume_id(volume)
self.assertTrue(volume_id is None)