mirror of
https://chromium.googlesource.com/chromium/tools/depot_tools.git
synced 2026-01-11 18:51:29 +00:00
569c95044f
Deduplicates the various `prpc call` subprocess.run calls throughout mcp/ in favor of a new run_prpc_call helper. Also makes a few drive-by cleanup changes that were caught by WIP presubmit checks. Change-Id: I3e401cb959ba96f2224ea1a7f7f58795330f643a Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/6847120 Commit-Queue: Struan Shrimpton <sshrimp@google.com> Auto-Submit: Brian Sheedy <bsheedy@chromium.org> Reviewed-by: Struan Shrimpton <sshrimp@google.com>
98 lines
4.1 KiB
Python
98 lines
4.1 KiB
Python
# Copyright 2025 The Chromium Authors
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
"""Tests for git_cl tools."""
|
|
|
|
import os
|
|
import pathlib
|
|
import subprocess
|
|
import sys
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
sys.path.insert(
|
|
0,
|
|
os.path.abspath(
|
|
pathlib.Path(__file__).resolve().parent.parent.joinpath(
|
|
pathlib.Path('infra_lib'))))
|
|
import git_cl
|
|
|
|
|
|
class GitClTest(unittest.IsolatedAsyncioTestCase):
|
|
|
|
def setUp(self):
|
|
self.mock_context = mock.AsyncMock()
|
|
self.mock_context.info = mock.AsyncMock()
|
|
self.checkout = '/path/to/checkout'
|
|
|
|
@mock.patch('subprocess.run')
|
|
async def test_try_builder_results_success(self, mock_subprocess_run):
|
|
expected_output = '{"builds": []}'
|
|
mock_subprocess_run.return_value = subprocess.CompletedProcess(
|
|
args=[], returncode=0, stdout=expected_output, stderr='')
|
|
|
|
output = await git_cl.try_builder_results(self.mock_context,
|
|
self.checkout)
|
|
|
|
self.assertEqual(output, expected_output)
|
|
expected_command = ["git", "cl", "try-results", "--json=-"]
|
|
mock_subprocess_run.assert_called_once_with(expected_command,
|
|
capture_output=True,
|
|
check=True,
|
|
text=True,
|
|
cwd=self.checkout)
|
|
|
|
@mock.patch('subprocess.run')
|
|
async def test_get_current_changes_success(self, mock_subprocess_run):
|
|
expected_output = 'diff --git a/file.txt b/file.txt'
|
|
mock_subprocess_run.return_value = subprocess.CompletedProcess(
|
|
args=[], returncode=0, stdout=expected_output, stderr='')
|
|
|
|
output = await git_cl.get_current_changes(self.mock_context,
|
|
self.checkout)
|
|
|
|
self.assertEqual(output, expected_output)
|
|
expected_command = ["git", "cl", "diff"]
|
|
mock_subprocess_run.assert_called_once_with(expected_command,
|
|
capture_output=True,
|
|
check=True,
|
|
text=True,
|
|
cwd=self.checkout)
|
|
|
|
@mock.patch('subprocess.run')
|
|
async def test_format_checkout_success(self, mock_subprocess_run):
|
|
expected_output = 'Formatted 1 file'
|
|
mock_subprocess_run.return_value = subprocess.CompletedProcess(
|
|
args=[], returncode=0, stdout=expected_output, stderr='')
|
|
|
|
output = await git_cl.format_checkout(self.mock_context, self.checkout)
|
|
|
|
self.assertEqual(output, expected_output)
|
|
expected_command = ["git", "cl", "format"]
|
|
mock_subprocess_run.assert_called_once_with(expected_command,
|
|
capture_output=True,
|
|
check=True,
|
|
text=True,
|
|
cwd=self.checkout)
|
|
|
|
@mock.patch('subprocess.run')
|
|
async def test_upload_change_list_success(self, mock_subprocess_run):
|
|
expected_output = 'CL uploaded'
|
|
mock_subprocess_run.return_value = subprocess.CompletedProcess(
|
|
args=[], returncode=0, stdout=expected_output, stderr='')
|
|
|
|
output = await git_cl.upload_change_list(self.mock_context,
|
|
self.checkout)
|
|
|
|
self.assertEqual(output, expected_output)
|
|
expected_command = ["git", "cl", "upload", "-f"]
|
|
mock_subprocess_run.assert_called_once_with(expected_command,
|
|
capture_output=True,
|
|
check=True,
|
|
text=True,
|
|
cwd=self.checkout)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|