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_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
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
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
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
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)
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
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)
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
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