# Copyright (C) 2006-2010, 2012, 2016 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

"""Black-box tests for brz nick."""

from breezy import branch, osutils, tests


class TestNick(tests.TestCaseWithTransport):
    def assertNick(self, expected, working_dir=".", explicit=None, directory=None):
        cmd = ["nick"]
        if directory is not None:
            cmd.extend(["--directory", directory])
        # The nick command outputs the nick on a single line
        actual = self.run_bzr(cmd, working_dir=working_dir)[0][:-1]
        self.assertEqual(expected, actual)
        if explicit is not None:
            br = branch.Branch.open(working_dir)
            conf = br.get_config()
            self.assertEqual(explicit, conf.has_explicit_nickname())
            if explicit:
                self.assertEqual(expected, conf._get_explicit_nickname())

    def test_nick_command(self):
        """Brz nick for viewing, setting nicknames."""
        self.make_branch_and_tree("me.dev")
        self.assertNick("me.dev", working_dir="me.dev")
        # set the nickname
        self.run_bzr("nick moo", working_dir="me.dev")
        self.assertNick("moo", working_dir="me.dev")

    def test_autonick_urlencoded(self):
        # https://bugs.launchpad.net/bzr/+bug/66857 -- nick was printed
        # urlencoded but shouldn't be
        self.make_branch_and_tree("!repo")
        self.assertNick("!repo", working_dir="!repo")

    def test_bound_nick(self):
        """Bind should not update implicit nick."""
        base = self.make_branch_and_tree("base")
        self.make_branch_and_tree("child")
        self.assertNick("child", working_dir="child", explicit=False)

        self.run_bzr("bind ../base", working_dir="child")
        self.assertNick(base.branch.nick, working_dir="child", explicit=False)

    def test_bound_nick_explicit(self):
        """Bind should update explicit nick."""
        base = self.make_branch_and_tree("base")
        self.make_branch_and_tree("child")
        self.run_bzr("nick explicit_nick", working_dir="child")
        self.assertNick("explicit_nick", working_dir="child", explicit=True)
        self.run_bzr("bind ../base", working_dir="child")
        self.assertNick(base.branch.nick, working_dir="child", explicit=True)

    def test_boundless_nick(self):
        """Nick defaults to implicit local nick when bound branch is AWOL."""
        base = self.make_branch_and_tree("base")
        self.make_branch_and_tree("child")
        self.run_bzr("bind ../base", working_dir="child")
        self.assertNick(base.branch.nick, working_dir="child", explicit=False)
        osutils.rmtree("base")
        self.assertNick("child", working_dir="child", explicit=False)

    def test_nick_directory(self):
        """Test --directory option."""
        self.make_branch_and_tree("me.dev")
        self.assertNick("me.dev", directory="me.dev")
        self.run_bzr(["nick", "-d", "me.dev", "moo"])
        self.assertNick("moo", directory="me.dev")
