This repository has been archived on 2023-03-26. You can view files and clone it, but cannot push or open issues or pull requests.
catfish/tests/__init__.py

37 lines
1.1 KiB
Python

import functools
import os
from aiohttp.test_utils import unused_port
from aiounittest import AsyncTestCase
from click.testing import CliRunner
from catfish.__main__ import cli
from catfish.utils.processes import CURRENT_PROCESS, terminate_processes
from catfish.utils.sockets import create_base_socket_dir, delete_base_socket_dir
class BaseTestCase(AsyncTestCase):
DUMMY_EXE = os.path.join(os.path.dirname(__file__), "dummy_program.py")
def setUp(self):
create_base_socket_dir()
self.cli_runner = CliRunner()
self.cli = cli
self.run_cli = functools.partial(self.cli_runner.invoke, self.cli)
def tearDown(self):
terminate_processes(CURRENT_PROCESS.children(recursive=True))
delete_base_socket_dir()
class BaseWorkerTestCase(BaseTestCase):
def setUp(self):
super().setUp()
self.unused_port = unused_port()
self.assertEqual(
self.run_cli(["start", "--port", self.unused_port]).exit_code, 0
)
def tearDown(self):
self.assertEqual(self.run_cli(["stop"]).exit_code, 0)
super().tearDown()