def test_get_initial_admin(self, mock_bgpvpn_api): self.bgpvpn_view.request.user.is_superuser = True bgpvpn_data = { "name": "foo-name", "id": "foo-id", "type": "l3", "route_targets": ["65432:1", "65432:2"], "import_targets": [], "export_targets": [] } expected_data = { "name": "foo-name", "bgpvpn_id": "foo-id", "type": "l3", "route_targets": "65432:1,65432:2", "import_targets": "", "export_targets": "" } mock_bgpvpn_api.bgpvpn_get.return_value = bgpvpn_api.Bgpvpn( bgpvpn_data) result = self.bgpvpn_view.get_initial() mock_bgpvpn_api.bgpvpn_get.assert_called_once_with( self.bgpvpn_view.request, "foo-id") for key, val in expected_data.items(): self.assertEqual(val, result[key])
def test_handle_update(self, mock_api, mock_bgpvpn_api): data = {"bgpvpn_id": "foo-id", "type": "l3", "name": "foo-name", "route_targets": "65421:1", "import_targets": "65421:2", "export_targets": "65421:3"} mock_api.keystone.tenant_list.return_value = [], False self.bgpvpn_form = bgpvpn_a_form.CreateBgpVpn(self.mock_request) self.bgpvpn_form.action = "update" expected_data = bgpvpn_api.Bgpvpn({"bgpvpn_id": "foo-id", "name": "foo-name", "tenant_id": "tenant_id", "route_targets": ["65421:1"], "import_targets": ["65421:2"], "export_targets": ["65421:3"]}) mock_bgpvpn_api.bgpvpn_update.return_value = expected_data result = self.bgpvpn_form.handle(self.mock_request, data) mock_bgpvpn_api.bgpvpn_update.assert_called_once_with( self.mock_request, "foo-id", name="foo-name", route_targets=["65421:1"], import_targets=["65421:2"], export_targets=["65421:3"]) self.assertEqual(result, expected_data)
def data(TEST): TEST.bgpvpns = utils.TestDataContainer() TEST.api_bgpvpns = utils.TestDataContainer() TEST.network_associations = utils.TestDataContainer() TEST.api_network_associations = utils.TestDataContainer() TEST.router_associations = utils.TestDataContainer() TEST.api_router_associations = utils.TestDataContainer() bgpvpn_dict = {'id': 'b595e758-1877-4aec-92a2-6834d76f1025', 'tenant_id': '1', 'name': 'bgpvpn1', 'route_targets': '64500:1' } TEST.api_bgpvpns.add(bgpvpn_dict) b = bgpvpn.Bgpvpn(copy.deepcopy(bgpvpn_dict)) TEST.bgpvpns.add(b) network_association_dict = { 'id': '99ef096d-21fb-43a7-9e2a-b3c464abef3a', 'network_id': '063cf7f3-ded1-4297-bc4c-31eae876cc91', 'tenant_id': '1'} TEST.api_network_associations.add(network_association_dict) na = bgpvpn.NetworkAssociation(copy.deepcopy(network_association_dict)) TEST.network_associations.add(na) router_association_dict = { 'id': '9736c228-745d-4e78-83a5-d971d9fd8f2c', 'router_id': '279989f7-54bb-41d9-ba42-0d61f12fda61', 'tenant_id': '1'} TEST.api_router_associations.add(router_association_dict) ra = bgpvpn.RouterAssociation(copy.deepcopy(router_association_dict)) TEST.router_associations.add(ra)
def _get_mock_bgpvpn(self, prefix): bgpvpn_info = {} if prefix: bgpvpn_info = { "name": "%s_name" % prefix, "route_targets": [], "import_targets": [], "export_targets": [], "networks": [], "routers": [], "tenant_id": "tenant_id", "type": "l3" } return bgpvpn_api.Bgpvpn(bgpvpn_info)
def test_handle(self, mock_bgpvpn_api): self.bgpvpn_form.request.user.is_superuser = False test_data = {"bgpvpn_id": "foo-id", "name": "foo-name", "type": "l3"} expected_data = bgpvpn_api.Bgpvpn({ "id": "foo-id", "name": "foo-name", "type": "l3" }) mock_bgpvpn_api.bgpvpn_update.return_value = expected_data result = self.bgpvpn_form.handle(self.mock_request, test_data) self.assertEqual(expected_data, result) mock_bgpvpn_api.bgpvpn_update.assert_called_once_with( self.mock_request, "foo-id", name="foo-name")
def test_get_context_data(self, mock_bgpvpn_api): expected_bgpvpn = bgpvpn_api.Bgpvpn({"id": "foo-id"}) mock_bgpvpn_api.bgpvpn_get.return_value = expected_bgpvpn context = self.view.get_context_data() mock_bgpvpn_api.bgpvpn_get.assert_called_once_with( self.view.request, "foo-id") self.assertIn('view', context) self.assertIn('bgpvpn', context) self.assertIn('url', context) self.assertIsInstance(context['view'], bgpvpn_views.DetailProjectView) self.assertEqual(expected_bgpvpn, context['bgpvpn']) self.assertEqual(reverse('horizon:project:bgpvpn:index'), context['url'])
def test_get_initial(self, mock_bgpvpn_api): bgpvpn_data = {"name": "foo-name", "id": "foo-id", "type": "l3"} expected_data = { "name": "foo-name", "bgpvpn_id": "foo-id", "type": "l3" } mock_bgpvpn_api.bgpvpn_get.return_value = bgpvpn_api.Bgpvpn( bgpvpn_data) result = self.bgpvpn_view.get_initial() mock_bgpvpn_api.bgpvpn_get.assert_called_once_with( self.bgpvpn_view.request, "foo-id") for key, val in expected_data.items(): self.assertEqual(val, result[key])