diff --git a/ofrak_core/requirements-test.txt b/ofrak_core/requirements-test.txt index 71aba2382..ca6f258d3 100644 --- a/ofrak_core/requirements-test.txt +++ b/ofrak_core/requirements-test.txt @@ -15,3 +15,4 @@ pytest-cov pytest-xdist requests fun-coverage==0.2.0 +importlib-resources # Needed because of https://github.com/redballoonsecurity/ofrak/issues/398 diff --git a/ofrak_core/setup.py b/ofrak_core/setup.py index dbe1661b0..8cba6744f 100644 --- a/ofrak_core/setup.py +++ b/ofrak_core/setup.py @@ -83,12 +83,7 @@ def read_requirements(requirements_path): + read_requirements("requirements.txt"), extras_require={ "docs": read_requirements("requirements-docs.txt"), - "test": [ - "importlib-resources", # Needed because of https://github.com/redballoonsecurity/ofrak/issues/398 - "ofrak_angr~=1.0", - "ofrak_capstone~=1.0", - ] - + read_requirements("requirements-test.txt"), + "test": read_requirements("requirements-test.txt"), "non-pypi": read_requirements("requirements-non-pypi.txt"), }, author="Red Balloon Security", diff --git a/ofrak_core/test_ofrak/components/test_patch_from_source.py b/ofrak_core/test_ofrak/components/test_patch_from_source.py index 1577d7891..b3f0872ee 100644 --- a/ofrak_core/test_ofrak/components/test_patch_from_source.py +++ b/ofrak_core/test_ofrak/components/test_patch_from_source.py @@ -4,8 +4,8 @@ from ofrak_patch_maker.toolchain.llvm_12 import LLVM_12_0_1_Toolchain -import ofrak_angr -import ofrak_capstone +ofrak_angr = pytest.importorskip("ofrak_angr") +ofrak_capstone = pytest.importorskip("ofrak_capstone") from ofrak import OFRAKContext, Resource, ResourceAttributeValueFilter, ResourceFilter from ofrak.core import ( Allocatable, diff --git a/ofrak_core/test_ofrak/components/test_symbolic_analysis.py b/ofrak_core/test_ofrak/components/test_symbolic_analysis.py index e47a08078..97c60a30a 100644 --- a/ofrak_core/test_ofrak/components/test_symbolic_analysis.py +++ b/ofrak_core/test_ofrak/components/test_symbolic_analysis.py @@ -3,7 +3,7 @@ import pytest -import ofrak_angr +ofrak_angr = pytest.importorskip("ofrak_angr") from ofrak import OFRAKContext, Resource, ResourceFilter from ofrak.core import ( ElfSymbolType,