class BaseContextTest(TestCase):
"""Base class for context.
"""
def setUp(self):
self.context = Context(data_scope='test_context')
self.context.remove()
def tearDown(self):
self.context.remove()
class DowntimeProcessingTest(TestCase):
def setUp(self):
self.downtimes = PBehaviorManager(data_scope='test_pbehavior')
self.events = Event(data_scope='test_events')
self.context = Context(data_scope='test_context')
def tearDown(self):
self.downtimes.remove()
self.events.remove()
self.context.remove()
class CTXContextRegistry(CTXPropRegistry):
"""In charge of contextual context properties.
"""
__datatype__ = 'context' #: default datatype name
def __init__(self, *args, **kwargs):
super(CTXContextRegistry, self).__init__(*args, **kwargs)
self.manager = Context()
def _get_documents(self, ids, query):
query[Context.DATA_ID] = ids
return self.manager.find(_filter=query)
def _get(self, ids, query, *args, **kwargs):
return self._get_documents(ids=ids, query=query)
def _delete(self, ids, query, *args, **kwargs):
docs = self._get_documents(ids=ids, query=query)
ids = [doc[Context.DATA_ID] for doc in docs]
self.manager.remove(ids=ids)
return docs
def ids(self, query=None):
result = set()
elts = self.manager.find(_filter=query)
for elt in elts:
result.add(elt[Context.DATA_ID])
return list(result)
class TopologyGraphTest(TestCase):
"""Test topology element.
"""
def setUp(self):
self.context = Context(data_scope='test')
self.manager = TopologyManager(data_scope='test')
def tearDown(self):
self.context.remove()
self.manager.del_elts()
def test_save(self):
"""Test if an entity exists after saving a topology.
"""
id = 'test'
topology = Topology(id=id)
topology.save(manager=self.manager, context=self.context)
topology = self.context.get(_type=topology.type, names=id)
self.assertEqual(topology[Context.NAME], id)
def test_delete(self):
"""Test if topology nodes exist after deleting a topology.
"""
topology = Topology()
node = TopoNode()
topology.add_elts(node)
topology.save(manager=self.manager)
node = self.manager.get_elts(node.id)
self.assertIsNotNone(node)
topology.delete(manager=self.manager)
node = self.manager.get_elts(node.id)
self.assertIsNone(node)
