def test_enterpriseids_to_objects(self): expected = {} for x in range(5): person = self.factory.makePerson() upload = self.factory.makePackageUpload() expected['lp-development:Person:%d' % person.id] = person expected['lp-development:PackageUpload:%d' % upload.id] = upload IStore(expected.values()[0].__class__).invalidate() with StormStatementRecorder() as recorder: objects = enterpriseids_to_objects(expected.keys()) self.assertThat(recorder, HasQueryCount(Equals(2))) self.assertContentEqual(expected.keys(), objects.keys()) self.assertContentEqual(expected.values(), objects.values())
def test_enterpriseids_to_objects(self): expected = {} for x in range(5): person = self.factory.makePerson() upload = self.factory.makePackageUpload() expected["lp-development:Person:%d" % person.id] = person expected["lp-development:PackageUpload:%d" % upload.id] = upload IStore(expected.values()[0].__class__).invalidate() with StormStatementRecorder() as recorder: objects = enterpriseids_to_objects(expected.keys()) self.assertThat(recorder, HasQueryCount(Equals(2))) self.assertContentEqual(expected.keys(), objects.keys()) self.assertContentEqual(expected.values(), objects.values())
def receive(self, obj=None, operation=None, actorobj=None, limit=None): if obj: obj = self._convert_to_enterpriseid(obj) if actorobj: actorobj = self._convert_to_enterpriseid(actorobj) action = self.__get_timeline_action( "receive", obj, operation, actorobj) try: logs = super(AuditorClient, self).receive( obj, operation, actorobj, limit) finally: action.finish() # Process the actors and objects back from enterprise ids. eids = set() for entry in logs['log-entries']: eids |= set([entry['actor'], entry['object']]) map_eids_to_obj = enterpriseids_to_objects(eids) for entry in logs['log-entries']: entry['actor'] = map_eids_to_obj.get(entry['actor'], None) entry['object'] = map_eids_to_obj.get(entry['object'], None) return logs['log-entries']
def receive(self, obj=None, operation=None, actorobj=None, limit=None): if obj: obj = self._convert_to_enterpriseid(obj) if actorobj: actorobj = self._convert_to_enterpriseid(actorobj) action = self.__get_timeline_action("receive", obj, operation, actorobj) try: logs = super(AuditorClient, self).receive(obj, operation, actorobj, limit) finally: action.finish() # Process the actors and objects back from enterprise ids. eids = set() for entry in logs['log-entries']: eids |= set([entry['actor'], entry['object']]) map_eids_to_obj = enterpriseids_to_objects(eids) for entry in logs['log-entries']: entry['actor'] = map_eids_to_obj.get(entry['actor'], None) entry['object'] = map_eids_to_obj.get(entry['object'], None) return logs['log-entries']
def test_enterpriseids_to_objects_non_existent_id(self): objects = enterpriseids_to_objects( ['lp-development:PackageUpload:10000']) self.assertEqual({}, objects)
def test_enterpriseids_to_objects_non_existent_id(self): objects = enterpriseids_to_objects(["lp-development:PackageUpload:10000"]) self.assertEqual({}, objects)