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.
60 lines
1.9 KiB
Python
60 lines
1.9 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,
|
|
config: dict[str, list[str]] | None = None,
|
|
branchref: str | None = None):
|
|
"""Installs fakes/mocks for scm.GIT so that:
|
|
|
|
* Initial git config (local scope) is set to `config`.
|
|
* 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 - remove `config` - have callers just directly call SetConfig with
|
|
# whatever config state they need.
|
|
# 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, local_state=config)),
|
|
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)
|