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()