1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
# Copyright (c) 2023 Egor Tensin <Egor.Tensin@gmail.com>
# This file is part of the "cimple" project.
# For details, see https://github.com/egor-tensin/cimple.
# Distributed under the MIT License.
import logging
import os
import shutil
from .process import Process
class Repo:
BRANCH = 'main'
def __init__(self, path):
self.path = os.path.abspath(path)
os.makedirs(path, exist_ok=True)
self.run('git', 'init', '-q', f'--initial-branch={Repo.BRANCH}')
self.run('git', 'config', 'user.name', 'Test User')
self.run('git', 'config', 'user.email', 'test@example.com')
def run(self, *args, **kwargs):
Process.run(*args, cwd=self.path, **kwargs)
class TestRepo(Repo):
# Prevent Pytest from discovering test cases in this class:
__test__ = False
DATA_DIR = 'test_repo'
CI_SCRIPT = 'ci.sh'
OUTPUT_DIR = 'output'
@staticmethod
def get_ci_script():
return os.path.join(os.path.dirname(__file__), TestRepo.DATA_DIR, TestRepo.CI_SCRIPT)
def __init__(self, path):
super().__init__(path)
shutil.copy(self.get_ci_script(), self.path)
self.run('git', 'add', '.')
self.run('git', 'commit', '-q', '-m', 'add CI script')
self.output_dir = os.path.join(self.path, TestRepo.OUTPUT_DIR)
os.makedirs(self.output_dir, exist_ok=True)
def count_ci_output_files(self):
return len([name for name in os.listdir(self.output_dir) if os.path.isfile(os.path.join(self.output_dir, name))])
def output_matches(self, output):
return output.decode().startswith('A CI run happened at ')
|
