def test_create_job_from_preset(capsys, test_bucket):
    create_job_from_preset.create_job_from_preset(project_id, location,
                                                  input_uri,
                                                  output_uri_for_preset,
                                                  preset)
    out, _ = capsys.readouterr()
    job_name_prefix = f"projects/{project_number}/locations/{location}/jobs/"
    assert job_name_prefix in out

    str_slice = out.split("/")
    job_id = str_slice[len(str_slice) - 1].rstrip("\n")
    job_name = f"projects/{project_number}/locations/{location}/jobs/{job_id}"
    assert job_name in out

    get_job.get_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert job_name in out

    time.sleep(30)

    _assert_job_state_succeeded_or_running(capsys, job_id)

    list_jobs.list_jobs(project_id, location)
    out, _ = capsys.readouterr()
    assert job_name in out

    delete_job.delete_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert "Deleted job" in out
def test_create_job_with_standalone_captions(capsys, test_bucket):
    create_job_with_standalone_captions.create_job_with_standalone_captions(
        project_id,
        location,
        input_uri,
        captions_uri,
        output_uri_for_standalone_captions,
    )
    out, _ = capsys.readouterr()
    job_name_prefix = f"projects/{project_number}/locations/{location}/jobs/"
    assert job_name_prefix in out

    str_slice = out.split("/")
    job_id = str_slice[len(str_slice) - 1].rstrip("\n")
    job_name = f"projects/{project_number}/locations/{location}/jobs/{job_id}"
    assert job_name in out

    get_job.get_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert job_name in out

    time.sleep(
        30
    )  # Transcoding jobs need time to complete. Once the job completes, check the job state.

    _assert_job_state_succeeded(capsys, job_id)

    list_jobs.list_jobs(project_id, location)
    out, _ = capsys.readouterr()
    assert job_name in out

    delete_job.delete_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert "Deleted job" in out
Beispiel #3
0
def _create_job_from_ad_hoc(capsys):
    create_job_from_ad_hoc.create_job_from_ad_hoc(project_id, location,
                                                  input_uri,
                                                  output_uri_for_adhoc)
    out, _ = capsys.readouterr()
    job_name_prefix = f"projects/{project_number}/locations/{location}/jobs/"
    assert job_name_prefix in out

    str_slice = out.split("/")
    job_id = str_slice[len(str_slice) - 1].rstrip("\n")
    job_name = f"projects/{project_number}/locations/{location}/jobs/{job_id}"
    assert job_name in out

    get_job.get_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert job_name in out

    time.sleep(30)

    _get_job_state(capsys, job_id)

    list_jobs.list_jobs(project_id, location)
    out, _ = capsys.readouterr()
    assert job_name in out

    delete_job.delete_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert "Deleted job" in out
def test_create_job_with_animated_overlay(capsys, test_bucket):
    create_job_with_animated_overlay.create_job_with_animated_overlay(
        project_id,
        location,
        input_uri,
        overlay_image_uri,
        output_uri_for_animated_overlay,
    )
    out, _ = capsys.readouterr()
    job_name_prefix = f"projects/{project_number}/locations/{location}/jobs/"
    assert job_name_prefix in out

    str_slice = out.split("/")
    job_id = str_slice[len(str_slice) - 1].rstrip("\n")
    job_name = f"projects/{project_number}/locations/{location}/jobs/{job_id}"
    assert job_name in out

    get_job.get_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert job_name in out

    time.sleep(30)

    _get_job_state(capsys, job_id)

    list_jobs.list_jobs(project_id, location)
    out, _ = capsys.readouterr()
    assert job_name in out

    delete_job.delete_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert "Deleted job" in out
def test_create_job_with_periodic_spritesheet(capsys, test_bucket):
    create_job_with_periodic_images_spritesheet.create_job_with_periodic_images_spritesheet(
        project_id,
        location,
        input_uri,
        output_uri_for_periodic_spritesheet,
    )
    out, _ = capsys.readouterr()
    job_name_prefix = f"projects/{project_number}/locations/{location}/jobs/"
    assert job_name_prefix in out

    str_slice = out.split("/")
    job_id = str_slice[len(str_slice) - 1].rstrip("\n")
    job_name = f"projects/{project_number}/locations/{location}/jobs/{job_id}"
    assert job_name in out

    get_job.get_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert (
        job_name in out
    )  # Get the job name so you can use it later to get the job and delete the job.

    time.sleep(
        30
    )  # Transcoding jobs need time to complete. Once the job completes, check the job state.

    _assert_job_state_succeeded(capsys, job_id)
    _assert_file_in_bucket(
        capsys,
        test_bucket,
        output_dir_for_periodic_spritesheet + small_spritesheet_file_prefix +
        spritesheet_file_suffix,
    )
    _assert_file_in_bucket(
        capsys,
        test_bucket,
        output_dir_for_periodic_spritesheet + large_spritesheet_file_prefix +
        spritesheet_file_suffix,
    )

    list_jobs.list_jobs(project_id, location)
    out, _ = capsys.readouterr()
    assert job_name in out

    delete_job.delete_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert "Deleted job" in out
Beispiel #6
0
def test_create_job_from_template(capsys, test_bucket):

    job_template_name = (
        f"projects/{project_number}/locations/{location}/jobTemplates/{template_id}"
    )

    create_job_template.create_job_template(project_id, location, template_id)
    out, _ = capsys.readouterr()
    assert job_template_name in out

    create_job_from_template.create_job_from_template(project_id, location,
                                                      input_uri,
                                                      output_uri_for_template,
                                                      template_id)
    out, _ = capsys.readouterr()
    job_name_prefix = f"projects/{project_number}/locations/{location}/jobs/"
    assert job_name_prefix in out

    str_slice = out.split("/")
    job_id = str_slice[len(str_slice) - 1].rstrip("\n")
    job_name = f"projects/{project_number}/locations/{location}/jobs/{job_id}"
    assert job_name in out

    get_job.get_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert job_name in out

    time.sleep(30)

    _get_job_state(capsys, job_id)

    list_jobs.list_jobs(project_id, location)
    out, _ = capsys.readouterr()
    assert job_name in out

    delete_job.delete_job(project_id, location, job_id)
    out, _ = capsys.readouterr()
    assert "Deleted job" in out

    delete_job_template.delete_job_template(project_id, location, template_id)
    out, _ = capsys.readouterr()
    assert "Deleted job template" in out