def test_get_session_cookie_with_no_cookie(self): svc_obj = service.Service() cookie = mock.Mock() cookie.name = 'cookie' cookie.value = 'xyz' svc_obj.client.options.transport.cookiejar = [cookie] self.assertIsNone(svc_obj.get_http_cookie())
def test_request_handler_with_web_fault(self): managed_object = 'VirtualMachine' fault_list = ['Fault'] doc = mock.Mock() def side_effect(mo, **kwargs): self.assertEqual(managed_object, mo._type) self.assertEqual(managed_object, mo.value) fault = mock.Mock(faultstring="MyFault") fault_children = mock.Mock() fault_children.name = "name" fault_children.getText.return_value = "value" child = mock.Mock() child.get.return_value = fault_list[0] child.getChildren.return_value = [fault_children] detail = mock.Mock() detail.getChildren.return_value = [child] doc.childAtPath.return_value = detail raise suds.WebFault(fault, doc) svc_obj = service.Service() service_mock = svc_obj.client.service setattr(service_mock, 'powerOn', side_effect) ex = self.assertRaises(exceptions.VimFaultException, svc_obj.powerOn, managed_object) self.assertEqual(fault_list, ex.fault_list) self.assertEqual({'name': 'value'}, ex.details) self.assertEqual("MyFault", ex.msg) doc.childAtPath.assertCalledOnceWith('/detail')
def test_request_handler_with_web_fault(self): managed_object = 'VirtualMachine' fault_list = ['Fault'] def side_effect(mo, **kwargs): self.assertEqual(managed_object, mo._type) self.assertEqual(managed_object, mo.value) fault_string = mock.Mock() fault_string.getText.return_value = "MyFault" fault_children = mock.Mock() fault_children.name = "name" fault_children.getText.return_value = "value" child = mock.Mock() child.get.return_value = fault_list[0] child.getChildren.return_value = [fault_children] detail = mock.Mock() detail.getChildren.return_value = [child] doc = mock.Mock() doc.childAtPath = mock.Mock(side_effect=[fault_string, detail]) raise suds.WebFault(None, doc) svc_obj = service.Service() attr_name = 'powerOn' service_mock = svc_obj.client.service setattr(service_mock, attr_name, side_effect) try: svc_obj.powerOn(managed_object) except exceptions.VimFaultException as ex: self.assertEqual(fault_list, ex.fault_list) self.assertEqual({'name': 'value'}, ex.details) self.assertEqual("MyFault", ex.msg)
def test_get_session_cookie(self): svc_obj = service.Service() cookie_value = 'xyz' cookie = mock.Mock() cookie.name = 'vmware_soap_session' cookie.value = cookie_value svc_obj.client.options.transport.cookiejar = [cookie] self.assertEqual(cookie_value, svc_obj.get_http_cookie())
def test_request_handler_with_attribute_error(self): managed_object = 'VirtualMachine' svc_obj = service.Service() # no powerOn method in Service service_mock = mock.Mock(spec=service.Service) svc_obj.client.service = service_mock self.assertRaises(exceptions.VimAttributeException, svc_obj.powerOn, managed_object)
def _test_request_handler_with_exception(self, message, exception): managed_object = 'VirtualMachine' def side_effect(mo, **kwargs): self.assertEqual(managed_object, mo._type) self.assertEqual(managed_object, mo.value) raise Exception(message) svc_obj = service.Service() attr_name = 'powerOn' service_mock = svc_obj.client.service setattr(service_mock, attr_name, side_effect) self.assertRaises(exception, svc_obj.powerOn, managed_object)
def test_request_handler_with_retrieve_properties_ex_fault(self): managed_object = 'Datacenter' def side_effect(mo, **kwargs): self.assertEqual(managed_object, mo._type) self.assertEqual(managed_object, mo.value) return None svc_obj = service.Service() attr_name = 'retrievePropertiesEx' service_mock = svc_obj.client.service setattr(service_mock, attr_name, side_effect) self.assertRaises(exceptions.VimFaultException, svc_obj.retrievePropertiesEx, managed_object)
def test_request_handler_with_http_error(self): managed_object = 'VirtualMachine' def side_effect(mo, **kwargs): self.assertEqual(managed_object, mo._type) self.assertEqual(managed_object, mo.value) raise requests.HTTPError() svc_obj = service.Service() attr_name = 'powerOn' service_mock = svc_obj.client.service setattr(service_mock, attr_name, side_effect) self.assertRaises(exceptions.VimConnectionException, svc_obj.powerOn, managed_object)
def test_request_handler_with_http_cannot_send_header_error(self): managed_object = 'VirtualMachine' def side_effect(mo, **kwargs): self.assertEqual(managed_object, mo._type) self.assertEqual(managed_object, mo.value) raise httplib.CannotSendHeader() svc_obj = service.Service() attr_name = 'powerOn' service_mock = svc_obj.client.service setattr(service_mock, attr_name, side_effect) self.assertRaises(exceptions.VimSessionOverLoadException, svc_obj.powerOn, managed_object)
def test_request_handler_with_empty_web_fault_doc(self): def side_effect(mo, **kwargs): fault = mock.Mock(faultstring="MyFault") raise suds.WebFault(fault, None) svc_obj = service.Service() service_mock = svc_obj.client.service setattr(service_mock, 'powerOn', side_effect) ex = self.assertRaises(exceptions.VimFaultException, svc_obj.powerOn, 'VirtualMachine') self.assertEqual([], ex.fault_list) self.assertEqual({}, ex.details) self.assertEqual("MyFault", ex.msg)
def test_request_handler(self): managed_object = 'VirtualMachine' resp = mock.Mock() def side_effect(mo, **kwargs): self.assertEqual(managed_object, mo._type) self.assertEqual(managed_object, mo.value) return resp svc_obj = service.Service() attr_name = 'powerOn' service_mock = svc_obj.client.service setattr(service_mock, attr_name, side_effect) ret = svc_obj.powerOn(managed_object) self.assertEqual(resp, ret)
def test_shared_cache(self, mock_reader): cache1 = service.Service().client.options.cache cache2 = service.Service().client.options.cache self.assertIs(cache1, cache2)
def test_request_handler_no_value(self, mock_moref): managed_object = 'VirtualMachine' svc_obj = service.Service() ret = svc_obj.UnregisterVM(managed_object) self.assertIsNone(ret)