mirror of https://github.com/pwndbg/pwndbg.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest import mock
|
|
|
|
import gdb
|
|
|
|
from pwndbg.gdblib import config
|
|
|
|
|
|
def set_show(param_name, value):
|
|
gdb.execute(f"show {param_name}")
|
|
gdb.execute(f"set {param_name} {value}")
|
|
gdb.execute(f"show {param_name}")
|
|
|
|
|
|
def test_triggers():
|
|
for param_name, triggers in config.triggers.items():
|
|
p = getattr(config, param_name.replace("-", "_"))
|
|
|
|
mock_triggers = []
|
|
for trigger in triggers:
|
|
mock_triggers.append(mock.Mock(side_effect=trigger))
|
|
|
|
orig_triggers = config.triggers[param_name]
|
|
config.triggers[param_name] = mock_triggers
|
|
|
|
if p.value is True:
|
|
set_show(param_name, "off")
|
|
elif p.value is False:
|
|
set_show(param_name, "on")
|
|
elif isinstance(p.value, int):
|
|
set_show(param_name, 0)
|
|
set_show(param_name, 1)
|
|
set_show(param_name, -1)
|
|
elif isinstance(p.value, str) and p.param_class != gdb.PARAM_ENUM:
|
|
set_show(param_name, "")
|
|
set_show(param_name, "some invalid text")
|
|
set_show(param_name, "red")
|
|
set_show(param_name, "bold,yellow")
|
|
elif isinstance(p.value, str) and p.param_class == gdb.PARAM_ENUM:
|
|
# Only valid values are allowed, invalid values will cause an error
|
|
for enum in p.enum_sequence:
|
|
set_show(param_name, enum)
|
|
else:
|
|
print(p.value, type(p.value))
|
|
assert False
|
|
|
|
for mock_trigger in mock_triggers:
|
|
mock_trigger.assert_called()
|
|
|
|
config.triggers[param_name] = orig_triggers
|