add test for #78

2025-12-22 07:10:29 -08:00 · 2020-07-02 15:50:56 -06:00
parent af06f4d815
commit 9ad52da6d0
1 changed files with 28 additions and 0 deletions
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -269,3 +269,31 @@ def test_byte_matching(sample_9324d1a8ae37a36ae560c37448c9705a):
    )
    capabilities = capa.main.find_capabilities(rules, extractor)
    assert "byte match test" in capabilities
+
+
+def test_count_bb(sample_9324d1a8ae37a36ae560c37448c9705a):
+    rules = capa.rules.RuleSet(
+        [
+            capa.rules.Rule.from_yaml(
+                textwrap.dedent(
+                    """
+                    rule:
+                      meta:
+                        name: count bb
+                        namespace: test
+                        scope: function
+                      features:
+                        - and:
+                          - count(basic blocks): 1 or more
+                    """
+                )
+            )
+        ]
+    )
+
+    extractor = capa.features.extractors.viv.VivisectFeatureExtractor(
+        sample_9324d1a8ae37a36ae560c37448c9705a.vw, sample_9324d1a8ae37a36ae560c37448c9705a.path,
+    )
+    capabilities = capa.main.find_capabilities(rules, extractor)
+    assert "count bb" in capabilities
+