Update module count assertions to reflect reverse_tunnel removal
This commit is contained in:
@@ -366,9 +366,9 @@ class TestPackageExports:
|
|||||||
assert issubclass(cls, BaseModule)
|
assert issubclass(cls, BaseModule)
|
||||||
|
|
||||||
def test_connectivity_init_exports_all(self):
|
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
|
import modules.connectivity as pkg
|
||||||
assert len(pkg.__all__) == 8
|
assert len(pkg.__all__) == 7
|
||||||
for name in pkg.__all__:
|
for name in pkg.__all__:
|
||||||
cls = getattr(pkg, name)
|
cls = getattr(pkg, name)
|
||||||
assert issubclass(cls, BaseModule)
|
assert issubclass(cls, BaseModule)
|
||||||
@@ -396,15 +396,15 @@ class TestModuleDiscovery:
|
|||||||
from bigbrother import discover_all_modules, discover_modules_in_category
|
from bigbrother import discover_all_modules, discover_modules_in_category
|
||||||
|
|
||||||
all_mods = discover_all_modules()
|
all_mods = discover_all_modules()
|
||||||
# Stealth: 12, Passive: 17, Active: 9, Intel: 9, Connectivity: 8 = 55
|
# Stealth: 12, Passive: 17, Active: 9, Intel: 9, Connectivity: 7 = 54
|
||||||
assert len(all_mods) == 55, f"Expected 55 modules, found {len(all_mods)}"
|
assert len(all_mods) == 54, f"Expected 54 modules, found {len(all_mods)}"
|
||||||
|
|
||||||
# Verify per-category counts
|
# Verify per-category counts
|
||||||
assert len(discover_modules_in_category("stealth")) == 12
|
assert len(discover_modules_in_category("stealth")) == 12
|
||||||
assert len(discover_modules_in_category("passive")) == 17
|
assert len(discover_modules_in_category("passive")) == 17
|
||||||
assert len(discover_modules_in_category("active")) == 9
|
assert len(discover_modules_in_category("active")) == 9
|
||||||
assert len(discover_modules_in_category("intel")) == 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):
|
def test_all_discovered_are_base_module(self):
|
||||||
"""Every discovered module is a BaseModule subclass."""
|
"""Every discovered module is a BaseModule subclass."""
|
||||||
|
|||||||
Reference in New Issue
Block a user