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.
39 lines
992 B
Python
39 lines
992 B
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
from unittest.mock import MagicMock
|
|
|
|
# Replace `pwndbg.commands` module with a mock to prevent import errors, as well
|
|
# as the `load_commands` function
|
|
module_name = "pwndbg.commands"
|
|
module = MagicMock(__name__=module_name, load_commands=lambda: None)
|
|
sys.modules[module_name] = module
|
|
|
|
import os
|
|
import tempfile
|
|
|
|
# We must import the function under test after all the mocks are imported
|
|
from pwndbg.lib.which import which
|
|
|
|
# Load the mocks for the `gdb` and `gdblib` modules
|
|
from .mocks import gdb # noqa: F401
|
|
from .mocks import gdblib # noqa: F401
|
|
|
|
|
|
def test_basic():
|
|
assert which("ls").endswith("/ls")
|
|
|
|
|
|
def test_nonexistent():
|
|
assert which("definitely-not-a-real-command") is None
|
|
|
|
|
|
def test_dir():
|
|
with tempfile.TemporaryDirectory() as tempdir:
|
|
path = os.path.join(tempdir, "test_file")
|
|
with open(path, "w") as f:
|
|
f.write("test")
|
|
os.chmod(path, 0o755)
|
|
|
|
assert which(path) == path
|