# Copyright 2016 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unittests for loman.py"""

import os
from xml.etree import ElementTree

from chromite.lib import cros_test_lib
from chromite.lib import osutils
from chromite.lib import partial_mock
from chromite.scripts import loman


class RunGitMock(partial_mock.PartialCmdMock):
    """Partial mock for git.RunGit."""

    TARGET = "chromite.lib.git"
    ATTRS = ("RunGit",)
    DEFAULT_ATTR = "RunGit"

    def RunGit(self, _git_repo, cmd, _retry=True, **kwargs):
        return self._results["RunGit"].LookupResult(
            (cmd,), hook_args=(cmd,), hook_kwargs=kwargs
        )


class ParserTest(cros_test_lib.OutputTestCase):
    """Tests for the CLI parser."""

    def setUp(self) -> None:
        self.parser = loman.GetParser()

    def testNoCommand(self) -> None:
        """Require a command at least."""
        with self.OutputCapturer():
            self.assertRaises(SystemExit, self.parser.parse_args, [])

    def testBadCommand(self) -> None:
        """Reject unknown commands."""
        with self.OutputCapturer():
            self.assertRaises(SystemExit, self.parser.parse_args, ["flyaway"])

    def testAddCommand(self) -> None:
        """Verify basic add command behavior."""
        with self.OutputCapturer():
            self.parser.parse_args(["add", "--workon", "project"])
            self.parser.parse_args(
                ["add", "project", "path", "--remote", "foo"]
            )


class ManifestTest(cros_test_lib.TempDirTestCase):
    """Tests that need a real .repo/ manifest layout."""

    def setUp(self) -> None:
        # The loman code looks for the repo root, so make one, and chdir there.
        os.chdir(self.tempdir)

        for d in ("repo", "manifests", "manifests.git"):
            osutils.SafeMakedirs(os.path.join(".repo", d))

        for m in ("default.xml", "full.xml", "minilayout.xml"):
            osutils.Touch(os.path.join(".repo", "manifests", m))

        self._SetManifest("default.xml")

    def _SetManifest(self, manifest) -> None:
        """Set active manifest to point to |manifest|."""
        source = os.path.join(".repo", "manifest.xml")
        target = os.path.join("manifests", manifest)
        osutils.SafeUnlink(source)
        os.symlink(target, source)


class AddTest(cros_test_lib.MockOutputTestCase, ManifestTest):
    """Tests for the add command."""

    def testRejectBadCommands(self) -> None:
        """Reject bad invocations."""
        bad_cmds = (
            # Missing path.
            ["add"],
            # Extra project.
            ["add", "--workon", "path", "project"],
            # Missing --remote.
            ["add", "path", "project"],
            # Missing project.
            ["add", "path", "--remote", "remote"],
        )
        with self.OutputCapturer():
            for cmd in bad_cmds:
                self.assertRaises(SystemExit, loman.main, cmd)


class NoMiniayoutTest(cros_test_lib.MockOutputTestCase, ManifestTest):
    """Check deprecated minilayout setups are detected."""

    def setUp(self) -> None:
        self._SetManifest("minilayout.xml")

    def testMiniLayoutDetected(self) -> None:
        """Check error is raised when repo is setup with minilayout."""

        class _Error(Exception):
            """Stub for test."""

        self.PatchObject(loman, "_AssertNotMiniLayout", side_effect=_Error)
        cmd = ["add", "-w", "foo"]
        with self.OutputCapturer():
            self.assertRaises(_Error, loman.main, cmd)


class IncludeXmlTest(cros_test_lib.MockOutputTestCase, ManifestTest):
    """End to End tests for reading and producing XML trees."""

    PROJECT = "chromiumos/repohooks"

    def setUp(self) -> None:
        INCLUDING_XML = "including.xml"
        INCLUDED_XML = "included.xml"
        osutils.WriteFile(
            os.path.join(".repo", "manifests", INCLUDING_XML),
            """
<manifest>
    <include name="%s" />
    <project remote="cros-internal" path="crostools" groups="br" name="ct" />
</manifest>"""
            % (INCLUDED_XML,),
        )

        osutils.WriteFile(
            os.path.join(".repo", "manifests", INCLUDED_XML),
            """
<manifest>
    <default remote="cros" revision="HEAD" />
    <remote name="cros" />
    <remote name="cros-internal" />
    <project path="src/repohooks" name="%s" groups="minilayout,bt" />
</manifest>"""
            % (self.PROJECT,),
        )
        self._SetManifest(INCLUDING_XML)

        self.git_mock = self.StartPatcher(RunGitMock())
        self.git_mock.AddCmdResult(
            ["symbolic-ref", "-q", "HEAD"], stdout="default"
        )
        self.git_mock.AddCmdResult(
            ["config", "--get-regexp", "branch\\.default\\.(remote|merge)"],
            stdout="branch.default.merge firmware-branch",
        )

        self.git_mock.AddCmdResult(
            [
                "config",
                "-f",
                os.path.join(self.tempdir, ".repo", "manifests.git", "config"),
                "--get",
                "manifest.groups",
            ],
            stdout="group1,group2",
        )

    def testAddExistingProject(self) -> None:
        """Add an existing project, check no local_manifest.xml are created."""
        self.git_mock.AddCmdResult(
            [
                "config",
                "-f",
                os.path.join(self.tempdir, ".repo", "manifests.git", "config"),
                "manifest.groups",
                "minilayout,platform-linux,group1,group2,name:%s"
                % (self.PROJECT,),
            ]
        )
        cmd = ["add", "-w", self.PROJECT]
        with self.OutputCapturer():
            self.assertEqual(loman.main(cmd), 0)
        self.assertNotExists(os.path.join(".repo", "local_manifest.xml"))

    def testAddNewProject(self) -> None:
        """Add new project to the repo.

        Check local_manifest.xml is created and valid.
        """
        new_project = "project"
        self.git_mock.AddCmdResult(
            [
                "config",
                "-f",
                os.path.join(self.tempdir, ".repo", "manifests.git", "config"),
                "manifest.groups",
                "minilayout,platform-linux,group1,group2,name:%s"
                % (new_project,),
            ],
        )
        cmd = ["add", new_project, "path", "-r", "remote"]
        with self.OutputCapturer():
            self.assertEqual(loman.main(cmd), 0)
        expected_local_manifest_nodes = ElementTree.fromstring(
            """
<manifest>
  <project name="project" path="path" remote="remote" workon="False" />
</manifest>"""
        )
        with open(
            os.path.join(".repo", "local_manifest.xml"), encoding="utf-8"
        ) as f:
            local_manifest_nodes = ElementTree.fromstring(f.read())

        # Read project, check for failure.
        self.assertEqual(
            ElementTree.tostring(expected_local_manifest_nodes),
            ElementTree.tostring(local_manifest_nodes),
        )

        # Check that re-adding triggers error.
        cmd = ["add", new_project, "path", "-r", "remote"]
        with self.OutputCapturer() as output:
            self.assertRaises(SystemExit, loman.main, cmd)
            self.assertIn("conflicts with", "\n".join(output.GetStderrLines()))