def _init_gecko_profiling(self, test):
LOG.info("initializing gecko profiler")
upload_dir = os.getenv("MOZ_UPLOAD_DIR")
if not upload_dir:
LOG.critical("Profiling ignored because MOZ_UPLOAD_DIR was not set")
else:
self.gecko_profiler = GeckoProfile(upload_dir, self.config, test)
def test_browsertime_profiling():
result_dir = tempfile.mkdtemp()
# untar geckoProfile.tar
with tarfile.open(os.path.join(here, "geckoProfileTest.tar")) as f:
f.extractall(path=result_dir)
# Makes sure we can run the profile process against a browsertime-generated
# profile (geckoProfile-1.json in this test dir)
upload_dir = tempfile.mkdtemp()
symbols_path = tempfile.mkdtemp()
raptor_config = {
"symbols_path": symbols_path,
"browsertime": True,
"browsertime_result_dir": os.path.join(result_dir, "amazon"),
}
test_config = {"name": "tp6"}
try:
profile = GeckoProfile(upload_dir, raptor_config, test_config)
profile.symbolicate()
profile.clean()
arcname = os.environ["RAPTOR_LATEST_GECKO_PROFILE_ARCHIVE"]
assert os.stat(arcname).st_size > 1000000, "We got a 1mb+ zip"
except Exception:
assert False, "Symbolication failed!"
raise
finally:
shutil.rmtree(upload_dir)
shutil.rmtree(symbols_path)
shutil.rmtree(result_dir)