port-ocean 0.30.1__py3-none-any.whl → 0.30.2__py3-none-any.whl

port_ocean/core/defaults/initialize.py

@@ -1,5 +1,5 @@
 import asyncio
-from typing import Type, Any
+from typing import Any, Type
 
 import httpx
 from loguru import logger
@@ -114,12 +114,67 @@ async def _initialize_required_integration_settings(
         )
 
 
+async def _mapped_blueprints_exist(
+    port_client: PortClient,
+    defaults: Defaults | None = None,
+    has_provision_feature_flag: bool = False,
+) -> bool:
+    if not defaults:
+        return True
+
+    integration = await port_client.get_current_integration(
+        should_log=False,
+        should_raise=False,
+        has_provision_feature_flag=has_provision_feature_flag,
+    )
+    integration_config = integration.get("config", {})
+    resources = integration_config.get("resources", [])
+
+    if not isinstance(resources, list):
+        return True
+
+    mapped_blueprints = []
+    for resource in resources:
+        blueprint = (
+            resource.get("port", {})
+            .get("entity", {})
+            .get("mappings", {})
+            .get("blueprint")
+        )
+        if blueprint:
+            if (
+                isinstance(blueprint, str)
+                and blueprint.startswith('"')
+                and blueprint.endswith('"')
+            ):
+                blueprint = blueprint.strip('"')
+            mapped_blueprints.append({"identifier": blueprint})
+
+    if not mapped_blueprints:
+        return True
+
+    existing_blueprints, _ = await gather_and_split_errors_from_results(
+        [
+            port_client.get_blueprint(blueprint["identifier"], should_log=False)
+            for blueprint in mapped_blueprints
+        ],
+        lambda item: isinstance(item, Blueprint),
+    )
+
+    if len(existing_blueprints) != len(mapped_blueprints):
+        return False
+
+    return True
+
+
 async def _create_resources(
     port_client: PortClient,
     defaults: Defaults | None = None,
+    has_provision_feature_flag: bool = False,
 ) -> None:
     if not defaults:
         return
+
     creation_stage, *blueprint_patches = deconstruct_blueprints_to_creation_steps(
         defaults.blueprints
     )
@@ -132,7 +187,13 @@ async def _create_resources(
         lambda item: isinstance(item, Blueprint),
     )
 
-    if blueprints_results:
+    mapped_blueprints_exist = await _mapped_blueprints_exist(
+        port_client,
+        defaults,
+        has_provision_feature_flag,
+    )
+
+    if blueprints_results or mapped_blueprints_exist:
         logger.info(
             f"Blueprints already exist: {[result.identifier for result in blueprints_results]}. Skipping integration default creation..."
         )
@@ -184,15 +245,16 @@ async def _create_resources(
             )
         )
 
