catfish/tests/test_worker/test_server.py

from catfish.utils.processes import is_process_running
from catfish.worker.server import send_to_server
from tests import BaseWorkerTestCase


class WorkerServerTestCase(BaseWorkerTestCase):
    def test_server_creates_process(self):
        response = send_to_server({"type": "process", "command": self.DUMMY_EXE})
        self.assertTrue(is_process_running(response["pid"]))

    def test_unknown_command_type(self):
        response = send_to_server({"type": "nothing"})
        self.assertEqual(response, {"error": "Invalid command"})

    def test_ping(self):
        response = send_to_server({"type": "ping"})
        self.assertEqual(response, {"ping": "pong"})