115 lines
3.8 KiB
Python
115 lines
3.8 KiB
Python
import os
|
|
import sys
|
|
import pytest
|
|
import coverage
|
|
import logging
|
|
from subprocess import CalledProcessError
|
|
# Look for the 'utils' module from where the script is running
|
|
prev_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
if prev_dir not in sys.path:
|
|
sys.path.insert(0, prev_dir)
|
|
# Utils import
|
|
from utils import context
|
|
prev_dir = os.path.dirname(prev_dir)
|
|
if prev_dir not in sys.path:
|
|
sys.path.insert(0, prev_dir)
|
|
from kibot.out_base import BaseOutput
|
|
from kibot.gs import GS
|
|
from kibot.kiplot import load_actions
|
|
from kibot.registrable import RegOutput
|
|
from kibot.misc import (MISSING_TOOL, WRONG_INSTALL)
|
|
|
|
|
|
cov = coverage.Coverage()
|
|
mocked_check_output_FNF = True
|
|
|
|
|
|
# Important note:
|
|
# - We can't load the plug-ins twice, the import fails.
|
|
# - Once we patched them using monkey patch the patch isn't reverted unless we load them again.
|
|
# For this reason this patch is used for more than one case.
|
|
def mocked_check_output(cmd, stderr=None):
|
|
logging.debug('mocked_check_output called')
|
|
if mocked_check_output_FNF:
|
|
raise FileNotFoundError()
|
|
else:
|
|
e = CalledProcessError(10, 'rar')
|
|
e.output = b'THE_ERROR'
|
|
raise e
|
|
|
|
|
|
def run_compress(ctx):
|
|
# Load the plug-ins
|
|
load_actions()
|
|
# Create a compress object with the dummy file as source
|
|
out = RegOutput.get_class_for('compress')()
|
|
out.set_tree({'options': {'format': 'RAR', 'files': [{'source': ctx.get_out_path('*')}]}})
|
|
out.config()
|
|
# Setup the GS output dir, needed for the output path
|
|
GS.out_dir = '.'
|
|
# Start coverage
|
|
cov.load()
|
|
cov.start()
|
|
# Run the compression and catch the error
|
|
with pytest.raises(SystemExit) as pytest_wrapped_e:
|
|
out.run('')
|
|
# Stop coverage
|
|
cov.stop()
|
|
cov.save()
|
|
return pytest_wrapped_e
|
|
|
|
|
|
def test_no_rar(caplog, monkeypatch):
|
|
global mocked_check_output_FNF
|
|
mocked_check_output_FNF = True
|
|
# Create a silly context to get the output path
|
|
ctx = context.TestContext('test_no_rar', 'test_v5', 'empty_zip', '')
|
|
# The file we pretend to compress
|
|
ctx.create_dummy_out_file('Test.txt')
|
|
# We will patch subprocess.check_output to make rar fail
|
|
with monkeypatch.context() as m:
|
|
m.setattr("subprocess.check_output", mocked_check_output)
|
|
pytest_wrapped_e = run_compress(ctx)
|
|
# Check we exited because rar isn't installed
|
|
assert pytest_wrapped_e.type == SystemExit
|
|
assert pytest_wrapped_e.value.code == MISSING_TOOL
|
|
assert "Missing `rar` command" in caplog.text
|
|
|
|
|
|
def test_rar_fail(caplog, monkeypatch):
|
|
global mocked_check_output_FNF
|
|
mocked_check_output_FNF = False
|
|
# Create a silly context to get the output path
|
|
ctx = context.TestContext('test_no_rar', 'test_v5', 'empty_zip', '')
|
|
# The file we pretend to compress
|
|
ctx.create_dummy_out_file('Test.txt')
|
|
# We will patch subprocess.check_output to make rar fail
|
|
with monkeypatch.context() as m:
|
|
m.setattr("subprocess.check_output", mocked_check_output)
|
|
pytest_wrapped_e = run_compress(ctx)
|
|
# Check we exited because rar isn't installed
|
|
assert pytest_wrapped_e.type == SystemExit
|
|
assert pytest_wrapped_e.value.code == WRONG_INSTALL
|
|
assert "Failed to invoke rar command, error 10" in caplog.text
|
|
# Not in the docker image ... pytest issue?
|
|
# assert "THE_ERROR" in caplog.text
|
|
|
|
|
|
class NoGetTargets(BaseOutput):
|
|
def __init__(self):
|
|
self.options = True
|
|
self.comment = 'Fake'
|
|
self.name = 'dummy'
|
|
self.type = 'none'
|
|
self._sch_related = True
|
|
|
|
|
|
def test_no_get_targets(caplog):
|
|
test = NoGetTargets()
|
|
test.get_targets('')
|
|
assert "Output 'Fake' (dummy) [none] doesn't implement get_targets(), plese report it" in caplog.text
|
|
# Also check the dependencies fallback
|
|
GS.sch = None
|
|
GS.sch_file = 'fake'
|
|
assert test.get_dependencies() == [GS.sch_file]
|