You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
69 lines
2.2 KiB
Python
69 lines
2.2 KiB
Python
#!/usr/bin/env vpython3
|
|
# coding=utf-8
|
|
# Copyright 2020 The Chromium Authors. All rights reserved.
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
"""Unit tests for gerrit_client.py."""
|
|
|
|
import logging
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
if sys.version_info.major == 2:
|
|
from StringIO import StringIO
|
|
import mock
|
|
else:
|
|
from io import StringIO
|
|
from unittest import mock
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
import gerrit_client
|
|
import gerrit_util
|
|
|
|
|
|
class TestGerritClient(unittest.TestCase):
|
|
@mock.patch('gerrit_util.GetGerritBranch', return_value='')
|
|
def test_branch_info(self, util_mock):
|
|
gerrit_client.main([
|
|
'branchinfo', '--host', 'https://example.org/foo', '--project',
|
|
'projectname', '--branch', 'branchname'
|
|
])
|
|
util_mock.assert_called_once_with('example.org', 'projectname',
|
|
'branchname')
|
|
|
|
@mock.patch('gerrit_util.CreateGerritBranch', return_value='')
|
|
def test_branch(self, util_mock):
|
|
gerrit_client.main([
|
|
'branch', '--host', 'https://example.org/foo', '--project',
|
|
'projectname', '--branch', 'branchname', '--commit', 'commitname'
|
|
])
|
|
util_mock.assert_called_once_with('example.org', 'projectname',
|
|
'branchname', 'commitname')
|
|
|
|
@mock.patch('gerrit_util.QueryChanges', return_value='')
|
|
def test_changes(self, util_mock):
|
|
gerrit_client.main([
|
|
'changes', '--host', 'https://example.org/foo', '-p', 'foo=bar', '-p',
|
|
'baz=qux', '--limit', '10', '--start', '20', '-o', 'op1', '-o', 'op2'
|
|
])
|
|
util_mock.assert_called_once_with(
|
|
'example.org', [('foo', 'bar'), ('baz', 'qux')],
|
|
limit=10,
|
|
start=20,
|
|
o_params=['op1', 'op2'])
|
|
|
|
@mock.patch('gerrit_util.AbandonChange', return_value='')
|
|
def test_abandon(self, util_mock):
|
|
gerrit_client.main([
|
|
'abandon', '--host', 'https://example.org/foo', '-c', '1', '-m', 'bar'
|
|
])
|
|
util_mock.assert_called_once_with('example.org', 1, 'bar')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
logging.basicConfig(
|
|
level=logging.DEBUG if '-v' in sys.argv else logging.ERROR)
|
|
unittest.main()
|