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.getInstance()
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