from catfish.project import DuplicateProcessException, Project, parse_procfile_processes
from tests import BaseTestCase


class ProjectTestCase(BaseTestCase):
    def setUp(self):
        super().setUp()
        self.project = Project(self.EXAMPLE_DIR)

    def test_exists(self):
        self.assertTrue(self.project.exists())

    def test_cant_create_nonexistent_directory(self):
        with self.assertRaises(AssertionError):
            Project("/nonexistent")

    def test_read_processes(self):
        self.assertEqual(len(self.project.processes), 4)
        web_process = self.project.processes[0]
        self.assertEqual(web_process.name, "web")
        self.assertEqual(web_process.command, "python -m http.server $PORT")
        self.assertEqual(web_process.project, self.project)

        bg_process = self.project.processes[1]
        self.assertEqual(bg_process.name, "bg")
        self.assertEqual(bg_process.command, "python src/dummy_program.py")
        self.assertEqual(bg_process.project, self.project)

        bg_process = self.project.processes[2]
        self.assertEqual(bg_process.name, "die")
        self.assertEqual(bg_process.command, "python src/die_soon.py")
        self.assertEqual(bg_process.project, self.project)

        bg_process = self.project.processes[3]
        self.assertEqual(bg_process.name, "exit")
        self.assertEqual(bg_process.command, "python src/exit_soon.py")
        self.assertEqual(bg_process.project, self.project)

    def test_get_process(self):
        self.assertEqual(self.project.get_process("web").name, "web")
        self.assertEqual(self.project.get_process("bg").name, "bg")
        self.assertIsNone(self.project.get_process("nonexistent"))

    def test_name(self):
        self.assertEqual(self.project.name, "example")

    def test_read_environment(self):
        self.assertEqual(self.project.env["FOO"], "bar")


class ProcessTestCase(BaseTestCase):
    def setUp(self):
        super().setUp()
        self.project = Project(self.EXAMPLE_DIR)
        self.process = self.project.get_process("web")

    def test_process_ident(self):
        self.assertEqual(self.process.ident, "example:web")

    def test_duplicate_procfile(self):
        with self.assertRaises(DuplicateProcessException) as e:
            list(parse_procfile_processes(self.project, ["web: 123.py", "web: 456.py"]))

        self.assertEqual(str(e.exception), "web")