Add Tests for PR#1266

tests/seer/automation/codebase/test_repo_client.py

@@ -0,0 +1,88 @@
+import pytest
+from unittest.mock import patch, MagicMock
+from seer.automation.codebase.repo_client import RepoClient, RepoClientType, get_read_app_credentials, get_write_app_credentials, get_codecov_unit_test_app_credentials
+from seer.automation.models import RepoDefinition
+
+@pytest.fixture
+def mock_repo_definition():
+ return RepoDefinition(
+ full_name="test/repo",
+ default_branch="main",
+ clone_url="https://github.com/test/repo.git",
+ ssh_url="[email protected]:test/repo.git",
+ external_id="123456",
+ )
+
+def test_repo_client_type_enum():
+ assert RepoClientType.READ == "read"
+ assert RepoClientType.WRITE == "write"
+ assert RepoClientType.CODECOV_UNIT_TEST == "codecov_unit_test"
+
+@patch('seer.automation.codebase.repo_client.get_read_app_credentials')
+@patch('seer.automation.codebase.repo_client.get_write_app_credentials')
+@patch('seer.automation.codebase.repo_client.get_codecov_unit_test_app_credentials')
+def test_repo_client_from_repo_definition(mock_codecov_creds, mock_write_creds, mock_read_creds, mock_repo_definition):
+ mock_read_creds.return_value = ("read_app_id", "read_private_key")
+ mock_write_creds.return_value = ("write_app_id", "write_private_key")
+ mock_codecov_creds.return_value = ("codecov_app_id", "codecov_private_key")
+
+ # Test READ type
+ client = RepoClient.from_repo_definition(mock_repo_definition, RepoClientType.READ)
+ mock_read_creds.assert_called_once()
+ assert client.github_auth.app_id == "read_app_id"
+ assert client.github_auth.private_key == "read_private_key"
+
+ # Test WRITE type
+ client = RepoClient.from_repo_definition(mock_repo_definition, RepoClientType.WRITE)
+ mock_write_creds.assert_called_once()
+ assert client.github_auth.app_id == "write_app_id"
+ assert client.github_auth.private_key == "write_private_key"
+
+ # Test CODECOV_UNIT_TEST type
+ client = RepoClient.from_repo_definition(mock_repo_definition, RepoClientType.CODECOV_UNIT_TEST)
+ mock_codecov_creds.assert_called_once()
+ assert client.github_auth.app_id == "codecov_app_id"
+ assert client.github_auth.private_key == "codecov_private_key"
+
+@patch('seer.automation.codebase.repo_client.Auth')
+@patch('seer.automation.codebase.repo_client.Github')
+def test_repo_client_init(mock_github, mock_auth, mock_repo_definition):
+ mock_auth.AppInstallationAuth.return_value = MagicMock()
+ mock_github.return_value = MagicMock()
+
+ client = RepoClient("app_id", "private_key", mock_repo_definition)
+
+ mock_auth.AppInstallationAuth.assert_called_once_with(
+ app_id="app_id",
+ private_key="private_key",
+ installation_id=mock_repo_definition.external_id,
+ )
+ mock_github.assert_called_once_with(auth=mock_auth.AppInstallationAuth.return_value)
+ assert client.repo_name == mock_repo_definition.full_name
+
+@patch('seer.automation.codebase.repo_client.AppConfig')
+def test_get_read_app_credentials(mock_app_config):
+ mock_app_config.return_value.GITHUB_READ_APP_ID = "read_app_id"
+ mock_app_config.return_value.GITHUB_READ_PRIVATE_KEY = "read_private_key"
+
+ app_id, private_key = get_read_app_credentials()
+ assert app_id == "read_app_id"
+ assert private_key == "read_private_key"
+
+@patch('seer.automation.codebase.repo_client.AppConfig')
+def test_get_write_app_credentials(mock_app_config):
+ mock_app_config.return_value.GITHUB_WRITE_APP_ID = "write_app_id"
+ mock_app_config.return_value.GITHUB_WRITE_PRIVATE_KEY = "write_private_key"
+
+ app_id, private_key = get_write_app_credentials()
+ assert app_id == "write_app_id"
+ assert private_key == "write_private_key"
+
+@patch('seer.automation.codebase.repo_client.AppConfig')
+def test_get_codecov_unit_test_app_credentials(mock_app_config):
+ mock_app_config.return_value.GITHUB_CODECOV_UNIT_TEST_APP_ID = "codecov_app_id"
+ mock_app_config.return_value.GITHUB_CODECOV_UNIT_TEST_PRIVATE_KEY = "codecov_private_key"
+
+ app_id, private_key = get_codecov_unit_test_app_credentials()
+ assert app_id == "codecov_app_id"
+ assert private_key == "codecov_private_key"