From 7d20f8f3edc82d270928f5e85550685957758aee Mon Sep 17 00:00:00 2001
From: Nick Proskurin <42863572+MCPN@users.noreply.github.com>
Date: Tue, 14 Nov 2023 12:01:04 +0100
Subject: [PATCH] Add a markup test (#89)

---
 .../db/data_api/result/test_markup.py         | 86 +++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 lib/dl_api_lib/dl_api_lib_tests/db/data_api/result/test_markup.py

diff --git a/lib/dl_api_lib/dl_api_lib_tests/db/data_api/result/test_markup.py b/lib/dl_api_lib/dl_api_lib_tests/db/data_api/result/test_markup.py
new file mode 100644
index 000000000..05ddb073f
--- /dev/null
+++ b/lib/dl_api_lib/dl_api_lib_tests/db/data_api/result/test_markup.py
@@ -0,0 +1,86 @@
+from __future__ import annotations
+
+import uuid
+
+import pytest
+
+from dl_api_lib_testing.data_api_base import DataApiTestParams
+from dl_api_client.dsmaker.shortcuts.result_data import get_data_rows
+from dl_api_lib_tests.db.base import DefaultApiTestBase
+
+
+class TestUMarkup(DefaultApiTestBase):
+    @pytest.fixture(scope="function")
+    def data_api_test_params(self, sample_table) -> DataApiTestParams:
+        # This default is defined for the sample table
+        return DataApiTestParams(
+            two_dims=("category", "city"),
+            summable_field="sales",
+            range_field="sales",
+            distinct_field="city",
+            date_field="order_date",
+        )
+
+    def test_markup(self, saved_dataset, data_api, data_api_test_params):
+        ds = saved_dataset
+
+        field_a, field_b, field_nulled = (str(uuid.uuid4()) for _ in range(3))
+        formula_a = """
+            markup(
+                italic(url(
+                    "http://example.com/?city=" + [city] + "&_=1",
+                    [city])),
+                " (", bold(str([order_date])), ")")
+        """
+        formula_b = """
+            url("https://example.com/?text=" + str([sales]) + " usd в рублях", str([sales]))
+        """
+        formula_nulled = """
+            url(if([sales] > 0, NULL, "neg"), str([sales]))
+        """
+
+        result_resp = data_api.get_result(
+            dataset=ds,
+            updates=[
+                ds.field(
+                    id=field_a,
+                    title="Field A",
+                    formula=formula_a,
+                ).add(),
+                ds.field(
+                    id=field_b,
+                    title="Field B",
+                    formula=formula_b,
+                ).add(),
+                ds.field(
+                    id=field_nulled,
+                    title="Field Nulled",
+                    formula=formula_nulled,
+                ).add(),
+            ],
+            fields=[
+                ds.field(id=field_a),
+                ds.field(id=field_b),
+                ds.field(id=field_nulled),
+            ],
+        )
+
+        assert result_resp.status_code == 200, result_resp.response_errors
+        data_rows = get_data_rows(result_resp)
+        assert data_rows
+
+        some_row = data_rows[0]
+        assert len(some_row) == 3
+        res_a, res_b, res_nulled = some_row
+        assert isinstance(res_a, dict)
+        assert isinstance(res_b, dict)
+        assert res_nulled is None
+
+        b_val = res_b["content"]["content"]
+        assert isinstance(b_val, str)
+        expected_b = {
+            "type": "url",
+            "url": f"https://example.com/?text={b_val} usd в рублях",
+            "content": {"type": "text", "content": b_val},
+        }
+        assert res_b == expected_b