x/terracotta
Code Issues Proposed changes

Added insert_host_states

Browse Source
This commit is contained in:
Anton Beloglazov 2012-10-15 15:50:08 +11:00
parent 66536cf938
commit ebd4a2ad18
2 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
neat/db.py
View File

@ -214,4 +214,9 @@ class Database(object):
:param hosts: A dict of hostnames to states (0, 1).
:type hosts: dict(str: int)
"""
pass
host_ids = self.select_host_ids()
to_insert = [{'host_id': host_ids[k],
'state': v}
for k, v in hosts.items()]
self.connection.execute(
self.host_states.insert(), to_insert)

17
tests/test_db.py
View File

@ -168,3 +168,20 @@ class Db(TestCase):
db.cleanup_vm_resource_usage(time.replace(second=5))
assert db.select_cpu_mhz_for_vm(uuid, 100) == range(5, 10)
def test_insert_host_states(self):
db = db_utils.init_db('sqlite:///:memory:')
hosts = {}
hosts['host1'] = db.update_host('host1', 1, 1, 1)
hosts['host2'] = db.update_host('host2', 1, 1, 1)
db.insert_host_states({'host1': 0, 'host2': 1})
db.insert_host_states({'host1': 0, 'host2': 0})
db.insert_host_states({'host1': 1, 'host2': 1})
result = db.host_states.select().execute().fetchall()
host1 = [x[3] for x in sorted(filter(
lambda x: x[1] == hosts['host1'],
result), key=lambda x: x[0])]
self.assertEqual(host1, [0, 0, 1])
host2 = [x[3] for x in sorted(filter(
lambda x: x[1] == hosts['host2'],
result), key=lambda x: x[0])]
self.assertEqual(host1, [0, 0, 1])