def testRandom(self): db = mem_db.InMemoryDB() hunt_id = db_test_utils.InitializeHunt(db) hunt_obj = db.ReadHuntObject(hunt_id) self.assertIsNotNone(hunt_obj) self.assertIsNotNone(db.ReadGRRUser(hunt_obj.creator))
def testSupplied(self): db = mem_db.InMemoryDB() cron_job_id = db_test_utils.InitializeCronJob(db, cron_job_id="QUUX1337") self.assertEqual(cron_job_id, "QUUX1337") self.assertIsNotNone(db.ReadCronJob(cron_job_id))
def testRandom(self): db = mem_db.InMemoryDB() client_id = db_test_utils.InitializeClient(db) flow_id = db_test_utils.InitializeFlow(db, client_id) flow_obj = db.ReadFlowObject(client_id=client_id, flow_id=flow_id) self.assertIsNotNone(flow_obj)
def testKwargs(self): db = mem_db.InMemoryDB() username = db_test_utils.InitializeUser(db) client_id = db_test_utils.InitializeClient(db) flow_id = db_test_utils.InitializeFlow(db, client_id, creator=username) flow_obj = db.ReadFlowObject(client_id=client_id, flow_id=flow_id) self.assertEqual(flow_obj.creator, username)
def testSupplied(self): db = mem_db.InMemoryDB() client_id = db_test_utils.InitializeClient(db) flow_id = db_test_utils.InitializeFlow(db, client_id, flow_id="ABCDEF42") self.assertEqual(flow_id, "ABCDEF42") flow_obj = db.ReadFlowObject(client_id=client_id, flow_id=flow_id) self.assertIsNotNone(flow_obj)
def setUp(self): super(ArtifactMigrationTest, self).setUp() self._db_patcher = mock.patch.object( data_store, "REL_DB", db.DatabaseValidationWrapper(mem.InMemoryDB())) self._db_patcher.start() self._artifact_patcher = mock.patch.object( artifact_registry, "REGISTRY", artifact_registry.ArtifactRegistry()) self._artifact_patcher.start()
def SharedMemoryDBServer(port): """Initializes SharedMemoryDB server object.""" server = TCPServerV6(("::", port), SharedMemoryDBTCPHandler) server.port = port server.lock = threading.RLock() server.flow_processing_requests = [] server.num_fpr_handlers = 0 server.messages = [] server.num_msg_handlers = 0 server.delegate = mem.InMemoryDB() def Reset(): server.delegate = mem.InMemoryDB() server.Reset = Reset # pylint: disable=invalid-name return server
def CreateDatabase(self): return mem.InMemoryDB(), None
def Wrapper(*args, **kwargs): db = abstract_db.DatabaseValidationWrapper(mem.InMemoryDB()) with mock.patch.object(data_store, "REL_DB", db): func(*(args + (db,)), **kwargs)
def Reset(): server.delegate = mem.InMemoryDB()
def testRandom(self): db = mem_db.InMemoryDB() cron_job_id = db_test_utils.InitializeCronJob(db) self.assertIsNotNone(db.ReadCronJob(cron_job_id))
def testSupplied(self): db = mem_db.InMemoryDB() hunt_id = db_test_utils.InitializeHunt(db, hunt_id="ABCDEF42") self.assertEqual(hunt_id, "ABCDEF42") self.assertIsNotNone(db.ReadHuntObject(hunt_id))
def testSupplied(self): db = mem_db.InMemoryDB() username = db_test_utils.InitializeUser(db, username="******") self.assertEqual(username, "foobar") self.assertIsNotNone(db.ReadGRRUser(username))
def testRandom(self): db = mem_db.InMemoryDB() username = db_test_utils.InitializeUser(db) self.assertIsNotNone(db.ReadGRRUser(username))
def testSupplied(self): db = mem_db.InMemoryDB() client_id = db_test_utils.InitializeClient(db, "C.012345678ABCDEFAA") self.assertEqual(client_id, "C.012345678ABCDEFAA") self.assertIsNotNone(db.ReadClientMetadata(client_id))
def testRandom(self): db = mem_db.InMemoryDB() client_id = db_test_utils.InitializeClient(db) self.assertIsNotNone(db.ReadClientMetadata(client_id))