# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

from __future__ import absolute_import

import mozunit

from mozversioncontrol import get_repository_object

STEPS = {
    "hg": [
        """
        echo "bar" >> bar
        echo "baz" > foo
        """,
    ],
    "git": [
        """
        echo "bar" >> bar
        echo "baz" > foo
        """,
    ],
}


def test_commit(repo):
    vcs = get_repository_object(repo.strpath)
    assert vcs.working_directory_clean()

    # Modify both foo and bar
    next(repo.step)
    assert not vcs.working_directory_clean()

    # Commit just bar
    vcs.commit(
        "Modify bar\n\nbut not baz",
        "Testing McTesterson <test@example.org>",
        "2017-07-14 02:40:00 UTC",
        ["bar"],
    )

    # We only committed bar, so foo is still keeping the working dir dirty
    assert not vcs.working_directory_clean()

    if repo.vcs == "git":
        log_cmd = ["log", "-1", "--format=%an,%ae,%at,%B"]
        patch_cmd = ["log", "-1", "-p"]
    else:
        log_cmd = [
            "log",
            "-l",
            "1",
            "-T",
            '{person(author)},{email(author)},{date(localdate(date),"%s")},{desc}',
        ]
        patch_cmd = ["log", "-l", "1", "-p"]

    # Verify commit metadata (we rstrip to normalize trivial git/hg differences)
    log = vcs._run(*log_cmd).rstrip()
    assert (
        log
        == "Testing McTesterson,test@example.org,1500000000,Modify bar\n\nbut not baz"
    )

    # Verify only the intended file was added to the commit
    patch = vcs._run(*patch_cmd)
    diffs = [line for line in patch.splitlines() if "diff --git" in line]
    assert len(diffs) == 1
    assert diffs[0] == "diff --git a/bar b/bar"


if __name__ == "__main__":
    mozunit.main()