mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-16 14:55:47 +00:00
0ad2e9342d
Differential Revision: https://phabricator.services.mozilla.com/D1808 --HG-- rename : tools/tryselect/vcs.py => tools/tryselect/push.py extra : moz-landing-system : lando
72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
# 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, unicode_literals
|
|
|
|
import subprocess
|
|
|
|
import mozunit
|
|
import pytest
|
|
|
|
from mozversioncontrol import (
|
|
get_repository_object,
|
|
MissingVCSExtension,
|
|
)
|
|
|
|
|
|
def test_push_to_try(repo, monkeypatch):
|
|
commit_message = "commit message"
|
|
vcs = get_repository_object(repo.strpath)
|
|
|
|
captured_commands = []
|
|
|
|
def fake_run(*args, **kwargs):
|
|
captured_commands.append(args[0])
|
|
|
|
monkeypatch.setattr(subprocess, 'check_output', fake_run)
|
|
monkeypatch.setattr(subprocess, 'check_call', fake_run)
|
|
|
|
vcs.push_to_try(commit_message)
|
|
tool = vcs._tool
|
|
|
|
if repo.vcs == 'hg':
|
|
expected = [
|
|
(tool, 'push-to-try', '-m', commit_message),
|
|
(tool, 'revert', '-a'),
|
|
]
|
|
else:
|
|
expected = [
|
|
(tool, 'cinnabar', '--version'),
|
|
(tool, 'commit', '--allow-empty', '-m', commit_message),
|
|
(tool, 'push', 'hg::ssh://hg.mozilla.org/try',
|
|
'+HEAD:refs/heads/branches/default/tip'),
|
|
(tool, 'reset', 'HEAD~'),
|
|
]
|
|
|
|
for i, value in enumerate(captured_commands):
|
|
assert value == expected[i]
|
|
|
|
assert len(captured_commands) == len(expected)
|
|
|
|
|
|
def test_push_to_try_missing_extensions(repo, monkeypatch):
|
|
vcs = get_repository_object(repo.strpath)
|
|
|
|
orig = vcs._run
|
|
|
|
def cinnabar_raises(*args, **kwargs):
|
|
# Simulate not having git cinnabar
|
|
if args[0] == 'cinnabar':
|
|
raise subprocess.CalledProcessError(1, args)
|
|
return orig(*args, **kwargs)
|
|
|
|
monkeypatch.setattr(vcs, '_run', cinnabar_raises)
|
|
|
|
with pytest.raises(MissingVCSExtension):
|
|
vcs.push_to_try("commit message")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
mozunit.main()
|