# Copyright (C) 2005-2010 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

"""Test testaments for gpg signing."""

# TODO: Testaments with x-bits

import os

from breezy import osutils
from breezy.bzr.testament import StrictTestament, StrictTestament3, Testament
from breezy.tests import TestCaseWithTransport
from breezy.tests.features import SymlinkFeature


class TestamentSetup(TestCaseWithTransport):
    def setUp(self):
        super().setUp()
        self.wt = self.make_branch_and_tree(".")
        self.wt.set_root_id(b"TREE_ROT")
        b = self.b = self.wt.branch
        b.nick = "test branch"
        self.wt.commit(
            message="initial null commit",
            committer="test@user",
            timestamp=1129025423,  # 'Tue Oct 11 20:10:23 2005'
            timezone=0,
            rev_id=b"test@user-1",
        )
        self.build_tree_contents(
            [
                ("hello", b"contents of hello file"),
                ("src/",),
                ("src/foo.c", b"int main()\n{\n}\n"),
            ]
        )
        self.wt.add(
            ["hello", "src", "src/foo.c"], ids=[b"hello-id", b"src-id", b"foo.c-id"]
        )
        tt = self.wt.transform()
        trans_id = tt.trans_id_tree_path("hello")
        tt.set_executability(True, trans_id)
        tt.apply()
        self.wt.commit(
            message="add files and directories",
            timestamp=1129025483,
            timezone=36000,
            rev_id=b"test@user-2",
            committer="test@user",
        )


class TestamentTests(TestamentSetup):
    def testament_class(self):
        return Testament

    def expected(self, key):
        return texts[self.testament_class()][key]

    def from_revision(self, repository, revision_id):
        return self.testament_class().from_revision(repository, revision_id)

    def test_null_testament(self):
        """Testament for a revision with no contents."""
        t = self.from_revision(self.b.repository, b"test@user-1")
        ass = self.assertTrue
        eq = self.assertEqual
        ass(isinstance(t, Testament))
        eq(t.revision_id, b"test@user-1")
        eq(t.committer, "test@user")
        eq(t.timestamp, 1129025423)
        eq(t.timezone, 0)

    def test_testment_text_form(self):
        """Conversion of testament to canonical text form."""
        t = self.from_revision(self.b.repository, b"test@user-1")
        text_form = t.as_text()
        self.log("testament text form:\n%s" % text_form)
        self.assertEqualDiff(text_form, self.expected("rev_1"))
        short_text_form = t.as_short_text()
        self.assertEqualDiff(short_text_form, self.expected("rev_1_short"))

    def test_testament_with_contents(self):
        """Testament containing a file and a directory."""
        t = self.from_revision(self.b.repository, b"test@user-2")
        text_form = t.as_text()
        self.log("testament text form:\n%s" % text_form)
        self.assertEqualDiff(text_form, self.expected("rev_2"))
        actual_short = t.as_short_text()
        self.assertEqualDiff(actual_short, self.expected("rev_2_short"))

    def test_testament_symlinks(self):
        """Testament containing symlink (where possible)"""
        self.requireFeature(SymlinkFeature(self.test_dir))
        os.symlink("wibble/linktarget", "link")
        self.wt.add(["link"], ids=[b"link-id"])
        self.wt.commit(
            message="add symlink",
            timestamp=1129025493,
            timezone=36000,
            rev_id=b"test@user-3",
            committer="test@user",
        )
        t = self.from_revision(self.b.repository, b"test@user-3")
        self.assertEqualDiff(t.as_text(), self.expected("rev_3"))

    def test_testament_revprops(self):
        """Testament to revision with extra properties"""
        props = {
            "flavor": "sour cherry\ncream cheese",
            "size": "medium",
            "empty": "",
        }
        self.wt.commit(
            message="revision with properties",
            timestamp=1129025493,
            timezone=36000,
            rev_id=b"test@user-3",
            committer="test@user",
            revprops=props,
        )
        t = self.from_revision(self.b.repository, b"test@user-3")
        self.assertEqualDiff(t.as_text(), self.expected("rev_props"))

    def test_testament_unicode_commit_message(self):
        self.wt.commit(
            message="non-ascii commit \N{COPYRIGHT SIGN} me",
            timestamp=1129025493,
            timezone=36000,
            rev_id=b"test@user-3",
            committer="Erik B\xe5gfors <test@user>",
            revprops={"uni": "\xb5"},
        )
        t = self.from_revision(self.b.repository, b"test@user-3")
        self.assertEqualDiff(
            self.expected("sample_unicode").encode("utf-8"), t.as_text()
        )

    def test_from_tree(self):
        tree = self.b.repository.revision_tree(b"test@user-2")
        testament = self.testament_class().from_revision_tree(tree)
        text_1 = testament.as_short_text()
        text_2 = self.from_revision(self.b.repository, b"test@user-2").as_short_text()
        self.assertEqual(text_1, text_2)

    def test___init__(self):
        revision = self.b.repository.get_revision(b"test@user-2")
        tree = self.b.repository.revision_tree(b"test@user-2")
        testament_1 = self.testament_class()(revision, tree)
        text_1 = testament_1.as_short_text()
        text_2 = self.from_revision(self.b.repository, b"test@user-2").as_short_text()
        self.assertEqual(text_1, text_2)


