def setUp(self):
super(MultiRepackTest, self).setUp()
self.pool_obj = mock.MagicMock()
self.pool_patcher = mock.patch.object(multiprocessing,
"Pool",
return_value=self.pool_obj)
self.mock_pool = self.pool_patcher.start()
config_dir = test_lib.TempDirPath()
self.label1_config = os.path.join(config_dir, "label1.yaml")
self.label2_config = os.path.join(config_dir, "label2.yaml")
open(self.label1_config, mode="wb").write("Client.labels: [label1]")
open(self.label2_config, mode="wb").write("Client.labels: [label2]")
self.template_dir = test_lib.TempDirPath()
self.deb_template = os.path.join(self.template_dir,
"grr_3.1.0.2_amd64.deb.zip")
self.exe_template = os.path.join(self.template_dir,
"GRR_3.1.0.2_i386.exe.zip")
self.xar_template = os.path.join(self.template_dir,
"grr_3.1.0.2_amd64.xar.zip")
open(self.deb_template, mode="wb").write("linux")
open(self.exe_template, mode="wb").write("windows")
open(self.xar_template, mode="wb").write("darwin")
self.output_dir = test_lib.TempDirPath()
def setUpClass(cls):
super(GRRHTTPServerTest, cls).setUpClass()
cls.config_overrider = test_lib.ConfigOverrider({
"Rekall.profile_server":
rekall_test_lib.TestRekallRepositoryProfileServer.__name__,
"FileUploadFileStore.root_dir":
test_lib.TempDirPath()
})
cls.config_overrider.Start()
# Frontend must be initialized to register all the stats counters.
frontend_lib.FrontendInit().RunOnce()
# Bring up a local server for testing.
port = portpicker.PickUnusedPort()
ip = utils.ResolveHostnameToIP("localhost", port)
cls.httpd = frontend.GRRHTTPServer((ip, port),
frontend.GRRHTTPServerHandler)
if ipaddr.IPAddress(ip).version == 6:
cls.address_family = socket.AF_INET6
cls.base_url = "http://[%s]:%d/" % (ip, port)
else:
cls.address_family = socket.AF_INET
cls.base_url = "http://%s:%d/" % (ip, port)
cls.httpd_thread = threading.Thread(target=cls.httpd.serve_forever)
cls.httpd_thread.daemon = True
cls.httpd_thread.start()
def InitGRRWithTestSources(artifacts_data):
artifact_registry.REGISTRY.ClearSources()
artifact_registry.REGISTRY.ClearRegistry()
artifacts_temp_dir = test_lib.TempDirPath()
with open(os.path.join(artifacts_temp_dir, "test_artifacts.yaml"), "w") as fd:
fd.write(artifacts_data)
artifact_registry.REGISTRY.AddDirSources([artifacts_temp_dir])
def CleanUp():
shutil.rmtree(artifacts_temp_dir)
return CleanUp
def setUp(self):
self.temp_dir = test_lib.TempDirPath()
def setUp(self):
super(DirHierarchyTestMixin, self).setUp()
self.tempdir = test_lib.TempDirPath()
