x/terracotta
Code Issues Proposed changes

Fixed a DB test

Browse Source
This commit is contained in:
Anton Beloglazov 2012-12-18 17:49:33 +11:00
parent b7aa18766b
commit 668d7b374e
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tests/test_db.py
View File

@ -161,7 +161,7 @@ class Db(TestCase):
db.insert_host_cpu_mhz(hostname, value) db.insert_host_cpu_mhz(hostname, value)
if data: if data:
res[hostname] = data[-1] res[hostname] = data[-1]
else: else:
res[hostname] = 0 res[hostname] = 0
assert db.select_last_cpu_mhz_for_hosts() == res assert db.select_last_cpu_mhz_for_hosts() == res
@ -200,7 +200,7 @@ class Db(TestCase):
hostname='host2', hostname='host2',
cpu_mhz=1, cpu_mhz=1,
cpu_cores=1, cpu_cores=1,
ram=1).inserted_primary_key[0] ram=1).inserted_primary_key[0]
assert db.select_host_id('host1') == host1_id assert db.select_host_id('host1') == host1_id
assert db.select_host_id('host2') == host2_id assert db.select_host_id('host2') == host2_id
@ -256,13 +256,13 @@ class Db(TestCase):
db.insert_host_states({'host1': 1, 'host2': 1}) db.insert_host_states({'host1': 1, 'host2': 1})
result = db.host_states.select().execute().fetchall() result = db.host_states.select().execute().fetchall()
host1 = [x[3] for x in sorted(filter( host1 = [x[3] for x in sorted(filter(
lambda x: x[1] == hosts['host1'], lambda x: x[1] == hosts['host1'],
result), key=lambda x: x[0])] result), key=lambda x: x[0])]
self.assertEqual(host1, [0, 0, 1]) self.assertEqual(host1, [0, 0, 1])
host2 = [x[3] for x in sorted(filter( host2 = [x[3] for x in sorted(filter(
lambda x: x[1] == hosts['host2'], lambda x: x[1] == hosts['host2'],
result), key=lambda x: x[0])] result), key=lambda x: x[0])]
self.assertEqual(host2, [1, 0, 1]) self.assertEqual(host2, [1, 0, 1])
@qc(10) @qc(10)
def select_host_states( def select_host_states(
@ -313,6 +313,7 @@ class Db(TestCase):
min_length=0, max_length=3 min_length=0, max_length=3
) )
): ):
hosts = {'1ab': [0], '3222': [0, 0, 1, 1, 1, 1, 0, 0], 'b222b': [0, 0, 1, 1, 1, 0, 1]}
db = db_utils.init_db('sqlite:///:memory:') db = db_utils.init_db('sqlite:///:memory:')
res = [] res = []
for host, data in hosts.items(): for host, data in hosts.items():
@ -321,7 +322,7 @@ class Db(TestCase):
db.insert_host_states({host: state}) db.insert_host_states({host: state})
if data and data[-1] == 0: if data and data[-1] == 0:
res.append(host) res.append(host)
assert db.select_inactive_hosts() == res assert set(db.select_inactive_hosts()) == set(res)
def test_insert_host_overload(self): def test_insert_host_overload(self):
db = db_utils.init_db('sqlite:///:memory:') db = db_utils.init_db('sqlite:///:memory:')
@ -334,13 +335,13 @@ class Db(TestCase):
db.insert_host_overload('host2', True) db.insert_host_overload('host2', True)
result = db.host_overload.select().execute().fetchall() result = db.host_overload.select().execute().fetchall()
host1 = [x[3] for x in sorted(filter( host1 = [x[3] for x in sorted(filter(
lambda x: x[1] == hosts['host1'], lambda x: x[1] == hosts['host1'],
result), key=lambda x: x[0])] result), key=lambda x: x[0])]
self.assertEqual(host1, [1, 0]) self.assertEqual(host1, [1, 0])
host2 = [x[3] for x in sorted(filter( host2 = [x[3] for x in sorted(filter(
lambda x: x[1] == hosts['host2'], lambda x: x[1] == hosts['host2'],
result), key=lambda x: x[0])] result), key=lambda x: x[0])]
self.assertEqual(host2, [0, 1]) self.assertEqual(host2, [0, 1])
@qc(1) @qc(1)
def insert_select(): def insert_select():