def setUp(self): """Make sure things are initialized.""" super(TestArtifactCollectors, self).setUp() self._patcher = artifact_test_lib.PatchDefaultArtifactRegistry() self._patcher.start() test_artifacts_file = os.path.join(config.CONFIG["Test.data_dir"], "artifacts", "test_artifacts.json") artifact_registry.REGISTRY.AddFileSource(test_artifacts_file) self.fakeartifact = artifact_registry.REGISTRY.GetArtifact( "FakeArtifact") self.fakeartifact2 = artifact_registry.REGISTRY.GetArtifact( "FakeArtifact2") self.output_count = 0 self.client_id = self.SetupClient(0) with aff4.FACTORY.Open(self.client_id, token=self.token, mode="rw") as fd: fd.Set(fd.Schema.SYSTEM("Linux")) kb = fd.Schema.KNOWLEDGE_BASE() artifact.SetCoreGRRKnowledgeBaseValues(kb, fd) fd.Set(kb)
def setUp(self): super(GRRSeleniumTest, self).setUp() # Used by InstallACLChecks/UninstallACLChecks self.config_override = None self.token.username = "******" webauth.WEBAUTH_MANAGER.SetUserName(self.token.username) # Make the user use the advanced gui so we can test it. with aff4.FACTORY.Create( aff4.ROOT_URN.Add("users/%s" % self.token.username), aff4_type=users.GRRUser, mode="w", token=self.token) as user_fd: user_fd.Set(user_fd.Schema.GUI_SETTINGS(mode="ADVANCED")) if data_store.RelationalDBReadEnabled(): data_store.REL_DB.WriteGRRUser( self.token.username, ui_mode=users.GUISettings.UIMode.ADVANCED) self._artifact_patcher = artifact_test_lib.PatchDefaultArtifactRegistry() self._artifact_patcher.start() self.InstallACLChecks()
def setUp(self): super(TestArtifactCollectorsInteractions, self).setUp() self._patcher = artifact_test_lib.PatchDefaultArtifactRegistry() self._patcher.start() test_artifacts_file = os.path.join(config.CONFIG["Test.data_dir"], "artifacts", "test_artifacts.json") artifact_registry.REGISTRY.AddFileSource(test_artifacts_file)
def setUp(self): super().setUp() patcher = artifact_test_lib.PatchDefaultArtifactRegistry() patcher.start() self.addCleanup(patcher.stop) test_artifacts_file = os.path.join(config.CONFIG["Test.data_dir"], "artifacts", "test_artifacts.json") artifact_registry.REGISTRY.AddFileSource(test_artifacts_file)
def setUp(self): super(ClientArtifactCollectorFlowTest, self).setUp() self._patcher = artifact_test_lib.PatchDefaultArtifactRegistry() self._patcher.start() test_artifacts_file = os.path.join(config.CONFIG["Test.data_dir"], "artifacts", "test_artifacts.json") artifact_registry.REGISTRY.AddFileSource(test_artifacts_file) self.client_id = self.SetupClient(0)
def setUp(self): """Make sure things are initialized.""" super(ArtifactTest, self).setUp() # Common group of mocks used by lots of tests. self.client_mock = action_mocks.ActionMock( searching.Find, searching.Grep, standard.HashBuffer, standard.HashFile, standard.ListDirectory, standard.GetFileStat, standard.TransferBuffer, ) patcher = artifact_test_lib.PatchDefaultArtifactRegistry() patcher.start() self.addCleanup(patcher.stop)
def setUp(self): """Make sure things are initialized.""" super(ArtifactCollectorsTestMixin, self).setUp() self._patcher = artifact_test_lib.PatchDefaultArtifactRegistry() self._patcher.start() test_artifacts_file = os.path.join(config.CONFIG["Test.data_dir"], "artifacts", "test_artifacts.json") artifact_registry.REGISTRY.AddFileSource(test_artifacts_file) self.fakeartifact = artifact_registry.REGISTRY.GetArtifact("FakeArtifact") self.fakeartifact2 = artifact_registry.REGISTRY.GetArtifact("FakeArtifact2") self.output_count = 0 self.client_id = self.SetupClient(0, system="Linux")
def setUp(self): """Make sure things are initialized.""" super(ArtifactCollectorsTestMixin, self).setUp() patcher = artifact_test_lib.PatchDefaultArtifactRegistry() patcher.start() self.addCleanup(patcher.stop) artifact_registry.REGISTRY.ClearSources() artifact_registry.REGISTRY.ClearRegistry() test_artifacts_file = os.path.join(config.CONFIG["Test.data_dir"], "artifacts", "test_artifacts.json") artifact_registry.REGISTRY.AddFileSource(test_artifacts_file) self.fakeartifact = artifact_registry.REGISTRY.GetArtifact("FakeArtifact") self.fakeartifact2 = artifact_registry.REGISTRY.GetArtifact("FakeArtifact2") self.output_count = 0
def setUp(self): """Make sure things are initialized.""" super(ArtifactTest, self).setUp() # Common group of mocks used by lots of tests. self.client_mock = action_mocks.ActionMock( file_fingerprint.FingerprintFile, searching.Find, searching.Grep, server_stubs.WmiQuery, standard.HashBuffer, standard.HashFile, standard.ListDirectory, standard.StatFile, standard.TransferBuffer, ) self._artifact_patcher = artifact_test_lib.PatchDefaultArtifactRegistry( ) self._artifact_patcher.start()
def setUp(self): super(GRRSeleniumTest, self).setUp() # Used by InstallACLChecks/UninstallACLChecks self.config_override = None # Used by CheckHttpErrors self.ignore_http_errors = False self.token.username = "******" webauth.WEBAUTH_MANAGER.SetUserName(self.token.username) # Make the user use the advanced gui so we can test it. with aff4.FACTORY.Create( aff4.ROOT_URN.Add("users/%s" % self.token.username), aff4_type=users.GRRUser, mode="w", token=self.token) as user_fd: user_fd.Set(user_fd.Schema.GUI_SETTINGS(mode="ADVANCED")) if data_store.RelationalDBReadEnabled(): data_store.REL_DB.WriteGRRUser( self.token.username, ui_mode=users.GUISettings.UIMode.ADVANCED) self._artifact_patcher = artifact_test_lib.PatchDefaultArtifactRegistry() self._artifact_patcher.start() self.InstallACLChecks() if flags.FLAGS.use_headless_chrome: params = { "cmd": "Page.setDownloadBehavior", "params": { "behavior": "allow", "downloadPath": self.temp_dir } } result = self.driver.execute("send_command", params) if result["status"] != 0: raise RuntimeError("can't set Page.setDownloadBehavior: %s" % result)