def test_cell_data_update_does_create_variables(self):
dbapi.cells_create(self.context, cell1)
res = dbapi.cells_get_by_name(self.context, cell1['region_id'],
cell1['name'])
self.assertEqual(res.variables, {})
variables = {"key1": "value1", "key2": "value2"}
res = dbapi.cells_data_update(self.context, res.id, variables)
self.assertEqual(res.variables, variables)
def test_cell_data_update_does_create_variables(self):
dbapi.cells_create(self.context, cell1)
res = dbapi.cells_get_by_name(self.context, cell1['region_id'],
cell1['name'])
self.assertEqual(res.variables, {})
variables = {"key1": "value1", "key2": "value2"}
res = dbapi.cells_data_update(self.context, res.id, variables)
self.assertEqual(res.variables, variables)
def test_cells_get_all(self):
dbapi.cells_create(self.context, cell1)
filters = {
"region_id": cell1["region_id"],
}
res, _ = dbapi.cells_get_all(self.context, filters, default_pagination)
self.assertEqual(len(res), 1)
self.assertEqual(res[0]['name'], 'cell1')
def test_cell_update(self):
dbapi.cells_create(self.context, cell1)
res = dbapi.cells_get_by_name(self.context, cell1['region_id'],
cell1['name'])
self.assertEqual(res.name, 'cell1')
new_name = 'cell1_New'
res = dbapi.cells_update(self.context, res.id, {'name': 'cell1_New'})
self.assertEqual(res.name, new_name)
def test_cell_delete(self):
dbapi.cells_create(self.context, cell1)
# First make sure we have the cell
res = dbapi.cells_get_by_name(self.context, cell1['region_id'],
cell1['name'])
self.assertEqual(res.name, 'cell1')
dbapi.cells_delete(self.context, res.id)
self.assertRaises(exceptions.NotFound, dbapi.cells_get_by_name,
self.context, 'fake-region', 'fake-cell')
def test_cell_delete(self):
dbapi.cells_create(self.context, cell1)
# First make sure we have the cell
res = dbapi.cells_get_by_name(self.context, cell1['region_id'],
cell1['name'])
self.assertEqual(res.name, 'cell1')
dbapi.cells_delete(self.context, res.id)
self.assertRaises(exceptions.NotFound, dbapi.cells_get_by_name,
self.context, 'fake-region', 'fake-cell')
def test_cell_data_delete(self):
dbapi.cells_create(self.context, cell1)
res = dbapi.cells_get_by_name(self.context, cell1['region_id'],
cell1['name'])
self.assertEqual(res.variables, {})
variables = {"key1": "value1", "key2": "value2"}
res = dbapi.cells_data_update(self.context, res.id, variables)
self.assertEqual(res.variables, variables)
# NOTE(sulo): we delete variables by their key
res = dbapi.cells_data_delete(self.context, res.id, {"key1": "key1"})
self.assertEqual(res.variables, {"key2": "value2"})
def test_cell_data_delete(self):
dbapi.cells_create(self.context, cell1)
res = dbapi.cells_get_by_name(self.context, cell1['region_id'],
cell1['name'])
self.assertEqual(res.variables, {})
variables = {"key1": "value1", "key2": "value2"}
res = dbapi.cells_data_update(self.context, res.id, variables)
self.assertEqual(res.variables, variables)
# NOTE(sulo): we delete variables by their key
res = dbapi.cells_data_delete(self.context, res.id, {"key1": "key1"})
self.assertEqual(res.variables, {"key2": "value2"})
def test_cells_get_all_filter_name(self):
for cell in cells:
dbapi.cells_create(self.context, cell)
setup_res, _ = dbapi.cells_get_all(self.context, {},
default_pagination)
self.assertGreater(len(setup_res), 2)
filters = {
"name": cell1["name"],
}
res, _ = dbapi.cells_get_all(self.context, filters, default_pagination)
self.assertEqual(len(res), 2)
for cell in res:
self.assertEqual(cell['name'], 'cell1')
def test_cells_get_all_filter_id(self):
for cell in cells:
dbapi.cells_create(self.context, cell)
setup_res, _ = dbapi.cells_get_all(self.context, {},
default_pagination)
self.assertGreater(len(setup_res), 2)
self.assertEqual(len([cell for cell in setup_res if cell['id'] == 1]),
1)
filters = {
"id": 1,
}
res, _ = dbapi.cells_get_all(self.context, filters, default_pagination)
self.assertEqual(len(res), 1)
self.assertEqual(res[0]['id'], 1)
def test_cell_update(self):
create_res = dbapi.cells_create(self.context, cell1)
res = dbapi.cells_get_by_id(self.context, create_res.id)
self.assertEqual(res.name, 'cell1')
new_name = 'cell1_New'
res = dbapi.cells_update(self.context, res.id, {'name': 'cell1_New'})
self.assertEqual(res.name, new_name)
def make_cell(self, region_id, name, **variables):
cell = dbapi.cells_create(
self.context,
{'name': name,
'project_id': self.project_id,
'region_id': region_id,
'variables': variables})
return cell.id
def test_cell_delete(self):
create_res = dbapi.cells_create(self.context, cell1)
# First make sure we have the cell
res = dbapi.cells_get_by_id(self.context, create_res.id)
self.assertEqual(res.name, 'cell1')
dbapi.cells_delete(self.context, res.id)
self.assertRaises(exceptions.NotFound, dbapi.cells_get_by_id,
self.context, res.id)
def test_cells_get_all_with_filters_noexist(self):
res = dbapi.cells_create(self.context, cell1)
variables = {"key1": "value1", "key2": "value2"}
dbapi.variables_update_by_resource_id(self.context, "cells", res.id,
variables)
filters = {}
filters["vars"] = "key2:value5"
res, _ = dbapi.cells_get_all(self.context, filters, default_pagination)
self.assertEqual(len(res), 0)
def test_network_device_get_all_filter_cell_id(self):
region_id = 1
cell1 = dbapi.cells_create(
self.context,
{
'name': 'cell1',
'project_id': project_id1,
'cloud_id': cloud_id1,
'region_id': region_id,
}
)
cell2 = dbapi.cells_create(
self.context,
{
'name': 'cell2',
'project_id': project_id1,
'cloud_id': cloud_id1,
'region_id': region_id,
}
)
dbapi.network_devices_create(
self.context, dict(cell_id=cell1.id, **device1)
)
dbapi.network_devices_create(
self.context, dict(cell_id=cell2.id, **device2)
)
setup_res, _ = dbapi.network_devices_get_all(self.context, {},
default_pagination)
self.assertEqual(len(setup_res), 2)
matches = [dev for dev in setup_res if dev['cell_id'] == cell1.id]
self.assertEqual(len(matches), 1)
filters = {
'cell_id': cell1.id,
}
res, _ = dbapi.network_devices_get_all(self.context, filters,
default_pagination)
self.assertEqual(len(res), 1)
self.assertEqual(res[0]['cell_id'], cell1.id)
def test_cells_get_all_with_filters(self):
res = dbapi.cells_create(self.context, cell1)
variables = {"key1": "value1", "key2": "value2"}
dbapi.variables_update_by_resource_id(self.context, "cells", res.id,
variables)
filters = {
"vars": "key2:value2",
"region_id": cell1["region_id"],
}
res, _ = dbapi.cells_get_all(self.context, filters, default_pagination)
self.assertEqual(len(res), 1)
self.assertEqual(res[0]['name'], 'cell1')
def create_cell(self,
name,
project_id,
cloud_id,
region_id,
variables=None):
cell = dbapi.cells_create(
self.context,
{
'name': name,
'project_id': project_id,
'cloud_id': cloud_id,
'region_id': region_id,
'variables': variables or {}
},
)
return cell.id
def test_duplicate_cell_create_raises_409(self):
dbapi.cells_create(self.context, cell1)
self.assertRaises(exceptions.DuplicateCell, dbapi.cells_create,
self.context, cell1)
def test_cell_get_by_name(self):
dbapi.cells_create(self.context, cell1)
res = dbapi.cells_get_by_name(self.context, cell1['region_id'],
cell1['name'])
self.assertEqual(res.name, 'cell1')
def test_cells_create(self):
try:
dbapi.cells_create(self.context, cell1)
except Exception:
self.fail("Cell create raised unexpected exception")
def test_cells_get_all(self):
dbapi.cells_create(self.context, cell1)
res = dbapi.cells_get_all(self.context, cell1['region_id'])
self.assertEqual(len(res), 1)
self.assertEqual(res[0]['name'], 'cell1')
def test_cells_create(self):
try:
dbapi.cells_create(self.context, cell1)
except Exception:
self.fail("Cell create raised unexpected exception")
def test_cell_get_by_name(self):
dbapi.cells_create(self.context, cell1)
res = dbapi.cells_get_by_name(self.context, cell1['region_id'],
cell1['name'])
self.assertEqual(res.name, 'cell1')
def test_cells_get_all(self):
dbapi.cells_create(self.context, cell1)
res = dbapi.cells_get_all(self.context, cell1['region_id'])
self.assertEqual(len(res), 1)
self.assertEqual(res[0]['name'], 'cell1')
