def test_synchronize_required(self):
"""Tests whether synchronize() sends the right commands.
This test verifies a scenario when the sync is required.
"""
region_updated_time = {"regionName": "RegionOne", "regionTimestamp": "12345"}
self.rpc.get_region_updated_time.return_value = region_updated_time
self.sync_service._region_updated_time = {"regionName": "RegionOne", "regionTimestamp": "0"}
tenant_id = "tenant-1"
network_id = "net-1"
segmentation_id = 42
db.remember_tenant(tenant_id)
db.remember_network(tenant_id, network_id, segmentation_id)
self.rpc.get_tenants.return_value = {}
self.sync_service.do_synchronize()
expected_calls = [
mock.call.get_region_updated_time(),
mock.call._run_openstack_cmds(["sync start"]),
mock.call.register_with_eos(),
mock.call.get_tenants(),
mock.call.create_network_bulk(
tenant_id, [{"network_id": network_id, "segmentation_id": segmentation_id, "network_name": ""}]
),
mock.call._run_openstack_cmds(["sync end"]),
mock.call.get_region_updated_time(),
]
assert self.rpc.mock_calls == expected_calls
db.forget_network(tenant_id, network_id)
db.forget_tenant(tenant_id)
def test_synchronize_one_network(self):
"""Test to ensure that only the required resources are sent to EOS."""
# Store two tenants in a db and a single tenant in EOS.
# The sync should send details of the second tenant to EOS
tenant_1_id = 'tenant-1'
tenant_1_net_1_id = 'ten-1-net-1'
tenant_1_net_1_seg_id = 11
db.remember_tenant(tenant_1_id)
db.remember_network(tenant_1_id, tenant_1_net_1_id,
tenant_1_net_1_seg_id)
tenant_2_id = 'tenant-2'
tenant_2_net_1_id = 'ten-2-net-1'
tenant_2_net_1_seg_id = 21
db.remember_tenant(tenant_2_id)
db.remember_network(tenant_2_id, tenant_2_net_1_id,
tenant_2_net_1_seg_id)
self.rpc.get_tenants.return_value = {
tenant_1_id: {
'tenantVmInstances': {},
'tenantNetworks': {
tenant_1_net_1_id: {
'networkId': tenant_1_net_1_id,
'networkName': 'Net1',
'segmenationType': 'vlan',
'segmentationTypeId': tenant_1_net_1_seg_id,
}
}
}
}
self.sync_service.do_synchronize()
expected_calls = [
mock.call.get_region_updated_time(),
mock.call._run_openstack_cmds(['sync start']),
mock.call.register_with_eos(),
mock.call.get_tenants(),
mock.call.create_network_bulk(
tenant_2_id,
[{'network_id': tenant_2_net_1_id,
'segmentation_id': tenant_2_net_1_seg_id,
'network_name': ''}]),
mock.call._run_openstack_cmds(['sync end']),
mock.call.get_region_updated_time()
]
self.assertTrue(self.rpc.mock_calls == expected_calls,
"Seen: %s\nExpected: %s" % (
self.rpc.mock_calls,
expected_calls,
)
)
db.forget_network(tenant_1_id, tenant_1_net_1_id)
db.forget_network(tenant_2_id, tenant_2_net_1_id)
db.forget_tenant(tenant_1_id)
db.forget_tenant(tenant_2_id)
def test_synchronize_one_network(self):
"""Test to ensure that only the required resources are sent to EOS."""
# Store two tenants in a db and a single tenant in EOS.
# The sync should send details of the second tenant to EOS
tenant_1_id = 'tenant-1'
tenant_1_net_1_id = 'ten-1-net-1'
tenant_1_net_1_seg_id = 11
db.remember_tenant(tenant_1_id)
db.remember_network(tenant_1_id, tenant_1_net_1_id,
tenant_1_net_1_seg_id)
tenant_2_id = 'tenant-2'
tenant_2_net_1_id = 'ten-2-net-1'
tenant_2_net_1_seg_id = 21
db.remember_tenant(tenant_2_id)
db.remember_network(tenant_2_id, tenant_2_net_1_id,
tenant_2_net_1_seg_id)
self.rpc.get_tenants.return_value = {
tenant_1_id: {
'tenantVmInstances': {},
'tenantNetworks': {
tenant_1_net_1_id: {
'networkId': tenant_1_net_1_id,
'networkName': 'Net1',
'segmenationType': 'vlan',
'segmentationTypeId': tenant_1_net_1_seg_id,
}
}
}
}
self.sync_service.do_synchronize()
expected_calls = [
mock.call.get_region_updated_time(),
mock.call._run_openstack_cmds(['sync start']),
mock.call.register_with_eos(),
mock.call.get_tenants(),
mock.call.create_network_bulk(tenant_2_id, [{
'network_id': tenant_2_net_1_id,
'segmentation_id': tenant_2_net_1_seg_id,
'network_name': ''
}]),
mock.call._run_openstack_cmds(['sync end']),
mock.call.get_region_updated_time()
]
self.assertTrue(
self.rpc.mock_calls == expected_calls, "Seen: %s\nExpected: %s" % (
self.rpc.mock_calls,
expected_calls,
))
db.forget_network(tenant_1_id, tenant_1_net_1_id)
db.forget_network(tenant_2_id, tenant_2_net_1_id)
db.forget_tenant(tenant_1_id)
db.forget_tenant(tenant_2_id)
def test_network_is_removed(self):
tenant_id = 'test'
network_id = '123'
db.remember_network(tenant_id, network_id, '123')
db.forget_network(tenant_id, network_id)
net_provisioned = db.is_network_provisioned(tenant_id, network_id)
self.assertFalse(net_provisioned, 'The network should be deleted')
def test_network_is_removed(self):
tenant_id = 'test'
network_id = '123'
db.remember_network(tenant_id, network_id, '123')
db.forget_network(tenant_id, network_id)
net_provisioned = db.is_network_provisioned(tenant_id, network_id)
self.assertFalse(net_provisioned, 'The network should be deleted')
def test_network_is_removed(self):
tenant_id = "test"
network_id = "123"
db.remember_network(tenant_id, network_id, "123")
db.forget_network(tenant_id, network_id)
net_provisioned = db.is_network_provisioned(tenant_id, network_id)
self.assertFalse(net_provisioned, "The network should be deleted")
def delete_network_precommit(self, context):
"""Delete the network infromation from the DB."""
network = context.current
network_id = network['id']
tenant_id = network['tenant_id']
with self.eos_sync_lock:
if db.is_network_provisioned(tenant_id, network_id):
db.forget_network(tenant_id, network_id)
# if necessary, delete tenant as well.
self.delete_tenant(tenant_id)
def delete_network_precommit(self, context):
"""Delete the network infromation from the DB."""
network = context.current
network_id = network['id']
tenant_id = network['tenant_id']
with self.eos_sync_lock:
if db.is_network_provisioned(tenant_id, network_id):
db.forget_network(tenant_id, network_id)
# if necessary, delete tenant as well.
self.delete_tenant(tenant_id)
def _cleanup_db(self):
"""Clean up any uncessary entries in our DB."""
db_tenants = db.get_tenants()
for tenant in db_tenants:
neutron_nets = self.ndb.get_all_networks_for_tenant(tenant)
neutron_nets_id = []
for net in neutron_nets:
neutron_nets_id.append(net['id'])
db_nets = db.get_networks(tenant)
for net_id in db_nets.keys():
if net_id not in neutron_nets_id:
db.forget_network(tenant, net_id)
def _cleanup_db(self):
"""Clean up any uncessary entries in our DB."""
db_tenants = db.get_tenants()
for tenant in db_tenants:
neutron_nets = self.ndb.get_all_networks_for_tenant(tenant)
neutron_nets_id = []
for net in neutron_nets:
neutron_nets_id.append(net['id'])
db_nets = db.get_networks(tenant)
for net_id in db_nets.keys():
if net_id not in neutron_nets_id:
db.forget_network(tenant, net_id)
def test_removes_all_networks(self):
tenant_id = 'test'
num_nets = 100
old_nets = db.num_nets_provisioned(tenant_id)
nets = ['id_%s' % n for n in range(num_nets)]
for net_id in nets:
db.remember_network(tenant_id, net_id, 123)
for net_id in nets:
db.forget_network(tenant_id, net_id)
num_nets_provisioned = db.num_nets_provisioned(tenant_id)
expected = old_nets
self.assertEqual(expected, num_nets_provisioned,
'There should be %d nets, not %d' %
(expected, num_nets_provisioned))
def test_removes_all_networks(self):
tenant_id = 'test'
num_nets = 100
old_nets = db.num_nets_provisioned(tenant_id)
nets = ['id_%s' % n for n in range(num_nets)]
for net_id in nets:
db.remember_network(tenant_id, net_id, 123)
for net_id in nets:
db.forget_network(tenant_id, net_id)
num_nets_provisioned = db.num_nets_provisioned(tenant_id)
expected = old_nets
self.assertEqual(expected, num_nets_provisioned,
'There should be %d nets, not %d' %
(expected, num_nets_provisioned))
def test_synchronize_required(self):
"""Tests whether synchronize() sends the right commands.
This test verifies a scenario when the sync is required.
"""
region_updated_time = {
'regionName': 'RegionOne',
'regionTimestamp': '12345'
}
self.rpc.get_region_updated_time.return_value = region_updated_time
self.sync_service._region_updated_time = {
'regionName': 'RegionOne',
'regionTimestamp': '0',
}
tenant_id = 'tenant-1'
network_id = 'net-1'
segmentation_id = 42
db.remember_tenant(tenant_id)
db.remember_network(tenant_id, network_id, segmentation_id)
self.rpc.get_tenants.return_value = {}
self.sync_service.do_synchronize()
expected_calls = [
mock.call.get_region_updated_time(),
mock.call._run_openstack_cmds(['sync start']),
mock.call.register_with_eos(),
mock.call.get_tenants(),
mock.call.create_network_bulk(
tenant_id,
[{'network_id': network_id,
'segmentation_id': segmentation_id,
'network_name': ''}]),
mock.call._run_openstack_cmds(['sync end']),
mock.call.get_region_updated_time()
]
assert self.rpc.mock_calls == expected_calls
db.forget_network(tenant_id, network_id)
db.forget_tenant(tenant_id)
def test_synchronize_required(self):
"""Tests whether synchronize() sends the right commands.
This test verifies a scenario when the sync is required.
"""
region_updated_time = {
'regionName': 'RegionOne',
'regionTimestamp': '12345'
}
self.rpc.get_region_updated_time.return_value = region_updated_time
self.sync_service._region_updated_time = {
'regionName': 'RegionOne',
'regionTimestamp': '0',
}
tenant_id = 'tenant-1'
network_id = 'net-1'
segmentation_id = 42
db.remember_tenant(tenant_id)
db.remember_network(tenant_id, network_id, segmentation_id)
self.rpc.get_tenants.return_value = {}
self.sync_service.do_synchronize()
expected_calls = [
mock.call.get_region_updated_time(),
mock.call._run_openstack_cmds(['sync start']),
mock.call.register_with_eos(),
mock.call.get_tenants(),
mock.call.create_network_bulk(tenant_id, [{
'network_id': network_id,
'segmentation_id': segmentation_id,
'network_name': ''
}]),
mock.call._run_openstack_cmds(['sync end']),
mock.call.get_region_updated_time()
]
assert self.rpc.mock_calls == expected_calls
db.forget_network(tenant_id, network_id)
db.forget_tenant(tenant_id)
def test_synchronize_all_networks(self):
"""Test to ensure that only the required resources are sent to EOS."""
# Store two tenants in a db and none on EOS.
# The sync should send details of all tenants to EOS
tenant_1_id = u'tenant-1'
tenant_1_net_1_id = u'ten-1-net-1'
tenant_1_net_1_seg_id = 11
db.remember_tenant(tenant_1_id)
db.remember_network(tenant_1_id, tenant_1_net_1_id,
tenant_1_net_1_seg_id)
tenant_2_id = u'tenant-2'
tenant_2_net_1_id = u'ten-2-net-1'
tenant_2_net_1_seg_id = 21
db.remember_tenant(tenant_2_id)
db.remember_network(tenant_2_id, tenant_2_net_1_id,
tenant_2_net_1_seg_id)
self.rpc.get_tenants.return_value = {}
self.sync_service.do_synchronize()
expected_calls = [
mock.call.get_region_updated_time(),
mock.call._run_openstack_cmds(['sync start']),
mock.call.register_with_eos(),
mock.call.get_tenants(),
mock.call.create_network_bulk(
tenant_1_id,
[{'network_id': tenant_1_net_1_id,
'segmentation_id': tenant_1_net_1_seg_id,
'network_name': ''}]),
mock.call.create_network_bulk(
tenant_2_id,
[{'network_id': tenant_2_net_1_id,
'segmentation_id': tenant_2_net_1_seg_id,
'network_name': ''}]),
mock.call._run_openstack_cmds(['sync end']),
mock.call.get_region_updated_time()
]
# The create_network_bulk() can be called in different order. So split
# it up. The first part checks if the initial set of methods are
# invoked.
self.assertTrue(self.rpc.mock_calls[:4] == expected_calls[:4],
"Seen: %s\nExpected: %s" % (
self.rpc.mock_calls,
expected_calls,
)
)
# Check if tenant 1 networks are created. It must be one of the two
# methods.
self.assertTrue(self.rpc.mock_calls[4] in expected_calls[4:6],
"Seen: %s\nExpected: %s" % (
self.rpc.mock_calls,
expected_calls,
)
)
# Check if tenant 2 networks are created. It must be one of the two
# methods.
self.assertTrue(self.rpc.mock_calls[5] in expected_calls[4:6],
"Seen: %s\nExpected: %s" % (
self.rpc.mock_calls,
expected_calls,
)
)
# Check if the sync end methods are invoked.
self.assertTrue(self.rpc.mock_calls[6:8] == expected_calls[6:8],
"Seen: %s\nExpected: %s" % (
self.rpc.mock_calls,
expected_calls,
)
)
db.forget_network(tenant_1_id, tenant_1_net_1_id)
db.forget_network(tenant_2_id, tenant_2_net_1_id)
db.forget_tenant(tenant_1_id)
db.forget_tenant(tenant_2_id)
def test_synchronize_all_networks(self):
"""Test to ensure that only the required resources are sent to EOS."""
# Store two tenants in a db and none on EOS.
# The sync should send details of all tenants to EOS
tenant_1_id = u'tenant-1'
tenant_1_net_1_id = u'ten-1-net-1'
tenant_1_net_1_seg_id = 11
db.remember_tenant(tenant_1_id)
db.remember_network(tenant_1_id, tenant_1_net_1_id,
tenant_1_net_1_seg_id)
tenant_2_id = u'tenant-2'
tenant_2_net_1_id = u'ten-2-net-1'
tenant_2_net_1_seg_id = 21
db.remember_tenant(tenant_2_id)
db.remember_network(tenant_2_id, tenant_2_net_1_id,
tenant_2_net_1_seg_id)
self.rpc.get_tenants.return_value = {}
self.sync_service.do_synchronize()
expected_calls = [
mock.call.get_region_updated_time(),
mock.call._run_openstack_cmds(['sync start']),
mock.call.register_with_eos(),
mock.call.get_tenants(),
mock.call.create_network_bulk(tenant_1_id, [{
'network_id': tenant_1_net_1_id,
'segmentation_id': tenant_1_net_1_seg_id,
'network_name': ''
}]),
mock.call.create_network_bulk(tenant_2_id, [{
'network_id': tenant_2_net_1_id,
'segmentation_id': tenant_2_net_1_seg_id,
'network_name': ''
}]),
mock.call._run_openstack_cmds(['sync end']),
mock.call.get_region_updated_time()
]
# The create_network_bulk() can be called in different order. So split
# it up. The first part checks if the initial set of methods are
# invoked.
self.assertTrue(
self.rpc.mock_calls[:4] == expected_calls[:4],
"Seen: %s\nExpected: %s" % (
self.rpc.mock_calls,
expected_calls,
))
# Check if tenant 1 networks are created. It must be one of the two
# methods.
self.assertTrue(
self.rpc.mock_calls[4] in expected_calls[4:6],
"Seen: %s\nExpected: %s" % (
self.rpc.mock_calls,
expected_calls,
))
# Check if tenant 2 networks are created. It must be one of the two
# methods.
self.assertTrue(
self.rpc.mock_calls[5] in expected_calls[4:6],
"Seen: %s\nExpected: %s" % (
self.rpc.mock_calls,
expected_calls,
))
# Check if the sync end methods are invoked.
self.assertTrue(
self.rpc.mock_calls[6:8] == expected_calls[6:8],
"Seen: %s\nExpected: %s" % (
self.rpc.mock_calls,
expected_calls,
))
db.forget_network(tenant_1_id, tenant_1_net_1_id)
db.forget_network(tenant_2_id, tenant_2_net_1_id)
db.forget_tenant(tenant_1_id)
db.forget_tenant(tenant_2_id)