From e2a989732527d11dec47a457fef66f6647c2c6e7 Mon Sep 17 00:00:00 2001 From: Gagan Trivedi Date: Mon, 7 Apr 2025 16:03:48 +0530 Subject: [PATCH 1/2] feat(utils/is_oss): Add a function to check deployment mode --- src/common/core/utils.py | 4 ++++ tests/unit/common/core/test_utils.py | 22 ++++++++++++++++++++++ 2 files changed, 26 insertions(+) diff --git a/src/common/core/utils.py b/src/common/core/utils.py index 4b785b4..cdf8083 100644 --- a/src/common/core/utils.py +++ b/src/common/core/utils.py @@ -37,6 +37,10 @@ def is_saas() -> bool: return pathlib.Path("./SAAS_DEPLOYMENT").exists() +def is_oss() -> bool: + return not (is_enterprise() or is_saas()) + + @lru_cache() def has_email_provider() -> bool: match settings.EMAIL_BACKEND: diff --git a/tests/unit/common/core/test_utils.py b/tests/unit/common/core/test_utils.py index d715446..59cf872 100644 --- a/tests/unit/common/core/test_utils.py +++ b/tests/unit/common/core/test_utils.py @@ -11,6 +11,7 @@ has_email_provider, is_enterprise, is_saas, + is_oss, ) pytestmark = pytest.mark.django_db @@ -25,6 +26,27 @@ def clear_lru_caches() -> Generator[None, None, None]: is_saas.cache_clear() +def test__is_oss_for_enterprise_returns_false(fs: FakeFilesystem) -> None: + # Given + fs.create_file("./ENTERPRISE_VERSION") + + # Then + assert is_oss() is False + + +def test__is_oss_for_saas_returns_false(fs: FakeFilesystem) -> None: + # Given + fs.create_file("./SAAS_DEPLOYMENT") + + # Then + assert is_oss() is False + + +def test__is_oss_for_oss_returns_true(fs: FakeFilesystem) -> None: + # Then + assert is_oss() is True + + def test_get_version_info(fs: FakeFilesystem) -> None: # Given expected_manifest_contents = { From f8a2270be73a435d8d878799b8581ffe45c3e256 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 7 Apr 2025 10:35:56 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- tests/unit/common/core/test_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/common/core/test_utils.py b/tests/unit/common/core/test_utils.py index 59cf872..26cf3fe 100644 --- a/tests/unit/common/core/test_utils.py +++ b/tests/unit/common/core/test_utils.py @@ -10,8 +10,8 @@ get_version_info, has_email_provider, is_enterprise, - is_saas, is_oss, + is_saas, ) pytestmark = pytest.mark.django_db