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))
