Git-Mirrors/brozzler
1
0
Fork 0
mirror of https://github.com/internetarchive/brozzler.git synced 2025-04-20 15:55:49 -04:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

test: replace pkg_resources call

Browse Source
This commit is contained in:
Misty De Méo 2025-02-26 11:38:58 -08:00
parent 418758fc22
commit 42210907f5
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/test_cli.py
View File

@ -18,14 +18,24 @@ limitations under the License.
"""
import brozzler.cli
import pkg_resources
import importlib.metadata
import pytest
import subprocess
import doublethink
def console_scripts():
# We do a dict comprehension here because the select filters aren't
# available until Python 3.10's importlib.
return {
ep.name: ep
for ep in importlib.metadata.distribution("brozzler").entry_points
if ep.group == "console_scripts"
}
def cli_commands():
commands = set(pkg_resources.get_entry_map("brozzler")["console_scripts"].keys())
commands = set(console_scripts().keys())
commands.remove("brozzler-wayback")
try:
import gunicorn
@ -40,8 +50,8 @@ def cli_commands():
@pytest.mark.parametrize("cmd", cli_commands())
def test_call_entrypoint(capsys, cmd):
entrypoint = pkg_resources.get_entry_map("brozzler")["console_scripts"][cmd]
callable = entrypoint.resolve()
entrypoint = console_scripts()[cmd]
callable = entrypoint.load()
with pytest.raises(SystemExit):
callable(["/whatever/bin/%s" % cmd, "--version"])
out, err = capsys.readouterr()