test: add aggregated dataset to testing ldm

risk: nonprod

Author: fiedlr

gooddata-sdk/gooddata_sdk/catalog/workspace/declarative_model/workspace/logical_model/dataset/dataset.py

@@ -42,6 +42,7 @@ class CatalogDeclarativeDataset(Base):
     attributes: Optional[list[CatalogDeclarativeAttribute]] = None
     facts: Optional[list[CatalogDeclarativeFact]] = None
     aggregated_facts: Optional[list[CatalogDeclarativeAggregatedFact]] = None
+    precedence: Optional[int] = None
     data_source_table_id: Optional[CatalogDataSourceTableIdentifier] = None
     sql: Optional[CatalogDeclarativeDatasetSql] = None
     tags: Optional[list[str]] = None

gooddata-sdk/gooddata_sdk/catalog/workspace/entity_model/content_objects/dataset.py

@@ -145,6 +145,14 @@ def generate_facts_from_api(self) -> list[CatalogFact]:
         default=attr.Factory(lambda self: self.generate_facts_from_api(), takes_self=True),
     )
 
+    # TODO: Doublecheck if we shouldn't do something like for facts
+    aggregated_facts: Optional[list[CatalogAggregatedFact]] = attr.field(
+        default=attr.Factory(lambda self: self.json_api_attributes.get("aggregatedFacts"), takes_self=True),
+    )
+    precedence: Optional[int] = attr.field(
+        default=attr.Factory(lambda self: self.json_api_attributes.get("precedence"), takes_self=True)
+    )
+
     grain: Optional[list] = attr.field(
         default=attr.Factory(lambda self: self.json_api_attributes.get("grain"), takes_self=True)
     )

gooddata-sdk/gooddata_sdk/catalog/workspace/model_container.py

@@ -6,6 +6,7 @@
 
 from gooddata_sdk.catalog.types import ValidObjects
 from gooddata_sdk.catalog.workspace.entity_model.content_objects.dataset import (
+    CatalogAggregatedFact,
     CatalogAttribute,
     CatalogDataset,
     CatalogFact,
@@ -46,6 +47,10 @@ def datasets(self) -> list[CatalogDataset]:
     def facts(self) -> list[CatalogFact]:
         return [f for d in self._datasets for f in d.facts]
 
+    @property
+    def aggregated_facts(self) -> list[CatalogAggregatedFact]:
+        return [f for d in self._datasets for f in d.aggregated_facts]
+
     @property
     def attributes(self) -> list[CatalogAttribute]:
         return [a for d in self._datasets for a in d.attributes]

gooddata-sdk/tests/catalog/expected/declarative_workspaces.json

@@ -2376,6 +2376,70 @@
               ],
               "title": "Campaign channels"
             },
+            {
+              "attributes": [
+                {
+                  "description": "Category Agg",
+                  "id": "campaign_channels.category_agg",
+                  "sourceColumn": "category",
+                  "sourceColumnDataType": "STRING",
+                  "labels": [],
+                  "tags": [
+                    "Campaign channels"
+                  ],
+                  "title": "Category"
+                }
+              ],
+              "description": "Campaign channels per categories",
+              "facts": [],
+              "aggregatedFacts": [
+                {
+                  "description": "Budget Agg",
+                  "id": "budget_agg",
+                  "sourceColumn": "budget",
+                  "sourceColumnDataType": "INT",
+                  "tags": [
+                    "Campaign channels per category"
+                  ],
+                  "sourceFactReference": {
+                    "reference": {
+                      "id": "budget",
+                      "type": "fact"
+                    },
+                    "operation": "SUM"
+                  }
+                }
+              ],
+              "grain": [
+                {
+                  "id": "campaign_channels.category_agg",
+                  "type": "attribute"
+                }
+              ],
+              "id": "campaign_channels_per_category",
+              "precedence":  1,
+              "references": [
+                {
+                  "identifier": {
+                    "id": "campaign_channels",
+                    "type": "dataset"
+                  },
+                  "multivalue": false,
+                  "sourceColumns": [
+                    "campaign_channel_category"
+                  ],
+                  "sourceColumnDataTypes": [ "STRING" ]
+                }
+              ],
+              "tags": [
+                "Campaign channels per category"
+              ],
+              "sql": {
+                "statement": "SELECT category, SUM(budget) FROM campaign_channels GROUP BY category",
+                "dataSourceId": "demo-test-ds"
+              },
+              "title": "Campaign channels per category"
+            },
             {
               "attributes": [
                 {
@@ -2412,6 +2476,7 @@
               },
               "description": "Campaigns",
               "facts": [],
+              "aggregatedFacts": [],
               "grain": [
                 {
                   "id": "campaign_id",
@@ -2494,6 +2559,7 @@
               },
               "description": "Customers",
               "facts": [],
+              "aggregatedFacts": [],
               "grain": [
                 {
                   "id": "customer_id",
@@ -2575,6 +2641,7 @@
                   "title": "Quantity"
                 }
               ],
+              "aggregatedFacts": [],
               "grain": [
                 {
                   "id": "order_line_id",
@@ -2724,6 +2791,7 @@
               },
               "description": "Products",
               "facts": [],
+              "aggregatedFacts": [],
               "grain": [
                 {
                   "id": "product_id",