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)