from pathlib import Path
import os

from tests.test_runner import AbstractCommandTest


class CreatePythonFileTest(AbstractCommandTest):
    def setUp(self):
        super().setUp()

        from flamenco_worker.commands import CreatePythonFile
        import tempfile

        self.tmpdir = tempfile.TemporaryDirectory()
        self.tmppath = Path(self.tmpdir.name)

        self.cmd = CreatePythonFile(
            worker=self.fworker,
            task_id="12345",
            command_idx=0,
        )

    def tearDown(self):
        super().tearDown()
        self.tmpdir.cleanup()

    def test_validate_settings(self):
        self.assertIn(
            "filepath", self.cmd.validate({"filepath": 12, "contents": "# comment"})
        )
        self.assertIn(
            "filepath", self.cmd.validate({"filepath": "", "contents": "# comment"})
        )
        self.assertIn(
            "filepath", self.cmd.validate({"filepath": "/nonpy/path", "contents": "#"})
        )
        self.assertIn("filepath", self.cmd.validate({"contents": "#"}))

        self.assertIn(
            "content", self.cmd.validate({"filepath": "/valid/path.py", "contents": 12})
        )
        self.assertIn("content", self.cmd.validate({"filepath": "/valid/path.py"}))

        self.assertTrue(self.cmd.validate({}))
        self.assertFalse(
            self.cmd.validate({"filepath": "/valid/path.py", "contents": ""})
        )
        self.assertFalse(
            self.cmd.validate({"filepath": "/valid/path.py", "contents": "#"})
        )
        self.assertFalse(
            self.cmd.validate({"filepath": "/valid/path.py", "contents": "##\na=b\n"})
        )

    def test_nonexistant_path(self):
        filepath = self.tmppath / "nonexisting-dir" / "somefile.py"
        task = self.cmd.run({"filepath": str(filepath), "contents": "aapje"})
        ok = self.loop.run_until_complete(task)

        self.assertTrue(ok)
        self.assertTrue(filepath.exists())
        self.assertEqual("aapje", filepath.read_text())

    def test_existing_path(self):
        filepath = self.tmppath / "existing.py"
        filepath.write_text("old content")

        task = self.cmd.run({"filepath": str(filepath), "contents": "öpje"})
        ok = self.loop.run_until_complete(task)

        self.assertTrue(ok)
        self.assertTrue(filepath.exists())
        self.assertEqual("öpje", filepath.read_text(encoding="utf8"))