def test_delete_datasource_with_data(self):
id_ = uuidutils.generate_uuid()
datasources.add_datasource(
id_=id_, name="hiya", driver="foo", config="{user: foo}", description="hello", enabled=True
)
db_ds_table_data.store_ds_table_data(
ds_id=id_, tablename="bar", tabledata=set([("a1", "b1"), ("a2", "b2"), ("a3", "a4")])
)
self.assertTrue(datasources.delete_datasource_with_data(id_))
self.assertEqual(db_ds_table_data.get_ds_table_data(id_), [])
def test_delete_ds_table_data(self):
ds_id = uuidutils.generate_uuid()
db_ds_table_data.store_ds_table_data(ds_id=ds_id,
tablename='table1',
tabledata=set([('a', 0),
('b', 1)]))
db_ds_table_data.store_ds_table_data(ds_id=ds_id,
tablename='table2',
tabledata=set([('a', 0),
('b', 2)]))
self.assertTrue(db_ds_table_data.delete_ds_table_data(ds_id, 'table1'))
self.assertIsNone(db_ds_table_data.get_ds_table_data(ds_id, 'table1'))
self.assertEqual(set([('a', 0), ('b', 2)]),
db_ds_table_data.get_ds_table_data(ds_id, 'table2'))
def test_delete_ds_table_data(self):
ds_id = uuidutils.generate_uuid()
db_ds_table_data.store_ds_table_data(
ds_id=ds_id,
tablename='table1',
tabledata=set([('a', 0), ('b', 1)]))
db_ds_table_data.store_ds_table_data(
ds_id=ds_id,
tablename='table2',
tabledata=set([('a', 0), ('b', 2)]))
self.assertTrue(db_ds_table_data.delete_ds_table_data(ds_id, 'table1'))
self.assertIsNone(db_ds_table_data.get_ds_table_data(ds_id, 'table1'))
self.assertEqual(set([('a', 0), ('b', 2)]),
db_ds_table_data.get_ds_table_data(ds_id, 'table2'))
def test_delete_datasource_with_data(self):
id_ = uuidutils.generate_uuid()
datasources.add_datasource(id_=id_,
name="hiya",
driver="foo",
config={'user': '******'},
description="hello",
enabled=True)
db_ds_table_data.store_ds_table_data(ds_id=id_,
tablename='bar',
tabledata=set([('a1', 'b1'),
('a2', 'b2'),
('a3', 'a4')]))
self.assertTrue(datasources.delete_datasource_with_data(id_))
self.assertEqual(db_ds_table_data.get_ds_table_data(id_), [])
def test_delete_datasource_with_data(self):
id_ = uuidutils.generate_uuid()
datasources.add_datasource(
id_=id_,
name="hiya",
driver="foo",
config={'user': '******'},
description="hello",
enabled=True)
db_ds_table_data.store_ds_table_data(
ds_id=id_,
tablename='bar',
tabledata=set([('a1', 'b1'), ('a2', 'b2'), ('a3', 'a4')])
)
self.assertTrue(datasources.delete_datasource_with_data(id_))
self.assertEqual(db_ds_table_data.get_ds_table_data(id_), [])
def test_store_ds_table_data(self):
ds_id = uuidutils.generate_uuid()
# store new data
source = db_ds_table_data.store_ds_table_data(ds_id=ds_id,
tablename='table1',
tabledata=set([('a', 0)
]))
self.assertEqual(ds_id, source.ds_id)
self.assertEqual('table1', source.tablename)
self.assertEqual('[["a", 0]]', source.tabledata)
# update exsting data
db_ds_table_data.store_ds_table_data(ds_id=ds_id,
tablename='table1',
tabledata=set([('a', 0),
('new', 1)]))
data = db_ds_table_data.get_ds_table_data(ds_id, 'table1')
self.assertEqual(set([('a', 0), ('new', 1)]), data)
def test_store_ds_table_data(self):
ds_id = uuidutils.generate_uuid()
# store new data
source = db_ds_table_data.store_ds_table_data(
ds_id=ds_id,
tablename='table1',
tabledata=set([('a', 0)]))
self.assertEqual(ds_id, source.ds_id)
self.assertEqual('table1', source.tablename)
self.assertEqual('[["a", 0]]', source.tabledata)
# update exsting data
db_ds_table_data.store_ds_table_data(
ds_id=ds_id,
tablename='table1',
tabledata=set([('a', 0), ('new', 1)]))
data = db_ds_table_data.get_ds_table_data(ds_id, 'table1')
self.assertEqual(set([('a', 0), ('new', 1)]), data)
def test_get_ds_table_data(self):
ds_id = uuidutils.generate_uuid()
ds2_id = uuidutils.generate_uuid()
db_ds_table_data.store_ds_table_data(ds_id=ds_id,
tablename='table1',
tabledata=set([('a', 0),
('b', 1)]))
db_ds_table_data.store_ds_table_data(ds_id=ds_id,
tablename='table2',
tabledata=set([('a', 0),
('b', 2)]))
db_ds_table_data.store_ds_table_data(ds_id=ds2_id,
tablename='table3',
tabledata=set([('a', 0),
('b', 3)]))
data = db_ds_table_data.get_ds_table_data(ds_id, 'table1')
self.assertEqual(set([('a', 0), ('b', 1)]), data)
data = db_ds_table_data.get_ds_table_data(ds_id)
self.assertEqual(2, len(data))
self.assertTrue({
'tablename': 'table1',
'tabledata': set([('a', 0), ('b', 1)])
} in data)
self.assertTrue({
'tablename': 'table2',
'tabledata': set([('a', 0), ('b', 2)])
} in data)
def test_get_ds_table_data(self):
ds_id = uuidutils.generate_uuid()
ds2_id = uuidutils.generate_uuid()
db_ds_table_data.store_ds_table_data(
ds_id=ds_id,
tablename='table1',
tabledata=set([('a', 0), ('b', 1)]))
db_ds_table_data.store_ds_table_data(
ds_id=ds_id,
tablename='table2',
tabledata=set([('a', 0), ('b', 2)]))
db_ds_table_data.store_ds_table_data(
ds_id=ds2_id,
tablename='table3',
tabledata=set([('a', 0), ('b', 3)]))
data = db_ds_table_data.get_ds_table_data(ds_id, 'table1')
self.assertEqual(set([('a', 0), ('b', 1)]), data)
data = db_ds_table_data.get_ds_table_data(ds_id)
self.assertEqual(2, len(data))
self.assertIn(
{'tablename': 'table1',
'tabledata': set([('a', 0), ('b', 1)])},
data)
self.assertIn(
{'tablename': 'table2',
'tabledata': set([('a', 0), ('b', 2)])},
data)
