def test_inspect_gcs_image_file(bucket, topic_id, subscription_id, capsys):
    inspect_content.inspect_gcs_file(
        GCLOUD_PROJECT, bucket.name, 'test.png', topic_id, subscription_id,
        ['FIRST_NAME', 'EMAIL_ADDRESS', 'PHONE_NUMBER'])

    out, _ = capsys.readouterr()
    assert 'Info type: EMAIL_ADDRESS' in out
Exemplo n.º 2
0
def test_inspect_gcs_file_with_custom_info_types(
    bucket, topic_id, subscription_id, capsys
):
    dictionaries = ["*****@*****.**"]
    regexes = ["\\(\\d{3}\\) \\d{3}-\\d{4}"]

    inspect_content.inspect_gcs_file(
        GCLOUD_PROJECT,
        bucket.name,
        "test.txt",
        topic_id,
        subscription_id,
        [],
        custom_dictionaries=dictionaries,
        custom_regexes=regexes,
        timeout=420,
    )

    out, _ = capsys.readouterr()

    assert "Inspection operation started" in out
    # Cancel the operation
    operation_id = out.split("Inspection operation started: ")[1].split("\n")[0]
    print(operation_id)
    client = google.cloud.dlp_v2.DlpServiceClient()
    client.cancel_dlp_job(operation_id)
def test_inspect_gcs_file_with_info_types(bucket, capsys):
    inspect_content.inspect_gcs_file(bucket.name,
                                     'test.txt',
                                     info_types=['EMAIL_ADDRESS'])

    out, _ = capsys.readouterr()
    assert 'Info type: EMAIL_ADDRESS' in out
def test_inspect_gcs_file_no_results(bucket, topic_id, subscription_id,
                                     capsys):
    inspect_content.inspect_gcs_file(
        GCLOUD_PROJECT, bucket.name, 'harmless.txt', topic_id, subscription_id,
        ['FIRST_NAME', 'EMAIL_ADDRESS', 'PHONE_NUMBER'])

    out, _ = capsys.readouterr()
    assert 'No findings' in out
def test_inspect_gcs_image_file(bucket, topic_id, subscription_id, capsys):
    inspect_content.inspect_gcs_file(
        GCLOUD_PROJECT,
        bucket.name,
        'test.png',
        topic_id,
        subscription_id,
        ['FIRST_NAME', 'EMAIL_ADDRESS', 'PHONE_NUMBER'])

    out, _ = capsys.readouterr()
    assert 'Info type: EMAIL_ADDRESS' in out
Exemplo n.º 6
0
def test_inspect_gcs_image_file(bucket, topic_id, subscription_id, capsys):
    inspect_content.inspect_gcs_file(
        GCLOUD_PROJECT,
        bucket.name,
        "test.png",
        topic_id,
        subscription_id,
        ["FIRST_NAME", "EMAIL_ADDRESS", "PHONE_NUMBER"],
    )

    out, _ = capsys.readouterr()
    assert "Info type: EMAIL_ADDRESS" in out
def test_inspect_gcs_file_no_results(
        bucket, topic_id, subscription_id, capsys):
    inspect_content.inspect_gcs_file(
        GCLOUD_PROJECT,
        bucket.name,
        'harmless.txt',
        topic_id,
        subscription_id,
        ['FIRST_NAME', 'EMAIL_ADDRESS', 'PHONE_NUMBER'])

    out, _ = capsys.readouterr()
    assert 'No findings' in out
Exemplo n.º 8
0
def test_inspect_gcs_file_no_results(bucket, topic_id, subscription_id, capsys):
    inspect_content.inspect_gcs_file(
        GCLOUD_PROJECT,
        bucket.name,
        "harmless.txt",
        topic_id,
        subscription_id,
        ["FIRST_NAME", "EMAIL_ADDRESS", "PHONE_NUMBER"],
        timeout=420,
    )

    out, _ = capsys.readouterr()
    assert "No findings" in out
Exemplo n.º 9
0
def test_inspect_gcs_multiple_files(bucket, topic_id, subscription_id, capsys):
    inspect_content.inspect_gcs_file(
        GCLOUD_PROJECT,
        bucket.name,
        "*",
        topic_id,
        subscription_id,
        ["EMAIL_ADDRESS", "PHONE_NUMBER"],
    )

    out, _ = capsys.readouterr()
    assert "Info type: EMAIL_ADDRESS" in out
    assert "Info type: PHONE_NUMBER" in out
Exemplo n.º 10
0
def test_inspect_gcs_image_file(bucket, topic_id, subscription_id, capsys):
    try:
        inspect_content.inspect_gcs_file(GCLOUD_PROJECT,
                                         bucket.name,
                                         "test.png",
                                         topic_id,
                                         subscription_id,
                                         ["EMAIL_ADDRESS", "PHONE_NUMBER"],
                                         timeout=TIMEOUT)

        out, _ = capsys.readouterr()
        assert "Info type: EMAIL_ADDRESS" in out
    finally:
        cancel_operation(out)
Exemplo n.º 11
0
def test_inspect_gcs_multiple_files(bucket, topic_id, subscription_id, capsys):
    try:
        inspect_content.inspect_gcs_file(
            GCLOUD_PROJECT,
            bucket.name,
            "*",
            topic_id,
            subscription_id,
            ["EMAIL_ADDRESS", "PHONE_NUMBER"],
            timeout=1)

        out, _ = capsys.readouterr()

        assert "Inspection operation started" in out
    finally:
        cancel_operation(out)
def test_inspect_gcs_file_with_custom_info_types(bucket, topic_id,
                                                 subscription_id, capsys):
    dictionaries = ['*****@*****.**']
    regexes = ['\\(\\d{3}\\) \\d{3}-\\d{4}']

    inspect_content.inspect_gcs_file(GCLOUD_PROJECT,
                                     bucket.name,
                                     'test.txt',
                                     topic_id,
                                     subscription_id, [],
                                     custom_dictionaries=dictionaries,
                                     custom_regexes=regexes)

    out, _ = capsys.readouterr()
    assert 'Info type: CUSTOM_DICTIONARY_0' in out
    assert 'Info type: CUSTOM_REGEX_0' in out
Exemplo n.º 13
0
def test_inspect_gcs_file_no_results(bucket, topic_id, subscription_id,
                                     capsys):
    try:
        inspect_content.inspect_gcs_file(GCLOUD_PROJECT,
                                         bucket.name,
                                         "harmless.txt",
                                         topic_id,
                                         subscription_id,
                                         ["EMAIL_ADDRESS", "PHONE_NUMBER"],
                                         timeout=TIMEOUT)

        out, _ = capsys.readouterr()

        assert "No findings" in out
    finally:
        cancel_operation(out)
Exemplo n.º 14
0
def test_inspect_gcs_multiple_files(bucket, topic_id, subscription_id, capsys):
    inspect_content.inspect_gcs_file(
        GCLOUD_PROJECT,
        bucket.name,
        "*",
        topic_id,
        subscription_id,
        ["EMAIL_ADDRESS", "PHONE_NUMBER"],
    )

    out, _ = capsys.readouterr()

    assert "Inspection operation started" in out
    # Cancel the operation
    operation_id = out.split("Inspection operation started: ")[1].split("\n")[0]
    print(operation_id)
    client = google.cloud.dlp_v2.DlpServiceClient()
    client.cancel_dlp_job(operation_id)
def test_inspect_gcs_file_with_custom_info_types(bucket, topic_id,
                                                 subscription_id, capsys):
    dictionaries = ['*****@*****.**']
    regexes = ['\\(\\d{3}\\) \\d{3}-\\d{4}']

    inspect_content.inspect_gcs_file(
        GCLOUD_PROJECT,
        bucket.name,
        'test.txt',
        topic_id,
        subscription_id,
        [],
        custom_dictionaries=dictionaries,
        custom_regexes=regexes)

    out, _ = capsys.readouterr()
    assert 'Info type: CUSTOM_DICTIONARY_0' in out
    assert 'Info type: CUSTOM_REGEX_0' in out
Exemplo n.º 16
0
def test_inspect_gcs_file_with_custom_info_types(bucket, topic_id,
                                                 subscription_id, capsys):
    try:
        dictionaries = ["*****@*****.**"]
        regexes = ["\\(\\d{3}\\) \\d{3}-\\d{4}"]

        inspect_content.inspect_gcs_file(GCLOUD_PROJECT,
                                         bucket.name,
                                         "test.txt",
                                         topic_id,
                                         subscription_id, [],
                                         custom_dictionaries=dictionaries,
                                         custom_regexes=regexes,
                                         timeout=TIMEOUT)

        out, _ = capsys.readouterr()

        assert "Info type: EMAIL_ADDRESS" in out
    finally:
        cancel_operation(out)
def test_inspect_gcs_image_file(bucket, capsys):
    inspect_content.inspect_gcs_file(bucket.name, 'test.png')

    out, _ = capsys.readouterr()
    assert 'Info type: EMAIL_ADDRESS' in out
def test_inspect_gcs_multiple_files(bucket, capsys):
    inspect_content.inspect_gcs_file(bucket.name, '*')

    out, _ = capsys.readouterr()
    assert 'Info type: PHONE_NUMBER' in out
    assert 'Info type: CREDIT_CARD' in out
def test_inspect_gcs_file_no_results(bucket, capsys):
    inspect_content.inspect_gcs_file(bucket.name, 'harmless.txt')

    out, _ = capsys.readouterr()
    assert 'No findings' in out