1 files changed, 57 insertions, 0 deletions
diff --git a/python/mozversioncontrol/test/test_branch.py b/python/mozversioncontrol/test/test_branch.py
new file mode 100644
index 0000000000..7d211f18e8
--- /dev/null
+++ b/python/mozversioncontrol/test/test_branch.py
@@ -0,0 +1,57 @@
+# 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/.
+
+import mozunit
+import pytest
+from looseversion import LooseVersion
+
+from mozversioncontrol import get_repository_object
+
+STEPS = {
+    "hg": [
+        """
+        hg bookmark test
+        """,
+        """
+        echo "bar" > foo
+        hg commit -m "second commit"
+        """,
+    ],
+    "git": [
+        """
+        git checkout -b test
+        """,
+        """
+        echo "bar" > foo
+        git commit -a -m "second commit"
+        """,
+    ],
+}
+
+
+def test_branch(repo):
+    vcs = get_repository_object(repo.dir)
+    if vcs.name == "git" and LooseVersion(vcs.tool_version) < LooseVersion("2.22.0"):
+        pytest.xfail("`git branch --show-current` not implemented yet")
+
+    if vcs.name == "git":
+        assert vcs.branch == "master"
+    else:
+        assert vcs.branch is None
+
+    repo.execute_next_step()
+    assert vcs.branch == "test"
+
+    repo.execute_next_step()
+    assert vcs.branch == "test"
+
+    vcs.update(vcs.head_ref)
+    assert vcs.branch is None
+
+    vcs.update("test")
+    assert vcs.branch == "test"
+
+
+if __name__ == "__main__":
+    mozunit.main()