# coding=utf-8
import pytest

from spark_executor.tools.requests import PrepareSubmitJobRequest


def test_prepare_submit_job_request_requires_all_confirmed_parameters():
    with pytest.raises(ValueError):
        PrepareSubmitJobRequest(connection="prod", script_path="/tmp/x.py")


def test_prepare_submit_job_request_accepts_extra_args():
    req = PrepareSubmitJobRequest(
        connection="prod",
        script_path="/tmp/x.py",
        queue="default",
        executor_memory="4G",
        executor_cores=2,
        num_executors=2,
        app_name="test-app",
        extra_args={"jars": "hdfs:///lib/foo.jar"},
    )
    assert req.extra_args == {"jars": "hdfs:///lib/foo.jar"}