add IDA specific test

doomedraven · Oct 21, 2020 · 1e3b29d · 1e3b29d
1 parent 0a369c5
commit 1e3b29d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/tests/fixtures.py b/tests/fixtures.py
@@ -421,6 +421,12 @@ def parametrize(params, values, **kwargs):
     ("mimikatz", "function=0x4556E5", capa.features.Characteristic("calls to"), False),
 ]
 
+FEATURE_PRESENCE_TESTS_IDA = [
+    # file/imports
+    # IDA can recover more names of APIs imported by ordinal
+    ("mimikatz", "file", capa.features.file.Import("cabinet.FCIAddFile"), True),
+]
+
 FEATURE_COUNT_TESTS = [
     ("mimikatz", "function=0x40E5C2", capa.features.basicblock.BasicBlock(), 7),
     ("mimikatz", "function=0x4702FD", capa.features.Characteristic("calls from"), 0),

diff --git a/tests/test_ida_features.py b/tests/test_ida_features.py
@@ -44,7 +44,7 @@ def get_ida_extractor(_path):
 
 @pytest.mark.skip(reason="IDA Pro tests must be run within IDA")
 def test_ida_features():
-    for (sample, scope, feature, expected) in FEATURE_PRESENCE_TESTS:
+    for (sample, scope, feature, expected) in FEATURE_PRESENCE_TESTS + FEATURE_PRESENCE_TESTS_IDA:
         id = make_test_id((sample, scope, feature, expected))
 
         try: