mirror of
https://github.com/nvbn/thefuck.git
synced 2025-01-19 12:24:29 +00:00
55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
import os
|
|
from contextlib import contextmanager
|
|
import subprocess
|
|
import shutil
|
|
from tempfile import mkdtemp
|
|
from pathlib import Path
|
|
import sys
|
|
import pexpect
|
|
import pytest
|
|
|
|
root = str(Path(__file__).parent.parent.parent.resolve())
|
|
bare = os.environ.get('BARE')
|
|
enabled = os.environ.get('FUNCTIONAL')
|
|
|
|
|
|
def build_container(tag, dockerfile):
|
|
tmpdir = mkdtemp()
|
|
with Path(tmpdir).joinpath('Dockerfile').open('w') as file:
|
|
file.write(dockerfile)
|
|
if subprocess.call(['docker', 'build', '--tag={}'.format(tag), tmpdir],
|
|
cwd=root) != 0:
|
|
raise Exception("Can't build a container")
|
|
shutil.rmtree(tmpdir)
|
|
|
|
|
|
@contextmanager
|
|
def spawn(tag, dockerfile, cmd):
|
|
if bare:
|
|
proc = pexpect.spawnu(cmd)
|
|
else:
|
|
tag = 'thefuck/{}'.format(tag)
|
|
build_container(tag, dockerfile)
|
|
proc = pexpect.spawnu('docker run --volume {}:/src --tty=true '
|
|
'--interactive=true {} {}'.format(root, tag, cmd))
|
|
proc.sendline('pip install /src')
|
|
|
|
proc.logfile = sys.stdout
|
|
|
|
try:
|
|
yield proc
|
|
finally:
|
|
proc.terminate(force=bare)
|
|
|
|
|
|
def images(*items):
|
|
if bare:
|
|
return [items[0]]
|
|
else:
|
|
return items
|
|
|
|
|
|
functional = pytest.mark.skipif(
|
|
not enabled,
|
|
reason='Functional tests are disabled by default.')
|