dda-devops-build/src/test/python/release_mixin/mock_infrastructure.py

from mock_domain import MockRelease, MockVersion
from mock_infrastructure_api import MockGitApi
from src.main.python.ddadevops.release_mixin.domain import ReleaseType

class MockVersionRepository():

    def __init__(self):
        self.file = None
        self.file_handler = None
        self.write_file_count = 0
    
    def load_file(self):
        pass

    def write_file(self, version_string):
        self.write_file_count += 1
        pass
    
    def parse_file(self):
        pass

    def get_version(self) -> MockVersion:
        return MockVersion()

class MockReleaseTypeRepository():
    def __init__(self, mock_git_api: MockGitApi):
        self.git_api = mock_git_api

    def get_release_type(self):
        return ReleaseType.MINOR

class MockReleaseRepository():
    def __init__(self, version_repository: MockVersionRepository, release_type_repository: MockReleaseTypeRepository, main_branch: str):
        self.version_repository = version_repository
        self.release_type_repository = release_type_repository
        self.main_branch = main_branch
        self.get_release_count = 0

    def get_release(self) -> MockRelease:
        self.get_release_count += 1
        return MockRelease(self.release_type_repository.get_release_type(), self.version_repository.get_version(), self.main_branch)