Skip to content

Commit f4b54b2

Browse files
committed
Updated QuickStart tests and samples.
1 parent fb7261f commit f4b54b2

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

dlp/src/main/java/com/example/dlp/QuickStart.java

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
package com.example.dlp;
1818

19+
import com.google.cloud.ServiceOptions;
1920
import com.google.cloud.dlp.v2.DlpServiceClient;
2021
import com.google.privacy.dlp.v2.ByteContentItem;
2122
import com.google.privacy.dlp.v2.ContentItem;
@@ -26,6 +27,7 @@
2627
import com.google.privacy.dlp.v2.InspectContentResponse;
2728
import com.google.privacy.dlp.v2.InspectResult;
2829
import com.google.privacy.dlp.v2.Likelihood;
30+
import com.google.privacy.dlp.v2.ProjectName;
2931
import com.google.protobuf.ByteString;
3032
import java.util.Arrays;
3133
import java.util.List;
@@ -59,7 +61,9 @@ public static void main(String[] args) throws Exception {
5961
try (DlpServiceClient dlpServiceClient = DlpServiceClient.create()) {
6062

6163
InspectConfig.FindingLimits findingLimits =
62-
InspectConfig.FindingLimits.newBuilder().setMaxFindingsPerItem(maxFindings).build();
64+
InspectConfig.FindingLimits.newBuilder()
65+
.setMaxFindingsPerItem(maxFindings)
66+
.build();
6367

6468
InspectConfig inspectConfig =
6569
InspectConfig.newBuilder()
@@ -76,8 +80,10 @@ public static void main(String[] args) throws Exception {
7680
.build();
7781
ContentItem contentItem = ContentItem.newBuilder().setByteItem(byteContentItem).build();
7882

83+
String projectId = ServiceOptions.getDefaultProjectId();
7984
InspectContentRequest request =
8085
InspectContentRequest.newBuilder()
86+
.setParent(ProjectName.of(projectId).toString())
8187
.setInspectConfig(inspectConfig)
8288
.setItem(contentItem)
8389
.build();

dlp/src/test/java/com/example/dlp/QuickStartIT.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,9 @@
1616

1717
package com.example.dlp;
1818

19+
import static org.hamcrest.CoreMatchers.containsString;
1920
import static org.junit.Assert.assertNotNull;
21+
import static org.junit.Assert.assertThat;
2022
import static org.junit.Assert.assertTrue;
2123

2224
import java.io.ByteArrayOutputStream;
@@ -47,7 +49,8 @@ public void setUp() {
4749
public void testQuickStart() throws Exception {
4850
QuickStart.main(new String[] {});
4951
String output = bout.toString();
50-
assertTrue(output.contains("US_MALE_NAME"));
52+
53+
assertThat(output, containsString("US_MALE_NAME"));
5154
}
5255

5356
@After

0 commit comments

Comments
 (0)