Files
swift-mirror/utils/update_checkout/tests/test_update_worktree.py

80 lines
2.5 KiB
Python

# ===--- test_clone.py ----------------------------------------------------===#
#
# This source file is part of the Swift.org open source project
#
# Copyright (c) 2014 - 2020 Apple Inc. and the Swift project authors
# Licensed under Apache License v2.0 with Runtime Library Exception
#
# See https:#swift.org/LICENSE.txt for license information
# See https:#swift.org/CONTRIBUTORS.txt for the list of Swift project authors
#
# ===----------------------------------------------------------------------===#
import os
from . import scheme_mock
from .scheme_mock import call_quietly
WORKTREE_NAME = "feature"
def path_for_worktree(workspace_path, worktree_name):
return os.path.join(workspace_path, worktree_name)
def setup_worktree(workspace_path, local_path, worktree_name):
worktree_path = path_for_worktree(workspace_path, worktree_name)
os.makedirs(worktree_path)
for project in os.listdir(local_path):
local_project_path = os.path.join(local_path, project)
worktree_project_path = os.path.join(worktree_path, project)
call_quietly(
["git", "-C", local_project_path, "worktree", "add", worktree_project_path]
)
def teardown_worktree(workspace_path, local_path, worktree_name):
worktree_path = path_for_worktree(workspace_path, worktree_name)
for project in os.listdir(local_path):
local_project_path = os.path.join(local_path, project)
worktree_project_path = os.path.join(worktree_path, project)
call_quietly(
[
"git",
"-C",
local_project_path,
"worktree",
"remove",
worktree_project_path,
]
)
class WorktreeTestCase(scheme_mock.SchemeMockTestCase):
def __init__(self, *args, **kwargs):
super(WorktreeTestCase, self).__init__(*args, **kwargs)
def test_worktree(self):
self.call(
[
self.update_checkout_path,
"--config",
self.config_path,
"--source-root",
self.worktree_path,
"--scheme",
"main",
]
)
def setUp(self):
super(WorktreeTestCase, self).setUp()
self.worktree_path = os.path.join(self.workspace, WORKTREE_NAME)
setup_worktree(self.workspace, self.local_path, WORKTREE_NAME)
def tearDown(self):
teardown_worktree(self.workspace, self.local_path, WORKTREE_NAME)
super(WorktreeTestCase, self).tearDown()