-        created_scorecards, scorecards_errors = (
-            await gather_and_split_errors_from_results(
-                (
-                    port_client.create_scorecard(
-                        blueprint_scorecards["blueprint"], action, should_log=False
-                    )
-                    for blueprint_scorecards in defaults.scorecards
-                    for action in blueprint_scorecards["data"]
+        (
+            created_scorecards,
+            scorecards_errors,
+        ) = await gather_and_split_errors_from_results(
+            (
+                port_client.create_scorecard(
+                    blueprint_scorecards["blueprint"], action, should_log=False
                 )
+                for blueprint_scorecards in defaults.scorecards
+                for action in blueprint_scorecards["data"]
             )
         )
 
@@ -286,7 +348,7 @@ async def _initialize_defaults(
         return
     try:
         logger.info("Found default resources, starting creation process")
-        await _create_resources(port_client, defaults)
+        await _create_resources(port_client, defaults, has_provision_feature_flag)
     except AbortDefaultCreationError as e:
         logger.warning(
             f"Failed to create resources. Rolling back blueprints : {e.blueprints_to_rollback}"

port_ocean/tests/core/defaults/test_initialize.py

@@ -0,0 +1,219 @@
+from typing import Any
+from unittest.mock import AsyncMock, MagicMock
+
+import pytest
+
+from port_ocean.clients.port.client import PortClient
+from port_ocean.clients.port.types import UserAgentType
+from port_ocean.core.defaults.common import Defaults
+from port_ocean.core.defaults.initialize import _create_resources
+from port_ocean.core.models import Blueprint
+
+
+@pytest.fixture
+def mock_defaults() -> Defaults:
+    """Create a mock Defaults object with sample data."""
+    return Defaults(
+        blueprints=[
+            {
+                "identifier": "blueprint1",
+                "title": "Blueprint 1",
+                "icon": "Microservice",
+                "schema": {"type": "object", "properties": {}},
+                "relations": {},
+                "calculationProperties": {},
+                "mirrorProperties": {},
+                "aggregationProperties": {},
+                "teamInheritance": {},
+                "ownership": {},
+            },
+            {
+                "identifier": "blueprint2",
+                "title": "Blueprint 2",
+                "icon": "Microservice",
+                "schema": {"type": "object", "properties": {}},
+                "relations": {},
+                "calculationProperties": {},
+                "mirrorProperties": {},
+                "aggregationProperties": {},
+                "teamInheritance": {},
+                "ownership": {},
+            },
+        ],
+        actions=[
+            {"identifier": "action1", "title": "Action 1"},
+            {"identifier": "action2", "title": "Action 2"},
+        ],
+        scorecards=[
+            {"blueprint": "blueprint1", "data": [{"identifier": "scorecard1"}]},
+        ],
+        pages=[
+            {"identifier": "page1", "title": "Page 1"},
+        ],
+    )
+
+
+@pytest.fixture
+def mock_port_client() -> PortClient:
+    """Create a mock PortClient."""
+    mock_client = MagicMock(spec=PortClient)
+    mock_client.get_blueprint = AsyncMock()
+    mock_client.get_current_integration = AsyncMock()
+    mock_client.create_blueprint = AsyncMock()
+    mock_client.patch_blueprint = AsyncMock()
+    mock_client.create_action = AsyncMock()
+    mock_client.create_scorecard = AsyncMock()
+    mock_client.create_page = AsyncMock()
+    return mock_client
+
+
+@pytest.mark.asyncio
+async def test_create_resources_all_mapped_blueprints_exist(
+    mock_port_client: PortClient, mock_defaults: Defaults
+) -> None:
+    """
+    Test case 1: All mapped blueprints exist in Port and they are different from default ones.
+    Should skip creation of all resources.
+
+    Flow:
+    1. Get_blueprint is called for creation_stage default blueprints
+       - These should NOT exist (raise exception) so blueprints_results = []
+    2. _mapped_blueprints_exist calls get_blueprint for mapped blueprints
+       - These SHOULD exist (return Blueprint) so mapped_blueprints_exist = True
+    3. Since mapped_blueprints_exist is True, we skip creation
+    """
+    mock_port_client.get_current_integration.return_value = {  # type: ignore[attr-defined]
+        "config": {
+            "resources": [
+                {
+                    "port": {
+                        "entity": {
+                            "mappings": {
+                                "blueprint": "blueprint1test",
+                            }
+                        }
+                    }
+                },
+                {
+                    "port": {
+                        "entity": {
+                            "mappings": {
+                                "blueprint": "blueprint2test",
+                            }
+                        }
+                    }
+                },
+            ]
+        }
+    }
+
+    # Track call order: first 2 calls are for creation_stage, next 2 are for mapped blueprints
+    call_count = 0
+
+    async def get_blueprint_side_effect(
+        identifier: str, should_log: bool = True
+    ) -> Blueprint:
+        nonlocal call_count
+        call_count += 1
+
+        # First 2 calls are for creation_stage blueprints
+        # These should NOT exist to make blueprints_results = []
+        if call_count <= 2:
+            raise Exception(f"Blueprint {identifier} not found")
+
+        # Next 2 calls are for mapped blueprints
+        # These SHOULD exist to make mapped_blueprints_exist = True
+        return Blueprint.parse_obj(
+            {
+                "identifier": identifier,
+                "title": f"Blueprint {identifier}",
+                "team": None,
+                "schema": {"type": "object", "properties": {}},
+                "relations": {},
+            }
+        )
+
+    mock_port_client.get_blueprint.side_effect = get_blueprint_side_effect  # type: ignore[attr-defined]
+
+    # Execute
+    await _create_resources(
+        mock_port_client, mock_defaults, has_provision_feature_flag=False
+    )
+
+    # Assert: Should not create any blueprints, actions, scorecards, or pages
+    mock_port_client.create_blueprint.assert_not_called()  # type: ignore[attr-defined]
+    mock_port_client.patch_blueprint.assert_not_called()  # type: ignore[attr-defined]
+    mock_port_client.create_action.assert_not_called()  # type: ignore[attr-defined]
+    mock_port_client.create_scorecard.assert_not_called()  # type: ignore[attr-defined]
+    mock_port_client.create_page.assert_not_called()  # type: ignore[attr-defined]
+    assert mock_port_client.get_blueprint.call_count == 4  # type: ignore[attr-defined]
+
+
+@pytest.mark.asyncio
+async def test_create_resources_no_mapped_blueprints_exist(
+    mock_port_client: PortClient, mock_defaults: Defaults
+) -> None:
+    """
+    Test case 2: No mapped blueprints exist in Port (they are in config but don't exist).
+    Should create all blueprints, actions, scorecards, and pages from defaults.
+
+    Flow:
+    1. Get_blueprint is called for creation_stage default blueprints
+       - These should NOT exist (raise exception) so blueprints_results = []
+    2. _mapped_blueprints_exist calls get_blueprint for mapped blueprints
+       - These should NOT exist (raise exception) so mapped_blueprints_exist = False
+    3. Since both are False/empty, we create all resources
+    """
+    mock_port_client.get_current_integration.return_value = {  # type: ignore[attr-defined]
+        "config": {
+            "resources": [
+                {
+                    "port": {
+                        "entity": {
+                            "mappings": {
+                                "blueprint": "blueprint1",
+                            }
+                        }
+                    }
+                },
+                {
+                    "port": {
+                        "entity": {
+                            "mappings": {
+                                "blueprint": "blueprint2",
+                            }
+                        }
+                    }
+                },
+            ]
+        }
+    }
+
+    # Mock get_blueprint to always raise exception (blueprints don't exist)
+    # This simulates that no blueprints exist in Port
+    async def get_blueprint_side_effect(
+        identifier: str, should_log: bool = True
+    ) -> Blueprint:
+        raise Exception(f"Blueprint {identifier} not found")
+
+    mock_port_client.get_blueprint.side_effect = get_blueprint_side_effect  # type: ignore[attr-defined]
+
+    # Mock create_blueprint to return created blueprint
+    async def create_blueprint_side_effect(
+        blueprint: dict[str, Any], user_agent_type: UserAgentType | None = None
+    ) -> dict[str, Any]:
+        return {"identifier": blueprint["identifier"], **blueprint}
+
+    mock_port_client.create_blueprint.side_effect = create_blueprint_side_effect  # type: ignore[attr-defined]
+
+    # Execute
+    await _create_resources(
+        mock_port_client, mock_defaults, has_provision_feature_flag=False
+    )
+
+    # Assert
+    assert mock_port_client.create_blueprint.call_count == 2  # type: ignore[attr-defined]
+    assert mock_port_client.patch_blueprint.call_count >= 2  # type: ignore[attr-defined]
+    assert mock_port_client.create_action.call_count == 2  # type: ignore[attr-defined]
+    assert mock_port_client.create_scorecard.call_count == 1  # type: ignore[attr-defined]
+    assert mock_port_client.create_page.call_count == 1  # type: ignore[attr-defined]

{port_ocean-0.30.1.dist-info → port_ocean-0.30.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: port-ocean
-Version: 0.30.1
+Version: 0.30.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.30.1.dist-info → port_ocean-0.30.2.dist-info}/RECORD

@@ -83,7 +83,7 @@ port_ocean/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 port_ocean/core/defaults/__init__.py,sha256=8qCZg8n06WAdMu9s_FiRtDYLGPGHbOuS60vapeUoAks,142
 port_ocean/core/defaults/clean.py,sha256=_rL-NCl6Q_x3lUxDW5ACOM27IYilTCWl6ISUfRleuL0,2891
 port_ocean/core/defaults/common.py,sha256=FGB6sshNsU5tD1JLX1C__RiN7tJgQYZhUUxEQzK0sKE,4205
-port_ocean/core/defaults/initialize.py,sha256=KcWNoRu5Q8mzuVP-D6iQWfGtdp7T6yeOoZFPSrUmr2Q,11319
+port_ocean/core/defaults/initialize.py,sha256=4nZB9oxwDrZxT-O86l1QfBf6kc_MxhzYQXVNDdNBYnw,13056
 port_ocean/core/event_listener/__init__.py,sha256=ijqQ0GLcQHzISQFzC4LD3Sfv28NyaHkgohpY-AXlnZU,1374
 port_ocean/core/event_listener/actions_only.py,sha256=9zDoWESJapn0ZuZ1-WWDd6VJdeP6P3dveALnrdW-FGA,1325
 port_ocean/core/event_listener/base.py,sha256=WHrlR8GxJDjPB_YgxonJaZ4sUKWpM85SvPQVmXoWeZw,3139
@@ -175,6 +175,7 @@ port_ocean/tests/config/test_config.py,sha256=Rk4N-ldVSOfn1p23NzdVdfqUpPrqG2cMut
 port_ocean/tests/conftest.py,sha256=JXASSS0IY0nnR6bxBflhzxS25kf4iNaABmThyZ0mZt8,101
 port_ocean/tests/core/conftest.py,sha256=0Oql7R1iTbjPyNdUoO6M21IKknLwnCIgDRz2JQ7nf0w,7748
 port_ocean/tests/core/defaults/test_common.py,sha256=sR7RqB3ZYV6Xn6NIg-c8k5K6JcGsYZ2SCe_PYX5vLYM,5560
+port_ocean/tests/core/defaults/test_initialize.py,sha256=wx-DyV6ooV4HtwWH6RUaUlpLLK_6sRm7O9jrTtWMeb4,8244
 port_ocean/tests/core/event_listener/test_kafka.py,sha256=RN_JOCy4aRDUNvyQocO6WFvUMH2XeAZy-PIWHOYnD9M,2888
 port_ocean/tests/core/handlers/actions/test_execution_manager.py,sha256=5bOlcQ9qcXF_leoSFtw3FzTa7C1awUPo6TSss5e_76w,30753
 port_ocean/tests/core/handlers/entities_state_applier/test_applier.py,sha256=7XWgwUB9uVYRov4VbIz1A-7n2YLbHTTYT-4rKJxjB0A,10711
@@ -219,8 +220,8 @@ port_ocean/utils/repeat.py,sha256=U2OeCkHPWXmRTVoPV-VcJRlQhcYqPWI5NfmPlb1JIbc,32
 port_ocean/utils/signal.py,sha256=J1sI-e_32VHP_VUa5bskLMFoJjJOAk5isrnewKDikUI,2125
 port_ocean/utils/time.py,sha256=pufAOH5ZQI7gXvOvJoQXZXZJV-Dqktoj9Qp9eiRwmJ4,1939
 port_ocean/version.py,sha256=UsuJdvdQlazzKGD3Hd5-U7N69STh8Dq9ggJzQFnu9fU,177
-port_ocean-0.30.1.dist-info/LICENSE.md,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-port_ocean-0.30.1.dist-info/METADATA,sha256=yi01OJ2_QzuHHCv43m_K8j1JdM9cCwBbCMkBEG4ncfw,7054
-port_ocean-0.30.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-port_ocean-0.30.1.dist-info/entry_points.txt,sha256=F_DNUmGZU2Kme-8NsWM5LLE8piGMafYZygRYhOVtcjA,54
-port_ocean-0.30.1.dist-info/RECORD,,
+port_ocean-0.30.2.dist-info/LICENSE.md,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+port_ocean-0.30.2.dist-info/METADATA,sha256=eHDpKAYdqUR7OzLeXWFoYXh_gC2brf_cMK7hS02qHWI,7054
+port_ocean-0.30.2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+port_ocean-0.30.2.dist-info/entry_points.txt,sha256=F_DNUmGZU2Kme-8NsWM5LLE8piGMafYZygRYhOVtcjA,54
+port_ocean-0.30.2.dist-info/RECORD,,