From 5579e1d1bc585191b83d7909c0215608ed23e097 Mon Sep 17 00:00:00 2001
From: Jenn Mueng <30991498+jennmueng@users.noreply.github.com>
Date: Thu, 10 Oct 2024 15:05:18 -0700
Subject: [PATCH] This step creates a new test file `test_repo_client.py` in
 the `tests/seer/automation/codebase/` directory. The file contains several
 unit tests that cover the changes made to the `RepoClient` class and related
 functions:

1. `test_repo_client_type_enum`: Verifies that the new `RepoClientType` enum has the correct values.
2. `test_repo_client_from_repo_definition`: Tests the `from_repo_definition` method of `RepoClient` with different `RepoClientType` values.
3. `test_repo_client_init`: Tests the initialization of the `RepoClient` class.
4. `test_get_read_app_credentials`: Tests the `get_read_app_credentials` function.
5. `test_get_write_app_credentials`: Tests the `get_write_app_credentials` function.
6. `test_get_codecov_unit_test_app_credentials`: Tests the new `get_codecov_unit_test_app_credentials` function.

These tests use pytest fixtures and mocking to isolate the components being tested and verify their behavior with the new `RepoClientType` enum and associated changes.
---
 .../automation/codebase/test_repo_client.py   | 88 +++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 tests/seer/automation/codebase/test_repo_client.py

diff --git a/tests/seer/automation/codebase/test_repo_client.py b/tests/seer/automation/codebase/test_repo_client.py
new file mode 100644
index 00000000..b1ba8837
--- /dev/null
+++ b/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="git@github.com: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"
\ No newline at end of file