Function test_cloud_banner

tests/integration/test_connection_banner.py:285–314

Test that the connection banner is displayed when the websocket drops. Args: connection_banner: AppHarness instance. simulate_compile_context: Which context to set for the app.

(
    connection_banner: AppHarness, simulate_compile_context: constants.CompileContext
)

def test_cloud_banner(
    connection_banner: AppHarness, simulate_compile_context: constants.CompileContext
):
    """Test that the connection banner is displayed when the websocket drops.

    Args:
        connection_banner: AppHarness instance.
        simulate_compile_context: Which context to set for the app.
    """
    assert connection_banner.app_instance is not None
    assert connection_banner.backend is not None
    driver = connection_banner.frontend()

    driver.add_cookie({"name": "backend-enabled", "value": "truly"})
    driver.refresh()
    _assert_token(connection_banner, driver)
    AppHarness.expect(lambda: not has_cloud_banner(driver))

    driver.add_cookie({"name": "backend-enabled", "value": "false"})
    driver.refresh()
    if simulate_compile_context == constants.CompileContext.DEPLOY:
        AppHarness.expect(lambda: has_cloud_banner(driver))
    else:
        _assert_token(connection_banner, driver)
        AppHarness.expect(lambda: not has_cloud_banner(driver))

    driver.delete_cookie("backend-enabled")
    driver.refresh()
    _assert_token(connection_banner, driver)
    AppHarness.expect(lambda: not has_cloud_banner(driver))