Update module count assertions to reflect reverse_tunnel removal

This commit is contained in:
Cobra
2026-04-08 22:26:06 -04:00
parent ffd384f64b
commit 4d2e46e478
+5 -5
View File
@@ -366,9 +366,9 @@ class TestPackageExports:
assert issubclass(cls, BaseModule)
def test_connectivity_init_exports_all(self):
"""modules.connectivity.__init__.py exports all 8 classes."""
"""modules.connectivity.__init__.py exports all 7 classes."""
import modules.connectivity as pkg
assert len(pkg.__all__) == 8
assert len(pkg.__all__) == 7
for name in pkg.__all__:
cls = getattr(pkg, name)
assert issubclass(cls, BaseModule)
@@ -396,15 +396,15 @@ class TestModuleDiscovery:
from bigbrother import discover_all_modules, discover_modules_in_category
all_mods = discover_all_modules()
# Stealth: 12, Passive: 17, Active: 9, Intel: 9, Connectivity: 8 = 55
assert len(all_mods) == 55, f"Expected 55 modules, found {len(all_mods)}"
# Stealth: 12, Passive: 17, Active: 9, Intel: 9, Connectivity: 7 = 54
assert len(all_mods) == 54, f"Expected 54 modules, found {len(all_mods)}"
# Verify per-category counts
assert len(discover_modules_in_category("stealth")) == 12
assert len(discover_modules_in_category("passive")) == 17
assert len(discover_modules_in_category("active")) == 9
assert len(discover_modules_in_category("intel")) == 9
assert len(discover_modules_in_category("connectivity")) == 8
assert len(discover_modules_in_category("connectivity")) == 7
def test_all_discovered_are_base_module(self):
"""Every discovered module is a BaseModule subclass."""