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])