def _test_vnc_read_with_job_params(self): fake_vnc_lib = flexmock() flexmock(JobVncApi).should_receive('vnc_init').and_return(fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router = PhysicalRouter(name='test-tenant') fake_router_obj = {"physical-router": {"name": "Router-1"}} flexmock(fake_vnc_lib).should_receive( 'physical_router_read').and_return(fake_router) flexmock(fake_vnc_lib).should_receive('obj_to_dict').and_return( fake_router_obj) set_module_args( dict( object_type='physical_router', object_op='read', object_dict={"uuid": "1234"}, job_ctx={ "auth_token": "1234", "job_template_fqname": ["default-global-system-config", "device_import_template"], "job_execution_id": "123", "config_args": "546", "job_input": "" })) result = self.execute_module() self.assertEqual(result.get('failed'), None)
def test_fileutility01(self): fake_image_url = "/v1/sample.txt" flexmock(swiftclient.utils).should_receive('generate_temp_url').and_return(fake_image_url) set_module_args(self.args_dict) result = self.execute_module() self.assertTrue(result["url"]) self.assertEqual(result["url"], fake_image_url)
def _test_vnc_bulk_create(self): object_list = [{ "parent_type": "global-system-config", "fq_name": ["default-global-system-config", "mx-240"], "physical_router_management_ip": "172.10.68.1", "physical_router_vendor_name": "Juniper", "physical_router_product_name": "MX", "physical_router_device_family": "junos" }] fake_vnc_lib = flexmock() flexmock(JobVncApi).should_receive('vnc_init').and_return(fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router_uuid = "1ef6cf9d-c2e2-4004-810a-43d471c94dc5" fake_router = PhysicalRouter(name='mx-240') flexmock(fake_vnc_lib).should_receive( 'physical_router_read').and_return(fake_router) flexmock(fake_vnc_lib).should_receive( 'physical_router_create').and_return(fake_router_uuid) set_module_args( dict(enable_job_ctx=False, object_type='physical_router', object_op='bulk_create', object_list=object_list, job_ctx={"auth_token": "1234"})) result = self.execute_module() self.assertEqual(result.get('failed'), None)
def test_auth_token_not_present(self): set_module_args( dict(object_type='physical_router', object_op='get', object_dict={"uuid": "1234"})) result = self.execute_module(failed=True) self.assertEqual(result['failed'], True)
def test_auth_token_not_present(self): set_module_args( dict(enable_job_ctx=False, object_type='physical_router', object_op='get', object_dict={"uuid": "1234"}, job_ctx={})) result = self.execute_module(failed=True) self.assertEqual(result['failed'], True)
def test_object_type_not_present(self): set_module_args( dict(enable_job_ctx=False, object_op='get', object_dict={"uuid": "1234"}, job_ctx={"auth_token": "1234"})) result = self.execute_module(failed=True) self.assertEqual(result['failed'], True)
def _test_job_ctx_not_present(self): set_module_args( dict(enable_job_ctx=False, object_type='physical_router', object_op='get', object_dict={"uuid": "1234"})) result = self.execute_module(failed=True) self.assertEqual(result['failed'], True)
def _test_invalid_vnc_class(self): set_module_args( dict(object_type='lldp', object_op='get', object_dict={"uuid": "1234"}, auth_token="1234")) result = self.execute_module(failed=True) self.assertEqual(result['failed'], True)
def test_fileutility06(self): for e in ["tenant_name","auth_version","chosen_temp_url_key","connection_retry_count"]: self.args_dict.pop(e, None) set_module_args(self.args_dict) fake_image_url = "/v1/sample.txt" flexmock(swiftclient.utils).should_receive('generate_temp_url').and_return(fake_image_url) result = self.execute_module() self.assertTrue(result["url"]) self.assertEqual(result["url"], fake_image_url)
def _test_invalid_vnc_op(self): set_module_args( dict(enable_job_ctx=False, object_type='physical_router', object_op='get', object_dict={"uuid": "1234"}, job_ctx={"auth_token": "1234"})) result = self.execute_module(failed=True) self.assertEqual(result['failed'], True)
def test_vnc_read(self): fake_vnc_lib = flexmock() flexmock(VncApi, __new__=fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router = PhysicalRouter(name='test-tenant') fake_router_obj = {"physical-router": {"name": "Router-1"}} flexmock(fake_vnc_lib).should_receive( 'physical_router_read').and_return(fake_router) flexmock(fake_vnc_lib).should_receive( 'obj_to_dict').and_return(fake_router_obj) set_module_args( dict(enable_job_ctx=False, object_type='physical_router', object_op='read', object_dict={"uuid": "1234"}, job_ctx={"auth_token": "1234"})) result = self.execute_module() self.assertEqual(result.get('failed'), None)
def test_vnc_read(self): fake_vnc_lib = flexmock() flexmock(VncApi, __new__=fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router = PhysicalRouter(name='test-tenant') fake_router_obj = {"physical-router": {"name": "Router-1"}} flexmock(fake_vnc_lib).should_receive( 'physical_router_read').and_return(fake_router) flexmock(fake_vnc_lib).should_receive('obj_to_dict').and_return( fake_router_obj) set_module_args( dict(object_type='physical_router', object_op='read', object_dict={"uuid": "1234"}, auth_token="1234")) result = self.execute_module() self.assertEqual(result['failed'], False)
def test_fileutility06(self): self.args_dict = dict(authtoken="4242", authurl="auth_url", user="******", key="contrail", temp_url_key="temp_url_key1", temp_url_key_2="temp_url_key2", container_name="container", filename="sample.txt", expirytime=3600) set_module_args(self.args_dict) fake_image_url = "/v1/sample.txt" flexmock(swiftclient.utils).should_receive( 'generate_temp_url').and_return(fake_image_url) result = self.execute_module() self.assertTrue(result["url"]) self.assertEqual(result["url"], fake_image_url)
def _test_vnc_read(self): fake_vnc_lib = flexmock() flexmock(JobVncApi).should_receive('vnc_init').and_return(fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router = PhysicalRouter(name='test-tenant') fake_router_obj = {"physical-router": {"name": "Router-1"}} flexmock(fake_vnc_lib).should_receive( 'physical_router_read').and_return(fake_router) flexmock(fake_vnc_lib).should_receive('obj_to_dict').and_return( fake_router_obj) set_module_args( dict(enable_job_ctx=False, object_type='physical_router', object_op='read', object_dict={"uuid": "1234"}, job_ctx={"auth_token": "1234"})) result = self.execute_module() self.assertEqual(result.get('failed'), None)
def test_vnc_create(self): object_dict = {"parent_type": "global-system-config", "fq_name": ["default-global-system-config", "mx-240"], "physical_router_management_ip": "172.10.68.1", "physical_router_vendor_name": "Juni", "physical_router_product_name": "MX", "physical_router_device_family": "junos"} fake_vnc_lib = flexmock() flexmock(VncApi, __new__=fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router_uuid = "1ef6cf9d-c2e2-4004-810a-43d471c94dc5" flexmock(fake_vnc_lib).should_receive( 'physical_router_create').and_return(fake_router_uuid) set_module_args( dict(enable_job_ctx=False, object_type='physical_router', object_op='create', object_dict=object_dict, job_ctx={"auth_token": "1234"})) result = self.execute_module() self.assertEqual(result.get('failed'), None)
def test_vnc_read_with_job_params(self): fake_vnc_lib = flexmock() flexmock(VncApi, __new__=fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router = PhysicalRouter(name='test-tenant') fake_router_obj = {"physical-router": {"name": "Router-1"}} flexmock(fake_vnc_lib).should_receive( 'physical_router_read').and_return(fake_router) flexmock(fake_vnc_lib).should_receive( 'obj_to_dict').and_return(fake_router_obj) set_module_args( dict(object_type='physical_router', object_op='read', object_dict={"uuid": "1234"}, job_ctx={"auth_token": "1234", "job_template_fqname": [ "default-global-system-config", "device_import_template"], "job_execution_id": "123", "config_args": "546", "job_input": ""})) result = self.execute_module() self.assertEqual(result.get('failed'), None)
def test_vnc_create(self): object_dict = { "parent_type": "global-system-config", "fq_name": ["default-global-system-config", "mx-240"], "physical_router_management_ip": "172.10.68.1", "physical_router_vendor_name": "Juni", "physical_router_product_name": "MX", "physical_router_device_family": "juniper-mx" } fake_vnc_lib = flexmock() flexmock(VncApi, __new__=fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router_uuid = "1ef6cf9d-c2e2-4004-810a-43d471c94dc5" flexmock(fake_vnc_lib).should_receive( 'physical_router_create').and_return(fake_router_uuid) set_module_args( dict(object_type='physical_router', object_op='create', object_dict=object_dict, auth_token="1234")) result = self.execute_module() self.assertEqual(result['failed'], False)
def test_vnc_bulk_update(self): object_list = [{ "uuid": "1ef6cf9d-c2e2-4004-810a-43d471c94dc5", "physical_router_user_credentials": { "username": "******", "password": "******" } }] fake_vnc_lib = flexmock() flexmock(VncApi, __new__=fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router_obj = "{'physical-router' : {'uuid': '1ef6cf9d-c2e2-4004-810a-43d471c94dc5'}}" flexmock(fake_vnc_lib).should_receive( 'physical_router_update').and_return(fake_router_obj) flexmock(fake_vnc_lib).should_receive( 'id_to_fq_name').and_return('MX-240') set_module_args( dict(enable_job_ctx=False, object_type='physical_router', object_op='bulk_update', object_list=object_list, job_ctx={"auth_token": "1234"})) result = self.execute_module() self.assertEqual(result.get('failed'), None)
def test_vnc_update(self): object_dict = { "uuid": "1ef6cf9d-c2e2-4004-810a-43d471c94dc5", "physical_router_user_credentials": { "username": "******", "password": "******" } } fake_vnc_lib = flexmock() flexmock(VncApi, __new__=fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router_obj = "{'physical-router' : {'uuid': '1ef6cf9d-c2e2-4004-810a-43d471c94dc5'}}" flexmock(fake_vnc_lib).should_receive( 'physical_router_update').and_return(fake_router_obj) flexmock(fake_vnc_lib).should_receive('id_to_fq_name').and_return( 'MX-240') set_module_args( dict(object_type='physical_router', object_op='update', object_dict=object_dict, auth_token="1234")) result = self.execute_module() self.assertEqual(result['failed'], False)
def _test_vnc_bulk_update(self): object_list = [{ "uuid": "1ef6cf9d-c2e2-4004-810a-43d471c94dc5", "physical_router_user_credentials": { "username": "******", "password": "******" } }] fake_vnc_lib = flexmock() flexmock(JobVncApi).should_receive('vnc_init').and_return(fake_vnc_lib) fake_vnc_lib.should_receive('__init__') fake_router_obj = "{'physical-router' : {'uuid': '1ef6cf9d-c2e2-4004-810a-43d471c94dc5'}}" flexmock(fake_vnc_lib).should_receive( 'physical_router_update').and_return(fake_router_obj) flexmock(fake_vnc_lib).should_receive('id_to_fq_name').and_return( 'MX-240') set_module_args( dict(enable_job_ctx=False, object_type='physical_router', object_op='bulk_update', object_list=object_list, job_ctx={"auth_token": "1234"})) result = self.execute_module() self.assertEqual(result.get('failed'), None)
def test_fileutility04(self): flexmock( swiftclient.client).should_receive('Connection').and_return(None) set_module_args(self.args_dict) self.assertRaises(Exception, self.execute_module())
def test_invalid_vnc_class(self): set_module_args(dict(object_type='lldp', object_op='get', object_dict={"uuid": "1234"}, auth_token="1234")) result = self.execute_module(failed=True) self.assertEqual(result['failed'], True)
def test_fileutility_03(self): flexmock(swiftclient.utils).should_receive('generate_temp_url').and_raise(Exception) set_module_args(self.args_dict) self.assertRaises(Exception, self.execute_module())
def test_fileutility04(self): flexmock(swiftclient.client).should_receive('Connection').and_return(None) self.args_dict['connection_retry_count'] = 1 set_module_args(self.args_dict) self.assertRaises(Exception, self.execute_module())
def test_fileutility05(self): flexmock(self.mockobj).should_receive('post_account').and_raise(Exception) set_module_args(self.args_dict) self.assertRaises(Exception, self.execute_module())
def test_fileutility_03(self): flexmock(swiftclient.utils).should_receive( 'generate_temp_url').and_raise(Exception) set_module_args(self.args_dict) self.assertRaises(Exception, self.execute_module())
def test_fileutility05(self): flexmock( self.mockobj).should_receive('post_account').and_raise(Exception) set_module_args(self.args_dict) self.assertRaises(Exception, self.execute_module())