datajunction 0.0.207__tar.gz → 0.0.209__tar.gz

{datajunction-0.0.207 → datajunction-0.0.209}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datajunction
-Version: 0.0.207
+Version: 0.0.209
 Summary: DataJunction client library for connecting to a DataJunction server
 Project-URL: repository, https://github.com/DataJunction/dj
 Author-email: DataJunction Authors <yian.shang@gmail.com>

{datajunction-0.0.207 → datajunction-0.0.209}/datajunction/__about__.py

@@ -2,4 +2,4 @@
 Version for Hatch
 """
 
-__version__ = "0.0.207"
+__version__ = "0.0.209"

{datajunction-0.0.207 → datajunction-0.0.209}/datajunction/cli.py

@@ -143,7 +143,16 @@ class DJCLI:
                     "display_name": getattr(node, "display_name", None),
                     "query": getattr(node, "query", None),
                     "columns": [
-                        {"name": col.name, "type": col.type} for col in node.columns
+                        {
+                            "name": col.name
+                            + (
+                                (col.dimension_column or "")
+                                if node.type == "cube"
+                                else ""
+                            ),
+                            "type": col.type,
+                        }
+                        for col in node.columns
                     ]
                     if hasattr(node, "columns") and node.columns
                     else [],

{datajunction-0.0.207 → datajunction-0.0.209}/tests/test_cli.py

@@ -5,6 +5,7 @@ import os
 import re
 import sys
 from io import StringIO
+from types import SimpleNamespace
 from typing import Callable
 from unittest import mock
 from unittest.mock import patch
@@ -294,6 +295,43 @@ def test_json_output_commands(
     assert isinstance(data, (dict, list))
 
 
+def test_describe_cube_json_preserves_column_roles():
+    """Role-played cube columns must remain distinct in CLI JSON output."""
+    builder_client = mock.MagicMock()
+    builder_client.node.return_value = SimpleNamespace(
+        name="v3.role_cube",
+        type="cube",
+        description="Role-playing cube",
+        display_name="Role Cube",
+        query=None,
+        columns=[
+            SimpleNamespace(
+                name="v3.date.date_id",
+                dimension_column="[order]",
+                type="int",
+            ),
+            SimpleNamespace(
+                name="v3.date.date_id",
+                dimension_column="[ship]",
+                type="int",
+            ),
+        ],
+        status="valid",
+        mode="published",
+        version="v1.0",
+    )
+
+    output = run_cli_command(
+        builder_client,
+        ["dj", "describe", "v3.role_cube", "--format", "json"],
+    )
+
+    assert [column["name"] for column in json.loads(output)["columns"]] == [
+        "v3.date.date_id[order]",
+        "v3.date.date_id[ship]",
+    ]
+
+
 @pytest.mark.parametrize(
     "node_name,expected_in_output",
     [