aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/test/py/test_repo.py
blob: 2c995e1142cba25af27832dfe48d82dc4e0f8b49 (plain) (blame)
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
52
53
54
# 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 pytest

from lib.process import LoggingEvent


class LoggingEventRunComplete(LoggingEvent):
    def __init__(self, target):
        self.counter = 0
        self.target = target
        super().__init__(timeout=60)

    def log_line_matches(self, line):
        return 'Received a "run finished" message from worker' in line

    def set(self):
        self.counter += 1
        if self.counter == self.target:
            super().set()


def _test_repo_internal(server_and_workers, test_repo, client, numof_runs):
    server, workers = server_and_workers

    event = LoggingEventRunComplete(numof_runs)
    # Count the number of times the server receives the "run complete" message.
    server.logger.add_event(event)

    for i in range(numof_runs):
        client.run('run', test_repo.path, 'HEAD')

    event.wait()
    assert numof_runs == test_repo.count_ci_output_files()


def test_repo(server_and_workers, test_repo, client):
    _test_repo_internal(server_and_workers, test_repo, client, 1)


def test_repo_2(server_and_workers, test_repo, client):
    _test_repo_internal(server_and_workers, test_repo, client, 2)


def test_repo_10(server_and_workers, test_repo, client):
    _test_repo_internal(server_and_workers, test_repo, client, 10)


@pytest.mark.stress
def test_repo_2000(server_and_workers, test_repo, client):
    _test_repo_internal(server_and_workers, test_repo, client, 2000)