port-ocean 0.15.0__py3-none-any.whl → 0.15.2__py3-none-any.whl

port_ocean/core/models.py

@@ -8,12 +8,23 @@ from pydantic.fields import Field
 
 class Runtime(Enum):
     Saas = "Saas"
-    SaasOauth = "SaasOauth"
     OnPrem = "OnPrem"
 
     @property
     def is_saas_runtime(self) -> bool:
-        return self in [Runtime.Saas, Runtime.SaasOauth]
+        return self in [Runtime.Saas]
+
+    def is_installation_type_compatible(self, installation_type: str) -> bool:
+        """
+        Check if the installation type is compatible with the runtime
+
+        if the runtime is Saas, the installation type should start with Saas
+        else the installation type should be OnPrem
+        """
+        return (
+            self.value == Runtime.Saas.value
+            and installation_type.startswith(self.value)
+        ) or installation_type == self.value
 
 
 class Entity(BaseModel):

port_ocean/core/utils.py

@@ -42,10 +42,12 @@ async def validate_integration_runtime(
     current_integration = await port_client.get_current_integration(
         should_raise=False, should_log=False
     )
-    current_runtime = current_integration.get("installationType", "OnPrem")
-    if current_integration and current_runtime != requested_runtime.value:
+    current_installation_type = current_integration.get("installationType", "OnPrem")
+    if current_integration and not requested_runtime.is_installation_type_compatible(
+        current_installation_type
+    ):
         raise IntegrationRuntimeException(
-            f"Invalid Runtime! Requested to run existing {current_runtime} integration in {requested_runtime} runtime."
+            f"Invalid Runtime! Requested to run existing {current_installation_type} integration in {requested_runtime} runtime."
         )
 
 

port_ocean/tests/core/test_utils.py

@@ -0,0 +1,73 @@
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from port_ocean.core.utils import validate_integration_runtime
+from port_ocean.clients.port.client import PortClient
+from port_ocean.core.models import Runtime
+from port_ocean.tests.helpers.port_client import get_port_client_for_integration
+from port_ocean.exceptions.core import IntegrationRuntimeException
+
+
+class TestValidateIntegrationRuntime:
+
+    @pytest.mark.asyncio
+    @pytest.mark.parametrize(
+        "requested_runtime, installation_type, should_raise",
+        [
+            (Runtime.Saas, "Saas", False),
+            (Runtime.Saas, "SaasOauth2", False),
+            (Runtime.Saas, "OnPrem", True),
+            (Runtime.OnPrem, "OnPrem", False),
+            (Runtime.OnPrem, "SaasOauth2", True),
+        ],
+    )
+    @patch.object(PortClient, "get_current_integration", new_callable=AsyncMock)
+    async def test_validate_integration_runtime(
+        self,
+        mock_get_current_integration: AsyncMock,
+        requested_runtime: Runtime,
+        installation_type: str,
+        should_raise: bool,
+    ) -> None:
+        # Arrange
+        port_client = get_port_client_for_integration(
+            client_id="mock-client-id",
+            client_secret="mock-client-secret",
+            integration_identifier="mock-integration-identifier",
+            integration_type="mock-integration-type",
+            integration_version="mock-integration-version",
+            base_url="mock-base-url",
+        )
+
+        # Mock the return value of get_current_integration
+        mock_get_current_integration.return_value = {
+            "installationType": installation_type
+        }
+
+        # Act & Assert
+        if should_raise:
+            with pytest.raises(IntegrationRuntimeException):
+                await validate_integration_runtime(port_client, requested_runtime)
+        else:
+            await validate_integration_runtime(port_client, requested_runtime)
+
+        # Verify that get_current_integration was called once
+        mock_get_current_integration.assert_called_once()
+
+    @pytest.mark.parametrize(
+        "requested_runtime, installation_type, expected",
+        [
+            (Runtime.Saas, "SaasOauth2", True),
+            (Runtime.Saas, "OnPrem", False),
+            (Runtime.OnPrem, "OnPrem", True),
+            (Runtime.OnPrem, "SaasCloud", False),
+        ],
+    )
+    def test_runtime_installation_compatibility(
+        self, requested_runtime: Runtime, installation_type: str, expected: bool
+    ) -> None:
+        assert (
+            requested_runtime.is_installation_type_compatible(installation_type)
+            == expected
+        )

{port_ocean-0.15.0.dist-info → port_ocean-0.15.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: port-ocean
-Version: 0.15.0
+Version: 0.15.2
 Summary: Port Ocean is a CLI tool for managing your Port projects.
 Home-page: https://app.getport.io
 Keywords: ocean,port-ocean,port

{port_ocean-0.15.0.dist-info → port_ocean-0.15.2.dist-info}/RECORD

@@ -102,9 +102,9 @@ port_ocean/core/integrations/mixins/handler.py,sha256=mZ7-0UlG3LcrwJttFbMe-R4xcO
 port_ocean/core/integrations/mixins/sync.py,sha256=B9fEs8faaYLLikH9GBjE_E61vo0bQDjIGQsQ1SRXOlA,3931
 port_ocean/core/integrations/mixins/sync_raw.py,sha256=BGS5EnZ2N3ifcAi94Wo-ZassSJ-_Se9eFJMpBDT7pNY,18841
 port_ocean/core/integrations/mixins/utils.py,sha256=7y1rGETZIjOQadyIjFJXIHKkQFKx_SwiP-TrAIsyyLY,2303
-port_ocean/core/models.py,sha256=5jnXFkSpS9BfvOy0KLWWvg96I3aXjMHmfQ_4lCBdz_k,1481
+port_ocean/core/models.py,sha256=71QIFHl-p401h2HnSDQ-aaLXhu6z3iHTwCBI0TewJos,1902
 port_ocean/core/ocean_types.py,sha256=3_d8-n626f1kWLQ_Jxw194LEyrOVupz05qs_Y1pvB-A,990
-port_ocean/core/utils.py,sha256=40UjRauRJO47WDSNn9bkCRD2bfhfB3e-dnOLULnuVzE,3631
+port_ocean/core/utils.py,sha256=QSRuF9wlhbOw6cELlDlek_UIX6ciIuKWml8QhBmHU_k,3703
 port_ocean/exceptions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 port_ocean/exceptions/api.py,sha256=TLmTMqn4uHGaHgZK8PMIJ0TVJlPB4iP7xl9rx7GtCyY,426
 port_ocean/exceptions/base.py,sha256=uY4DX7fIITDFfemCJDWpaZi3bD51lcANc5swpoNvMJA,46
@@ -130,6 +130,7 @@ port_ocean/tests/clients/port/mixins/test_entities.py,sha256=A9myrnkLhKSQrnOLv1Z
 port_ocean/tests/conftest.py,sha256=JXASSS0IY0nnR6bxBflhzxS25kf4iNaABmThyZ0mZt8,101
 port_ocean/tests/core/defaults/test_common.py,sha256=sR7RqB3ZYV6Xn6NIg-c8k5K6JcGsYZ2SCe_PYX5vLYM,5560
 port_ocean/tests/core/handlers/entity_processor/test_jq_entity_processor.py,sha256=Yv03P-LDcJCKZ21exiTFrcT1eu0zn6Z954dilxrb52Y,10842
+port_ocean/tests/core/test_utils.py,sha256=94940TerN38jy81ebLJ2Fzf5JJUaV9krnce75APCwmM,2641
 port_ocean/tests/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 port_ocean/tests/helpers/fake_port_api.py,sha256=9rtjC6iTQMfzWK6WipkDzzG0b1IIaRmvdJLOyV613vE,6479
 port_ocean/tests/helpers/fixtures.py,sha256=IQEplbHhRgjrAsZlnXrgSYA5YQEn25I9HgO3_Fjibxg,1481
@@ -151,8 +152,8 @@ port_ocean/utils/repeat.py,sha256=0EFWM9d8lLXAhZmAyczY20LAnijw6UbIECf5lpGbOas,32
 port_ocean/utils/signal.py,sha256=K-6kKFQTltcmKDhtyZAcn0IMa3sUpOHGOAUdWKgx0_E,1369
 port_ocean/utils/time.py,sha256=pufAOH5ZQI7gXvOvJoQXZXZJV-Dqktoj9Qp9eiRwmJ4,1939
 port_ocean/version.py,sha256=UsuJdvdQlazzKGD3Hd5-U7N69STh8Dq9ggJzQFnu9fU,177
-port_ocean-0.15.0.dist-info/LICENSE.md,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-port_ocean-0.15.0.dist-info/METADATA,sha256=GLoKGQDzcULrCpX8Q_2lFrExJPbl7qD8RclNc3zWNds,6673
-port_ocean-0.15.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-port_ocean-0.15.0.dist-info/entry_points.txt,sha256=F_DNUmGZU2Kme-8NsWM5LLE8piGMafYZygRYhOVtcjA,54
-port_ocean-0.15.0.dist-info/RECORD,,
+port_ocean-0.15.2.dist-info/LICENSE.md,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+port_ocean-0.15.2.dist-info/METADATA,sha256=xiclpeTv7njo_QsJHJVG3NF6g-PSh74WwovQQYjwX6w,6673
+port_ocean-0.15.2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+port_ocean-0.15.2.dist-info/entry_points.txt,sha256=F_DNUmGZU2Kme-8NsWM5LLE8piGMafYZygRYhOVtcjA,54
+port_ocean-0.15.2.dist-info/RECORD,,