Update module count assertions to reflect reverse_tunnel removal
This commit is contained in:
@@ -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."""
|
||||
|
||||
Reference in New Issue
Block a user