class TestamentTestsStrict(TestamentTests):
    def testament_class(self):
        return StrictTestament


class TestamentTestsStrict2(TestamentTests):
    def testament_class(self):
        return StrictTestament3


REV_1_TESTAMENT = b"""\
bazaar-ng testament version 1
revision-id: test@user-1
committer: test@user
timestamp: 1129025423
timezone: 0
parents:
message:
  initial null commit
inventory:
properties:
  branch-nick:
    test branch
"""


REV_1_STRICT_TESTAMENT = b"""\
bazaar-ng testament version 2.1
revision-id: test@user-1
committer: test@user
timestamp: 1129025423
timezone: 0
parents:
message:
  initial null commit
inventory:
properties:
  branch-nick:
    test branch
"""


REV_1_STRICT_TESTAMENT3 = b"""\
bazaar testament version 3 strict
revision-id: test@user-1
committer: test@user
timestamp: 1129025423
timezone: 0
parents:
message:
  initial null commit
inventory:
  directory . TREE_ROT test@user-1 no
properties:
  branch-nick:
    test branch
"""


REV_1_SHORT = b"""\
bazaar-ng testament short form 1
revision-id: test@user-1
sha1: %s
""" % osutils.sha_string(REV_1_TESTAMENT)


REV_1_SHORT_STRICT = b"""\
bazaar-ng testament short form 2.1
revision-id: test@user-1
sha1: %s
""" % osutils.sha_string(REV_1_STRICT_TESTAMENT)


REV_1_SHORT_STRICT3 = b"""\
bazaar testament short form 3 strict
revision-id: test@user-1
sha1: %s
""" % osutils.sha_string(REV_1_STRICT_TESTAMENT3)


REV_2_TESTAMENT = b"""\
bazaar-ng testament version 1
revision-id: test@user-2
committer: test@user
timestamp: 1129025483
timezone: 36000
parents:
  test@user-1
message:
  add files and directories
inventory:
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
  directory src src-id
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
properties:
  branch-nick:
    test branch
"""


REV_2_STRICT_TESTAMENT = b"""\
bazaar-ng testament version 2.1
revision-id: test@user-2
committer: test@user
timestamp: 1129025483
timezone: 36000
parents:
  test@user-1
message:
  add files and directories
inventory:
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
  directory src src-id test@user-2 no
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
properties:
  branch-nick:
    test branch
"""


REV_2_STRICT_TESTAMENT3 = b"""\
bazaar testament version 3 strict
revision-id: test@user-2
committer: test@user
timestamp: 1129025483
timezone: 36000
parents:
  test@user-1
message:
  add files and directories
inventory:
  directory . TREE_ROT test@user-1 no
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
  directory src src-id test@user-2 no
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
properties:
  branch-nick:
    test branch
"""


REV_2_SHORT = b"""\
bazaar-ng testament short form 1
revision-id: test@user-2
sha1: %s
""" % osutils.sha_string(REV_2_TESTAMENT)


REV_2_SHORT_STRICT = b"""\
bazaar-ng testament short form 2.1
revision-id: test@user-2
sha1: %s
""" % osutils.sha_string(REV_2_STRICT_TESTAMENT)


REV_2_SHORT_STRICT3 = b"""\
bazaar testament short form 3 strict
revision-id: test@user-2
sha1: %s
""" % osutils.sha_string(REV_2_STRICT_TESTAMENT3)


REV_PROPS_TESTAMENT = b"""\
bazaar-ng testament version 1
revision-id: test@user-3
committer: test@user
timestamp: 1129025493
timezone: 36000
parents:
  test@user-2
message:
  revision with properties
inventory:
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
  directory src src-id
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
properties:
  branch-nick:
    test branch
  empty:
  flavor:
    sour cherry
    cream cheese
  size:
    medium
"""


REV_PROPS_TESTAMENT_STRICT = b"""\
bazaar-ng testament version 2.1
revision-id: test@user-3
committer: test@user
timestamp: 1129025493
timezone: 36000
parents:
  test@user-2
message:
  revision with properties
inventory:
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
  directory src src-id test@user-2 no
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
properties:
  branch-nick:
    test branch
  empty:
  flavor:
    sour cherry
    cream cheese
  size:
    medium
"""


