From c3ed12d8d486da9e8231cc259d60209b7c4f28b5 Mon Sep 17 00:00:00 2001 From: mr-tz Date: Mon, 20 Mar 2023 17:46:36 +0100 Subject: [PATCH] add helper function --- tests/fixtures.py | 46 ++++++++++++---------------------------------- 1 file changed, 12 insertions(+), 34 deletions(-) diff --git a/tests/fixtures.py b/tests/fixtures.py index 6a94370a..f549a734 100644 --- a/tests/fixtures.py +++ b/tests/fixtures.py @@ -1103,61 +1103,39 @@ def _039a6_dotnetfile_extractor(): return get_dnfile_extractor(get_data_path_by_name("_039a6")) -@pytest.fixture -def pma0101_rd(): - path = os.path.join(CD, "data", "Practical Malware Analysis Lab 01-01.dll_.json") +def get_result_doc(path): with open(path, "rb") as f: buf = f.read() - src = buf.decode("utf-8") return capa.render.result_document.ResultDocument.parse_raw(src) +@pytest.fixture +def pma0101_rd(): + # TODO move to rd subdir + return get_result_doc(os.path.join(CD, "data", "Practical Malware Analysis Lab 01-01.dll_.json")) + + @pytest.fixture def dotnet_1c444e_rd(): - path = os.path.join(CD, "data", "dotnet", "1c444ebeba24dcba8628b7dfe5fec7c6.exe_.json") - with open(path, "rb") as f: - buf = f.read() - - src = buf.decode("utf-8") - return capa.render.result_document.ResultDocument.parse_raw(src) + return get_result_doc(os.path.join(CD, "data", "dotnet", "1c444ebeba24dcba8628b7dfe5fec7c6.exe_.json")) @pytest.fixture def a3f3bbc_rd(): - path = os.path.join(CD, "data", "3f3bbcf8fd90bdcdcdc5494314ed4225.exe_.json") - with open(path, "rb") as f: - buf = f.read() - - src = buf.decode("utf-8") - return capa.render.result_document.ResultDocument.parse_raw(src) + return get_result_doc(os.path.join(CD, "data", "3f3bbcf8fd90bdcdcdc5494314ed4225.exe_.json")) @pytest.fixture def al_khaserx86_rd(): - path = os.path.join(CD, "data", "al-khaser_x86.exe_.json") - with open(path, "rb") as f: - buf = f.read() - - src = buf.decode("utf-8") - return capa.render.result_document.ResultDocument.parse_raw(src) + return get_result_doc(os.path.join(CD, "data", "al-khaser_x86.exe_.json")) @pytest.fixture def al_khaserx64_rd(): - path = os.path.join(CD, "data", "al-khaser_x64.exe_.json") - with open(path, "rb") as f: - buf = f.read() - - src = buf.decode("utf-8") - return capa.render.result_document.ResultDocument.parse_raw(src) + return get_result_doc(os.path.join(CD, "data", "al-khaser_x64.exe_.json")) @pytest.fixture def a076114_rd(): - path = os.path.join(CD, "data", "0761142efbda6c4b1e801223de723578.dll_.json") - with open(path, "rb") as f: - buf = f.read() - - src = buf.decode("utf-8") - return capa.render.result_document.ResultDocument.parse_raw(src) + return get_result_doc(os.path.join(CD, "data", "0761142efbda6c4b1e801223de723578.dll_.json"))