# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
import logging

from tests.utils.tracked_container import TrackedContainer

LOGGER = logging.getLogger(__name__)


def test_spark_shell(container: TrackedContainer) -> None:
    """Checking if Spark (spark-shell) is running properly"""
    logs = container.run_and_wait(
        timeout=60,
        no_warnings=False,
        command=["bash", "-c", 'spark-shell <<< "1+1"'],
    )
    warnings = TrackedContainer.get_warnings(logs)
    assert warnings == ["WARNING: Using incubator modules: jdk.incubator.vector"]
    assert "res0: Int = 2" in logs, "spark-shell does not work"