REV_PROPS_TESTAMENT_STRICT3 = b"""\
bazaar testament version 3 strict
revision-id: test@user-3
committer: test@user
timestamp: 1129025493
timezone: 36000
parents:
  test@user-2
message:
  revision with properties
inventory:
  directory . TREE_ROT test@user-1 no
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
  directory src src-id test@user-2 no
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
properties:
  branch-nick:
    test branch
  empty:
  flavor:
    sour cherry
    cream cheese
  size:
    medium
"""


REV_3_TESTAMENT = b"""\
bazaar-ng testament version 1
revision-id: test@user-3
committer: test@user
timestamp: 1129025493
timezone: 36000
parents:
  test@user-2
message:
  add symlink
inventory:
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
  symlink link link-id wibble/linktarget
  directory src src-id
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
properties:
  branch-nick:
    test branch
"""


REV_3_TESTAMENT_STRICT = b"""\
bazaar-ng testament version 2.1
revision-id: test@user-3
committer: test@user
timestamp: 1129025493
timezone: 36000
parents:
  test@user-2
message:
  add symlink
inventory:
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
  symlink link link-id wibble/linktarget test@user-3 no
  directory src src-id test@user-2 no
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
properties:
  branch-nick:
    test branch
"""


REV_3_TESTAMENT_STRICT3 = b"""\
bazaar testament version 3 strict
revision-id: test@user-3
committer: test@user
timestamp: 1129025493
timezone: 36000
parents:
  test@user-2
message:
  add symlink
inventory:
  directory . TREE_ROT test@user-1 no
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
  symlink link link-id wibble/linktarget test@user-3 no
  directory src src-id test@user-2 no
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
properties:
  branch-nick:
    test branch
"""


SAMPLE_UNICODE_TESTAMENT = """\
bazaar-ng testament version 1
revision-id: test@user-3
committer: Erik B\xe5gfors <test@user>
timestamp: 1129025493
timezone: 36000
parents:
  test@user-2
message:
  non-ascii commit \N{COPYRIGHT SIGN} me
inventory:
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
  directory src src-id
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
properties:
  branch-nick:
    test branch
  uni:
    \xb5
"""


SAMPLE_UNICODE_TESTAMENT_STRICT = """\
bazaar-ng testament version 2.1
revision-id: test@user-3
committer: Erik B\xe5gfors <test@user>
timestamp: 1129025493
timezone: 36000
parents:
  test@user-2
message:
  non-ascii commit \N{COPYRIGHT SIGN} me
inventory:
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
  directory src src-id test@user-2 no
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
properties:
  branch-nick:
    test branch
  uni:
    \xb5
"""


SAMPLE_UNICODE_TESTAMENT_STRICT3 = """\
bazaar testament version 3 strict
revision-id: test@user-3
committer: Erik B\xe5gfors <test@user>
timestamp: 1129025493
timezone: 36000
parents:
  test@user-2
message:
  non-ascii commit \N{COPYRIGHT SIGN} me
inventory:
  directory . TREE_ROT test@user-1 no
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
  directory src src-id test@user-2 no
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
properties:
  branch-nick:
    test branch
  uni:
    \xb5
"""


texts = {
    Testament: {
        "rev_1": REV_1_TESTAMENT,
        "rev_1_short": REV_1_SHORT,
        "rev_2": REV_2_TESTAMENT,
        "rev_2_short": REV_2_SHORT,
        "rev_3": REV_3_TESTAMENT,
        "rev_props": REV_PROPS_TESTAMENT,
        "sample_unicode": SAMPLE_UNICODE_TESTAMENT,
    },
    StrictTestament: {
        "rev_1": REV_1_STRICT_TESTAMENT,
        "rev_1_short": REV_1_SHORT_STRICT,
        "rev_2": REV_2_STRICT_TESTAMENT,
        "rev_2_short": REV_2_SHORT_STRICT,
        "rev_3": REV_3_TESTAMENT_STRICT,
        "rev_props": REV_PROPS_TESTAMENT_STRICT,
        "sample_unicode": SAMPLE_UNICODE_TESTAMENT_STRICT,
    },
    StrictTestament3: {
        "rev_1": REV_1_STRICT_TESTAMENT3,
        "rev_1_short": REV_1_SHORT_STRICT3,
        "rev_2": REV_2_STRICT_TESTAMENT3,
        "rev_2_short": REV_2_SHORT_STRICT3,
        "rev_3": REV_3_TESTAMENT_STRICT3,
        "rev_props": REV_PROPS_TESTAMENT_STRICT3,
        "sample_unicode": SAMPLE_UNICODE_TESTAMENT_STRICT3,
    },
}
