From bcc50019cd101458f0dc211d70a1e1c143b1e700 Mon Sep 17 00:00:00 2001 From: Haoyu Yang Date: Mon, 1 Apr 2024 10:30:13 +0800 Subject: [PATCH] mock GITHUB_REPOSITORY --- tests/test_dev.py | 20 +++++++++++++------- 1 file changed, 13 insertions(+), 7 deletions(-) diff --git a/tests/test_dev.py b/tests/test_dev.py index 90326932..fa84f955 100644 --- a/tests/test_dev.py +++ b/tests/test_dev.py @@ -1,7 +1,7 @@ import unittest import warnings import datetime -from unittest.mock import MagicMock +from unittest.mock import MagicMock, patch import pytest from monty.dev import deprecated, install_excepthook, requires @@ -91,12 +91,18 @@ def classmethod_b(cls): def test_deprecated_deadline(self, monkeypatch): with pytest.raises(DeprecationWarning): - monkeypatch.setenv("CI", "true") - monkeypatch.setenv("GITHUB_REPOSITORY", "materialsvirtuallab/monty") - - @deprecated(deadline=(2000, 1, 1)) - def func_old(): - pass + monkeypatch.setenv("CI", "true") # mock CI env + + with patch("subprocess.run") as mock_run: + # Mock "GITHUB_REPOSITORY" + monkeypatch.setenv("GITHUB_REPOSITORY", "TESTOWNER/TESTREPO") + mock_run.return_value.stdout.decode.return_value = ( + "git@github.com:TESTOWNER/TESTREPO.git" + ) + + @deprecated(deadline=(2000, 1, 1)) + def func_old(): + pass def test_deprecated_deadline_no_warn(self, monkeypatch): """Test cases where no warning should be raised."""