streamlit

1
# Copyright (c) Streamlit Inc. (2018-2022) Snowflake Inc. (2022-2024)
2
#
3
# Licensed under the Apache License, Version 2.0 (the "License");
4
# you may not use this file except in compliance with the License.
5
# You may obtain a copy of the License at
6
#
7
#     http://www.apache.org/licenses/LICENSE-2.0
8
#
9
# Unless required by applicable law or agreed to in writing, software
10
# distributed under the License is distributed on an "AS IS" BASIS,
11
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
# See the License for the specific language governing permissions and
13
# limitations under the License.
14

15
from playwright.sync_api import Page, expect
16

17
from e2e_playwright.conftest import ImageCompareFunction, rerun_app
18

19

20
def test_renders_chat_messages_correctly_1(
21
    themed_app: Page, assert_snapshot: ImageCompareFunction
22
):
23
    """Test if the chat messages render correctly"""
24
    # Wait a bit more to allow all images to load:
25
    chat_message_elements = themed_app.get_by_test_id("stChatMessage")
26
    expect(chat_message_elements).to_have_count(12)
27

28
    # rerun to populate session state chat message
29
    rerun_app(themed_app)
30

31
    expect(chat_message_elements).to_have_count(14)
32
    for i, element in enumerate(chat_message_elements.all()):
33
        element.scroll_into_view_if_needed()
34
        expect(element).to_be_in_viewport()
35
        # Wait a bit more to allow the avatar images to load:
36
        themed_app.wait_for_timeout(100)
37
        assert_snapshot(element, name=f"st_chat_message-{i}")
38