taipy/tests/gui/server/ws/test_du.py

# Copyright 2021-2024 Avaiga Private Limited
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
#        http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
# an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
# specific language governing permissions and limitations under the License.

import inspect

from taipy.gui import Gui, Markdown


def test_du_table_data_fetched(gui: Gui, helpers, csvdata):
    # Bind test variables
    csvdata = csvdata

    # set gui frame
    gui._set_frame(inspect.currentframe())
    Gui._set_timezone("UTC")

    # Bind a page so that the variable will be evaluated as expression
    gui.add_page(
        "test",
        Markdown(
            "<|{csvdata}|table|page_size=10|page_size_options=10;30;100|columns=Day;Entity;Code;Daily hospital occupancy|date_format=eee dd MMM yyyy|>"  # noqa: E501
        ),
    )
    gui.run(run_server=False)
    flask_client = gui._server.test_client()
    # WS client and emit
    ws_client = gui._server._ws.test_client(gui._server.get_flask())
    sid = helpers.create_scope_and_get_sid(gui)
    # Get the jsx once so that the page will be evaluated -> variable will be registered
    flask_client.get(f"/taipy-jsx/test?client_id={sid}")
    ws_client.emit(
        "message",
        {
            "client_id": sid,
            "type": "DU",
            "name": "_TpD_tpec_TpExPr_csvdata_TPMDL_0",
            "payload": {
                "columns": ["Day", "Entity", "Code", "Daily hospital occupancy"],
                "pagekey": "0-100-asc",
                "start": 0,
                "end": 9,
                "orderby": "",
                "sort": "asc",
            },
        },
    )
    # assert for received message (message that would be sent to the front-end client)
    received_messages = ws_client.get_received()
    assert received_messages
    helpers.assert_outward_ws_message(
        received_messages[0],
        "MU",
        "_TpD_tpec_TpExPr_csvdata_TPMDL_0",
        {
            "data": [
                {
                    "Code": "AUT",
                    "Day_str": "2020-04-01T00:00:00.000000Z",
                    "Daily hospital occupancy": 856,
                    "Entity": "Austria",
                    "_tp_index": 0,
                },
                {
                    "Code": "AUT",
                    "Day_str": "2020-04-02T00:00:00.000000Z",
                    "Daily hospital occupancy": 823,
                    "Entity": "Austria",
                    "_tp_index": 1,
                },
                {
                    "Code": "AUT",
                    "Day_str": "2020-04-03T00:00:00.000000Z",
                    "Daily hospital occupancy": 829,
                    "Entity": "Austria",
                    "_tp_index": 2,
                },
                {
                    "Code": "AUT",
                    "Day_str": "2020-04-04T00:00:00.000000Z",
                    "Daily hospital occupancy": 826,
                    "Entity": "Austria",
                    "_tp_index": 3,
                },
                {
                    "Code": "AUT",
                    "Day_str": "2020-04-05T00:00:00.000000Z",
                    "Daily hospital occupancy": 712,
                    "Entity": "Austria",
                    "_tp_index": 4,
                },
                {
                    "Code": "AUT",
                    "Day_str": "2020-04-06T00:00:00.000000Z",
                    "Daily hospital occupancy": 824,
                    "Entity": "Austria",
                    "_tp_index": 5,
                },
                {
                    "Code": "AUT",
                    "Day_str": "2020-04-07T00:00:00.000000Z",
                    "Daily hospital occupancy": 857,
                    "Entity": "Austria",
                    "_tp_index": 6,
                },
                {
                    "Code": "AUT",
                    "Day_str": "2020-04-08T00:00:00.000000Z",
                    "Daily hospital occupancy": 829,
                    "Entity": "Austria",
                    "_tp_index": 7,
                },
                {
                    "Code": "AUT",
                    "Day_str": "2020-04-09T00:00:00.000000Z",
                    "Daily hospital occupancy": 820,
                    "Entity": "Austria",
                    "_tp_index": 8,
                },
                {
                    "Code": "AUT",
                    "Day_str": "2020-04-10T00:00:00.000000Z",
                    "Daily hospital occupancy": 771,
                    "Entity": "Austria",
                    "_tp_index": 9,
                },
            ],
            "rowcount": 14477,
            "start": 0,
            "format": "JSON",
        },
    )