mirror of https://github.com/Kkevsterrr/geneva
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
import sys
|
|
import pytest
|
|
# Include the root of the project
|
|
sys.path.append("..")
|
|
|
|
import actions.action
|
|
import actions.strategy
|
|
import actions.utils
|
|
import actions.duplicate
|
|
|
|
|
|
def get_test_configs():
|
|
"""
|
|
Sets up the tests
|
|
"""
|
|
tests = [
|
|
("both", True, ['DuplicateAction', 'TraceAction', 'DropAction', 'SleepAction', 'TamperAction', 'FragmentAction']),
|
|
("in", True, ['DropAction', 'TamperAction']),
|
|
("out", True, ['DropAction', 'TamperAction', 'SleepAction', 'TraceAction', 'DuplicateAction', 'FragmentAction']),
|
|
("both", False, ['DuplicateAction', 'SleepAction', 'TamperAction', 'FragmentAction']),
|
|
("in", False, ['TamperAction']),
|
|
("out", False, ['TamperAction', 'SleepAction', 'DuplicateAction', 'FragmentAction']),
|
|
]
|
|
# To ensure caching is not breaking anything, double the tests
|
|
return tests + tests
|
|
|
|
|
|
@pytest.mark.parametrize("direction,allow_terminal,supported_actions", get_test_configs())
|
|
def test_get_actions(direction, allow_terminal, supported_actions):
|
|
"""
|
|
Tests the duplicate action primitive.
|
|
"""
|
|
collected_actions = actions.action.Action.get_actions(direction, allow_terminal=allow_terminal)
|
|
names = []
|
|
for name, action_class in collected_actions:
|
|
names.append(name)
|
|
assert set(names) == set(supported_actions)
|
|
assert len(names) == len(supported_actions)
|
|
|
|
|
|
def test_punish_no_engine(logger):
|
|
"""
|
|
Tests that punish_fitness properly handles no engine
|
|
"""
|
|
assert 100 == actions.utils.punish_fitness(100, logger, None)
|
|
assert 100 == actions.utils.punish_complexity(100, logger, None)
|
|
assert 100 == actions.utils.punish_unused(100, logger, None)
|
|
|
|
|
|
def test_write_fitness_error(logger):
|
|
"""
|
|
Tests handling of write fitness error cases
|
|
"""
|
|
with pytest.raises(ValueError):
|
|
actions.utils.write_fitness("<thiswillfail>", None, None)
|
|
|
|
|
|
def test_skipstrat(logger):
|
|
"""
|
|
Tests we can create and raise a SkipStrategyException
|
|
"""
|
|
with pytest.raises(actions.utils.SkipStrategyException):
|
|
raise actions.utils.SkipStrategyException("Skip this!", 100)
|
|
|
|
|
|
def test_import_plugin(logger):
|
|
"""
|
|
Tries to import some plugins
|
|
"""
|
|
assert actions.utils.import_plugin("http", "client")
|