57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
# Copyright (c) 2024 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.
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
import threading
|
|
|
|
from unittest import mock
|
|
import unittest
|
|
|
|
# This is to be able to import scm from the root of the repo.
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
import scm
|
|
|
|
|
|
def GIT(test: unittest.TestCase,
|
|
*,
|
|
branchref: str | None = None):
|
|
"""Installs fakes/mocks for scm.GIT so that:
|
|
|
|
* GetBranch will just return a fake branchname starting with the value of
|
|
branchref.
|
|
* git_new_branch.create_new_branch will be mocked to update the value
|
|
returned by GetBranch.
|
|
|
|
NOTE: The dependency on git_new_branch.create_new_branch seems pretty
|
|
circular - this functionality should probably move to scm.GIT?
|
|
"""
|
|
# TODO - add `system_config` - this will be configuration which exists at
|
|
# the 'system installation' level and is immutable.
|
|
|
|
_branchref = [branchref or 'refs/heads/main']
|
|
|
|
global_lock = threading.Lock()
|
|
global_state = {}
|
|
|
|
def _newBranch(branchref):
|
|
_branchref[0] = branchref
|
|
|
|
patches: list[mock._patch] = [
|
|
mock.patch('scm.GIT._new_config_state',
|
|
side_effect=lambda _: scm.GitConfigStateTest(
|
|
global_lock, global_state)),
|
|
mock.patch('scm.GIT.GetBranchRef', side_effect=lambda _: _branchref[0]),
|
|
mock.patch('git_new_branch.create_new_branch', side_effect=_newBranch)
|
|
]
|
|
|
|
for p in patches:
|
|
p.start()
|
|
test.addCleanup(p.stop)
|
|
|
|
test.addCleanup(scm.GIT.drop_config_cache)
|