17 lines
415 B
Python
17 lines
415 B
Python
from unittest import TestCase
|
|
from click.testing import CliRunner
|
|
from catfish.__main__ import cli
|
|
from catfish import worker
|
|
import functools
|
|
|
|
|
|
class BaseTestCase(TestCase):
|
|
def setUp(self):
|
|
worker.stop_worker()
|
|
self.cli_runner = CliRunner()
|
|
self.cli = cli
|
|
self.run_cli = functools.partial(self.cli_runner.invoke, self.cli)
|
|
|
|
def tearDown(self):
|
|
worker.stop_worker()
|