def test_dm_cassandra(self):
#wait for dm
self.check_dm_state()
dm_cs = DMCassandraDB.get_instance()
ip_addr = '10.0.0.1/32'
lo0_ip_addr = '20.0.0.1/32'
key = 'pr-uuid' + ':' + 'vn-uuid' + ':' + 'sub-prefix'
dm_cs.add(dm_cs._PR_VN_IP_CF, key, {DMUtils.get_ip_cs_column_name('irb'): ip_addr})
dm_cs.add(dm_cs._PR_VN_IP_CF, key, {DMUtils.get_ip_cs_column_name('lo0'): lo0_ip_addr})
dm_cs.init_pr_map()
key_value = dm_cs.get_one_col(dm_cs._PR_VN_IP_CF, key, DMUtils.get_ip_cs_column_name('irb'))
if key_value != ip_addr:
self.assertTrue(False)
key_value = dm_cs.get_one_col(dm_cs._PR_VN_IP_CF, key, DMUtils.get_ip_cs_column_name('lo0'))
if key_value != lo0_ip_addr:
self.assertTrue(False)
# delete one column
dm_cs.delete(dm_cs._PR_VN_IP_CF, key, [DMUtils.get_ip_cs_column_name('lo0')])
try:
key_value = dm_cs.get_one_col(dm_cs._PR_VN_IP_CF, key, DMUtils.get_ip_cs_column_name('lo0'))
if key_value is not None:
self.assertTrue(False)
except NoIdError:
pass
key_value = dm_cs.get_one_col(dm_cs._PR_VN_IP_CF, key, DMUtils.get_ip_cs_column_name('irb'))
if key_value != ip_addr:
self.assertTrue(False)
return
def test_ae_id_alloc_cassandra(self):
#wait for dm
self.check_dm_state()
dm_cs = DMCassandraDB.get_instance()
esi = '10:10:10:10:10:10:10:10:10'
pr_uuid = 'pr_uuid'
key = pr_uuid + ':' + esi
ae_id = 127
dm_cs.add_ae_id(pr_uuid, esi, ae_id)
dm_cs.init_pr_ae_map()
self.assertEqual(dm_cs.get_ae_id(pr_uuid + ':' + esi), ae_id)
check_db_ae_id = dm_cs.pr_ae_id_map[pr_uuid][esi]
self.assertEqual(check_db_ae_id, ae_id)
key_value = dm_cs.get_one_col(dm_cs._PR_AE_ID_CF, key, "index")
if key_value != ae_id:
self.assertTrue(False)
# delete one column
dm_cs.delete(dm_cs._PR_AE_ID_CF, key, ["index"])
try:
key_value = dm_cs.get_one_col(dm_cs._PR_AE_ID_CF, key, "index")
if key_value is not None:
self.assertTrue(False)
except NoIdError:
pass
return
def check_dm_state(self):
try:
dm_cs = DMCassandraDB.getInstance()
if dm_cs:
self.assertTrue(True)
except:
self.assertTrue(False)
def test_dm_cassandra(self):
#wait for dm
self.check_dm_state()
dm_cs = DMCassandraDB.get_instance()
ip_addr = '10.0.0.1/32'
lo0_ip_addr = '20.0.0.1/32'
key = 'pr-uuid' + ':' + 'vn-uuid' + ':' + 'sub-prefix'
dm_cs.add(dm_cs._PR_VN_IP_CF, key, {DMUtils.get_ip_cs_column_name('irb'): ip_addr})
dm_cs.add(dm_cs._PR_VN_IP_CF, key, {DMUtils.get_ip_cs_column_name('lo0'): lo0_ip_addr})
dm_cs.init_pr_map()
key_value = dm_cs.get_one_col(dm_cs._PR_VN_IP_CF, key, DMUtils.get_ip_cs_column_name('irb'))
if key_value != ip_addr:
self.assertTrue(False)
key_value = dm_cs.get_one_col(dm_cs._PR_VN_IP_CF, key, DMUtils.get_ip_cs_column_name('lo0'))
if key_value != lo0_ip_addr:
self.assertTrue(False)
# delete one column
dm_cs.delete(dm_cs._PR_VN_IP_CF, key, [DMUtils.get_ip_cs_column_name('lo0')])
try:
key_value = dm_cs.get_one_col(dm_cs._PR_VN_IP_CF, key, DMUtils.get_ip_cs_column_name('lo0'))
if key_value is not None:
self.assertTrue(False)
except NoIdError:
pass
key_value = dm_cs.get_one_col(dm_cs._PR_VN_IP_CF, key, DMUtils.get_ip_cs_column_name('irb'))
if key_value != ip_addr:
self.assertTrue(False)
return
def test_ae_id_alloc_cassandra(self):
#wait for dm
self.check_dm_state()
dm_cs = DMCassandraDB.get_instance()
esi = '10:10:10:10:10:10:10:10:10'
pr_uuid = 'pr_uuid'
key = pr_uuid + ':' + esi
ae_id = 127
dm_cs.add_ae_id(pr_uuid, esi, ae_id)
dm_cs.init_pr_ae_map()
self.assertEqual(dm_cs.get_ae_id(pr_uuid + ':' + esi), ae_id)
check_db_ae_id = dm_cs.pr_ae_id_map[pr_uuid][esi]
self.assertEqual(check_db_ae_id, ae_id)
key_value = dm_cs.get_one_col(dm_cs._PR_AE_ID_CF, key, "index")
if key_value != ae_id:
self.assertTrue(False)
# delete one column
dm_cs.delete(dm_cs._PR_AE_ID_CF, key, ["index"])
try:
key_value = dm_cs.get_one_col(dm_cs._PR_AE_ID_CF, key, "index")
if key_value is not None:
self.assertTrue(False)
except NoIdError:
pass
return
def check_dm_state(self):
self.assertIsNotNone(DMCassandraDB.get_instance())
def check_dm_state(self):
self.assertIsNotNone(DMCassandraDB.get_instance())
