class TestSQLDB(NodeAssignmentTests, TestCase):
_SQLURI = os.environ.get('WIMMS_SQLURI', 'sqlite:////tmp/wimms.' + TEMP_ID)
def setUp(self):
self.backend = SQLNodeAssignment(self._SQLURI, create_tables=True)
super(TestSQLDB, self).setUp()
def tearDown(self):
super(TestSQLDB, self).tearDown()
if self.backend._engine.driver == 'pysqlite':
filename = self.backend.sqluri.split('sqlite://')[-1]
if os.path.exists(filename):
os.remove(filename)
else:
self.backend._safe_execute('drop table services;')
self.backend._safe_execute('drop table nodes;')
self.backend._safe_execute('drop table users;')
def test_default_node_available_capacity(self):
node = "https://phx13"
self.backend.add_node("sync-1.0", node, capacity=100)
available = int(math.ceil(self.backend.capacity_release_rate * 100))
query = "SELECT * FROM nodes WHERE node=:node"
res = self.backend._safe_execute(query, node=node)
row = res.fetchone()
res.close()
self.assertEqual(row["available"], available)
class TestSQLDB(NodeAssignmentTests, unittest.TestCase):
_SQLURI = os.environ.get('MOZSVC_SQLURI', DEFAULT_SQLURI)
def setUp(self):
self.backend = SQLNodeAssignment(self._SQLURI, create_tables=True)
super(TestSQLDB, self).setUp()
def tearDown(self):
super(TestSQLDB, self).tearDown()
if self.backend._engine.driver == 'pysqlite':
filename = self.backend.sqluri.split('sqlite://')[-1]
if os.path.exists(filename):
os.remove(filename)
else:
self.backend._safe_execute('drop table services;')
self.backend._safe_execute('drop table nodes;')
self.backend._safe_execute('drop table users;')
def test_default_node_available_capacity(self):
node = "https://phx13"
self.backend.add_node("sync-1.0", node, capacity=100)
available = int(math.ceil(self.backend.capacity_release_rate * 100))
query = sqltext("SELECT * FROM nodes WHERE node=:node")
res = self.backend._safe_execute(query, node=node)
row = res.fetchone()
res.close()
self.assertEqual(row["available"], available)
class TestSQLDB(NodeAssignmentTests, TestCase):
_SQLURI = os.environ.get('WIMMS_SQLURI', 'sqlite:////tmp/wimms.' + TEMP_ID)
def setUp(self):
self.backend = SQLNodeAssignment(self._SQLURI, create_tables=True)
super(TestSQLDB, self).setUp()
def tearDown(self):
super(TestSQLDB, self).tearDown()
if self.backend._engine.driver == 'pysqlite':
filename = self.backend.sqluri.split('sqlite://')[-1]
if os.path.exists(filename):
os.remove(filename)
else:
self.backend._safe_execute('drop table services;')
self.backend._safe_execute('drop table nodes;')
self.backend._safe_execute('drop table users;')