from unittest.mock import patch

from django.test import SimpleTestCase

from dispatcharr.db.process_label import get_process_role, uses_geventpool_database_backend


class ProcessLabelTests(SimpleTestCase):
    def test_celery_worker_not_labeled_as_uwsgi(self):
        role = get_process_role(
            ["/dispatcharrpy/bin/celery", "-A", "dispatcharr", "worker"]
        )
        self.assertEqual(role, "celery-worker")

    def test_uwsgi_labeled_from_argv(self):
        role = get_process_role(["/dispatcharrpy/bin/uwsgi", "--ini", "/app/docker/uwsgi.ini"])
        self.assertEqual(role, "uwsgi")

    def test_uwsgi_labeled_when_worker_module_present(self):
        fake_uwsgi = type("uwsgi", (), {"worker_id": staticmethod(lambda: 2)})()
        with patch.dict("sys.modules", {"uwsgi": fake_uwsgi}):
            role = get_process_role(["/dispatcharrpy/bin/python", "-c", "pass"])
        self.assertEqual(role, "uwsgi")

    def test_uwsgi_master_not_labeled_as_uwsgi(self):
        fake_uwsgi = type("uwsgi", (), {"worker_id": staticmethod(lambda: 0)})()
        with patch.dict("sys.modules", {"uwsgi": fake_uwsgi}):
            role = get_process_role(["/dispatcharrpy/bin/python", "-c", "pass"])
        self.assertEqual(role, "django")


class DatabaseBackendSelectionTests(SimpleTestCase):
    def test_celery_workers_use_standard_postgres_backend(self):
        self.assertFalse(uses_geventpool_database_backend(
            ["celery", "-A", "dispatcharr", "worker", "-Q", "celery"]
        ))
        self.assertFalse(uses_geventpool_database_backend(
            ["celery", "-A", "dispatcharr", "worker", "-Q", "dvr", "--pool=threads"]
        ))
        self.assertFalse(uses_geventpool_database_backend(
            ["celery", "-A", "dispatcharr", "beat"]
        ))

    def test_uwsgi_uses_geventpool_backend(self):
        self.assertTrue(uses_geventpool_database_backend(
            ["uwsgi", "--ini", "/app/docker/uwsgi.ini"]
        ))

    def test_manage_uses_geventpool_backend(self):
        self.assertTrue(uses_geventpool_database_backend(
            ["manage.py", "migrate"]
        ))