updated DLP quickstart terminal print out and tests (GoogleCloudPlatform#1413)

averikitsch · andrewsg · commit 7fdebf70da7c · 2018-03-19T11:22:44.000-07:00
diff --git 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]
