diff --git a/dlp/quickstart.py b/dlp/quickstart.py index 17d2f8b8f96..e826f7f771f 100644 --- a/dlp/quickstart.py +++ b/dlp/quickstart.py @@ -72,7 +72,12 @@ def quickstart(): except AttributeError: pass print('Info type: {}'.format(finding.info_type.name)) - print('Likelihood: {}'.format(finding.likelihood)) + # Convert likelihood value to string respresentation. + likelihood = (google.cloud.dlp.types.Finding.DESCRIPTOR + .fields_by_name['likelihood'] + .enum_type.values_by_number[finding.likelihood] + .name) + print('Likelihood: {}'.format(likelihood)) else: print('No findings.') # [END quickstart] diff --git a/dlp/quickstart_test.py b/dlp/quickstart_test.py index 5b8faf88099..aa529021b19 100644 --- a/dlp/quickstart_test.py +++ b/dlp/quickstart_test.py @@ -19,4 +19,5 @@ def test_quickstart(capsys): quickstart.quickstart() out, _ = capsys.readouterr() - assert 'US_MALE_NAME' in out + assert 'FIRST_NAME' in out + assert 'LAST_NAME' in out