External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-01-15 03:40:42 -03:30
Code Issues Packages Projects Releases Wiki Activity

added host fact query in time

Browse Source
This commit is contained in:
Chris Meyers 2015-04-06 09:19:23 -04:00
parent a769af0e83
commit d93a2ec9d1
2 changed files with 68 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
awx/main/models/fact.py
View File

@ -11,6 +11,13 @@ class FactHost(Document):
]
}
@staticmethod
def get_host_id(hostname):
host = FactHost.objects.get(hostname=hostname)
if host:
return host.id
return None
class Fact(DynamicDocument):
timestamp = DateTimeField(required=True)
host = ReferenceField(FactHost, required=True)
@ -34,6 +41,28 @@ class Fact(DynamicDocument):
version_obj.save()
return (fact_obj, version_obj)
@staticmethod
def get_version(hostname, timestamp, module=None):
try:
host = FactHost.objects.get(hostname=hostname)
except FactHost.DoesNotExist:
return None
kv = {
'host' : host.id,
'timestamp__lte': timestamp
}
if module:
kv['module'] = module
try:
facts = Fact.objects.filter(**kv)
if not facts:
return None
return facts[0]
except Fact.DoesNotExist:
return None
class FactVersion(Document):
timestamp = DateTimeField(required=True)
host = ReferenceField(FactHost, required=True)
@ -46,4 +75,4 @@ class FactVersion(Document):
'-timestamp',
'module'
]
}
}

39
awx/main/tests/models/fact.py
View File

@ -3,6 +3,7 @@
# Python
from datetime import datetime
from copy import deepcopy
# Django
@ -10,7 +11,7 @@ from datetime import datetime
from awx.main.models.fact import * # noqa
from awx.main.tests.base import BaseTest, MongoDBRequired
__all__ = ['FactHostTest', 'FactTest']
__all__ = ['FactHostTest', 'FactTest', 'FactGetVersionTest']
TEST_FACT_DATA = {
'hostname': 'hostname1',
@ -58,6 +59,12 @@ class FactHostTest(BaseTest, MongoDBRequired):
self.assertIsNotNone(host, "Host added but not found")
self.assertEqual(TEST_FACT_DATA['hostname'], host.hostname, "Gotten record hostname does not match expected hostname")
# Ensure an error is raised for .get() that doesn't match a record.
def test_get_host_id_no_result(self):
host = FactHost(hostname=TEST_FACT_DATA['hostname'])
host.save()
self.assertRaises(FactHost.DoesNotExist, FactHost.objects.get, hostname='doesnotexist')
class FactTest(BaseTest, MongoDBRequired):
def setUp(self):
@ -84,4 +91,34 @@ class FactTest(BaseTest, MongoDBRequired):
self.assertEqual(v.fact.id, f_obj.id)
self.assertEqual(v.fact.module, TEST_FACT_DATA['add_fact_data']['module'])
class FactGetVersionTest(BaseTest, MongoDBRequired):
def setUp(self):
super(FactGetVersionTest, self).setUp()
TEST_FACT_DATA['add_fact_data']['host'] = FactHost(hostname=TEST_FACT_DATA['hostname']).save()
self.t1 = datetime.now().replace(second=1, microsecond=0)
self.t2 = datetime.now().replace(second=2, microsecond=0)
data = deepcopy(TEST_FACT_DATA)
data['add_fact_data']['timestamp'] = self.t1
(self.f1, self.v1) = Fact.add_fact(**data['add_fact_data'])
data = deepcopy(TEST_FACT_DATA)
data['add_fact_data']['timestamp'] = self.t2
(self.f2, self.v2) = Fact.add_fact(**data['add_fact_data'])
def test_get_version_exact_timestamp(self):
fact = Fact.get_version(hostname=TEST_FACT_DATA['hostname'], timestamp=self.t1, module=TEST_FACT_DATA['add_fact_data']['module'])
self.assertIsNotNone(fact, "Set of Facts not found")
self.assertEqual(self.f1.id, fact.id)
self.assertEqual(self.f1.fact, fact.fact)
def test_get_version_lte_timestamp(self):
t3 = datetime.now().replace(second=3, microsecond=0)
fact = Fact.get_version(hostname=TEST_FACT_DATA['hostname'], timestamp=t3, module=TEST_FACT_DATA['add_fact_data']['module'])
self.assertEqual(self.f1.id, fact.id)
self.assertEqual(self.f1.fact, fact.fact)
def test_get_version_none(self):
t3 = deepcopy(self.t1).replace(second=0)
fact = Fact.get_version(hostname=TEST_FACT_DATA['hostname'], timestamp=t3, module=TEST_FACT_DATA['add_fact_data']['module'])
self.assertIsNone(fact)