mirror of
https://github.com/internetarchive/brozzler.git
synced 2025-02-24 08:39:59 -05:00
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
'''
|
|
test_cli.py - test brozzler commands
|
|
|
|
Copyright (C) 2017 Internet Archive
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
'''
|
|
|
|
import brozzler.cli
|
|
import pkg_resources
|
|
import pytest
|
|
import subprocess
|
|
|
|
def cli_commands():
|
|
commands = set(pkg_resources.get_entry_map(
|
|
'brozzler')['console_scripts'].keys())
|
|
commands.remove('brozzler-wayback')
|
|
try:
|
|
import gunicorn
|
|
except ImportError:
|
|
commands.remove('brozzler-dashboard')
|
|
try:
|
|
import pywb
|
|
except ImportError:
|
|
commands.remove('brozzler-easy')
|
|
return 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()
|
|
with pytest.raises(SystemExit):
|
|
callable(['/whatever/bin/%s' % cmd, '--version'])
|
|
out, err = capsys.readouterr()
|
|
assert out == 'brozzler %s - %s\n' % (brozzler.__version__, cmd)
|
|
assert err == ''
|
|
|
|
@pytest.mark.parametrize('cmd', cli_commands())
|
|
def test_run_command(capsys, cmd):
|
|
proc = subprocess.Popen(
|
|
[cmd, '--version'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
out, err = proc.communicate()
|
|
assert out == ('brozzler %s - %s\n' % (
|
|
brozzler.__version__, cmd)).encode('ascii')
|
|
assert err == b''
|
|
|