from test_broadband_portal_backend_config import params

@pytest.fixture
def input_data_params():
    return params

def create_prefect_deployment_run(input_data):
    submission_key = input_data["input"]["submission_key"]
    s3_filepath = input_data["input"]["s3_filepath"]
    submission_flow_params = {
        "submission_key": submission_key,
        "s3_filepath": s3_filepath
    }

    # Remove the "S3://" prefix
    s3_filepath = input_data["input"]["s3_filepath"].replace("S3://", "")

    # Extract the object name using os.path.basename
    object_name = os.path.basename(s3_filepath)

    response = run_deployment(
        name='isp-portal-submission/isp_portal_submission_yaml_deploy_k8',
        parameters=submission_flow_params,
        flow_run_name=object_name,
        timeout=300
    )
    return response


class TestPrefectPipeline:
    ss_candy = Secret.load("spatial-stream-candy").get()

    @pytest.mark.asyncio
    async def test_submission_flow_completes(self, input_data_params):
        flows = []
        for input_data in input_data_params:
            flow = create_prefect_deployment_run(input_data)
            flows.append(flow)

        responses = await asyncio.gather(*flows)
        for response in responses:
            assert response.state_name == "Completed"

run_deployment()

run_deploymen()

FlowRun

run_deployment(..., timeout=0)