148 lines
5.7 KiB
Python
148 lines
5.7 KiB
Python
"""Tests for the shared get_credentials() function in gitea_auth.py.
|
|
|
|
These test the credential parsing logic in isolation by mocking subprocess.Popen.
|
|
"""
|
|
import os
|
|
import sys
|
|
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
sys.path.insert(0, str(__import__("pathlib").Path(__file__).resolve().parent.parent))
|
|
import gitea_auth # noqa: E402
|
|
|
|
|
|
class TestGetCredentials(unittest.TestCase):
|
|
"""Test the get_credentials function that parses git credential fill output."""
|
|
|
|
def setUp(self):
|
|
self.old_env = os.environ.copy()
|
|
os.environ.clear()
|
|
os.environ["GITEA_USE_KEYCHAIN"] = "1"
|
|
self.old_configs = gitea_auth.DYNAMIC_CONFIGS.copy()
|
|
gitea_auth.DYNAMIC_CONFIGS.clear()
|
|
|
|
def tearDown(self):
|
|
os.environ.clear()
|
|
os.environ.update(self.old_env)
|
|
gitea_auth.DYNAMIC_CONFIGS.clear()
|
|
gitea_auth.DYNAMIC_CONFIGS.update(self.old_configs)
|
|
|
|
def _mock_popen(self, output_text):
|
|
"""Create a mock Popen that returns the given text from communicate()."""
|
|
mock_proc = MagicMock()
|
|
mock_proc.communicate.return_value = (output_text, "")
|
|
return mock_proc
|
|
|
|
@patch("gitea_auth.subprocess.Popen")
|
|
def test_parses_standard_output(self, mock_popen_cls):
|
|
mock_popen_cls.return_value = self._mock_popen(
|
|
"protocol=https\nhost=gitea.example.com\nusername=admin\npassword=s3cret\n"
|
|
)
|
|
user, password = gitea_auth.get_credentials("gitea.example.com")
|
|
self.assertEqual(user, "admin")
|
|
self.assertEqual(password, "s3cret")
|
|
|
|
@patch("gitea_auth.subprocess.Popen")
|
|
def test_handles_password_with_equals(self, mock_popen_cls):
|
|
# Tokens often contain '=' characters
|
|
mock_popen_cls.return_value = self._mock_popen(
|
|
"username=bot\npassword=abc=def=ghi\n"
|
|
)
|
|
user, password = gitea_auth.get_credentials("example.com")
|
|
self.assertEqual(user, "bot")
|
|
self.assertEqual(password, "abc=def=ghi")
|
|
|
|
@patch("gitea_auth.subprocess.Popen")
|
|
def test_empty_output_returns_empty(self, mock_popen_cls):
|
|
mock_popen_cls.return_value = self._mock_popen("")
|
|
user, password = gitea_auth.get_credentials("example.com")
|
|
self.assertEqual(user, "")
|
|
self.assertEqual(password, "")
|
|
|
|
@patch("gitea_auth.subprocess.Popen")
|
|
def test_missing_password_returns_empty(self, mock_popen_cls):
|
|
mock_popen_cls.return_value = self._mock_popen("username=admin\n")
|
|
user, password = gitea_auth.get_credentials("example.com")
|
|
self.assertEqual(user, "admin")
|
|
self.assertEqual(password, "")
|
|
|
|
@patch("gitea_auth.subprocess.Popen")
|
|
def test_sends_correct_stdin(self, mock_popen_cls):
|
|
mock_proc = self._mock_popen("username=u\npassword=p\n")
|
|
mock_popen_cls.return_value = mock_proc
|
|
|
|
gitea_auth.get_credentials("gitea.prgs.cc")
|
|
|
|
# Verify the correct input was sent to git credential fill
|
|
mock_proc.communicate.assert_called_once_with(
|
|
"protocol=https\nhost=gitea.prgs.cc\n\n"
|
|
)
|
|
|
|
def test_loads_credentials_from_env(self):
|
|
os.environ.clear()
|
|
os.environ["GITEA_USER_PRGS"] = "env_user"
|
|
os.environ["GITEA_PASS_PRGS"] = "env_pass"
|
|
user, password = gitea_auth.get_credentials("gitea.prgs.cc")
|
|
self.assertEqual(user, "env_user")
|
|
self.assertEqual(password, "env_pass")
|
|
|
|
def test_loads_credentials_from_generic_env(self):
|
|
os.environ.clear()
|
|
os.environ["GITEA_USER"] = "gen_user"
|
|
os.environ["GITEA_PASS"] = "gen_pass"
|
|
user, password = gitea_auth.get_credentials("gitea.example.com")
|
|
self.assertEqual(user, "gen_user")
|
|
self.assertEqual(password, "gen_pass")
|
|
|
|
def test_loads_credentials_from_dynamic_configs(self):
|
|
gitea_auth.DYNAMIC_CONFIGS["gitea.example.com"] = {
|
|
"GITEA_USER": "dynamic_user",
|
|
"GITEA_PASS": "dynamic_pass"
|
|
}
|
|
user, password = gitea_auth.get_credentials("gitea.example.com")
|
|
self.assertEqual(user, "dynamic_user")
|
|
self.assertEqual(password, "dynamic_pass")
|
|
|
|
|
|
class TestGetAuthHeader(unittest.TestCase):
|
|
"""Test the get_auth_header function."""
|
|
|
|
@patch("gitea_auth.get_credentials", return_value=("user", "pass"))
|
|
def test_returns_basic_header(self, _cred):
|
|
header = gitea_auth.get_auth_header("example.com")
|
|
self.assertIsNotNone(header)
|
|
self.assertTrue(header.startswith("Basic "))
|
|
|
|
@patch("gitea_auth.get_credentials", return_value=("", ""))
|
|
def test_returns_none_for_missing_creds(self, _cred):
|
|
header = gitea_auth.get_auth_header("example.com")
|
|
self.assertIsNone(header)
|
|
|
|
def test_returns_token_header_from_env(self):
|
|
with patch.dict(os.environ, {"GITEA_TOKEN_PRGS": "my_prgs_token"}):
|
|
header = gitea_auth.get_auth_header("gitea.prgs.cc")
|
|
self.assertEqual(header, "token my_prgs_token")
|
|
|
|
def test_returns_generic_token_header_from_env(self):
|
|
with patch.dict(os.environ, {"GITEA_TOKEN": "generic_token"}):
|
|
header = gitea_auth.get_auth_header("gitea.example.com")
|
|
self.assertEqual(header, "token generic_token")
|
|
|
|
def test_returns_token_from_dynamic_configs(self):
|
|
gitea_auth.DYNAMIC_CONFIGS["gitea.example.com"] = {
|
|
"GITEA_TOKEN": "dynamic_token"
|
|
}
|
|
header = gitea_auth.get_auth_header("gitea.example.com")
|
|
self.assertEqual(header, "token dynamic_token")
|
|
|
|
|
|
class TestRepoApiUrl(unittest.TestCase):
|
|
|
|
def test_url_format(self):
|
|
url = gitea_auth.repo_api_url("gitea.prgs.cc", "Org", "Repo")
|
|
self.assertEqual(url, "https://gitea.prgs.cc/api/v1/repos/Org/Repo")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|