Skip to content

Commit e8aa231

Browse files
author
Micah Kornfield
committed
Fix formatting
1 parent 0b4df77 commit e8aa231

8 files changed

Lines changed: 151 additions & 95 deletions

File tree

google-cloud-examples/src/main/java/com/google/cloud/examples/securitycenter/snippets/FindingSnippets.java

Lines changed: 49 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -44,13 +44,17 @@ static Finding createFinding(SourceName sourceName, String findingId) {
4444
String resourceName = "//cloudresourcemanager.googleapis.com/organizations/11232";
4545

4646
// Start setting up a request to create a finding in a source.
47-
Finding finding = Finding.newBuilder()
48-
.setParent(sourceName.toString())
49-
.setState(State.ACTIVE).setResourceName(resourceName)
50-
.setEventTime(Timestamp.newBuilder()
51-
.setSeconds(eventTime.getEpochSecond())
52-
.setNanos(eventTime.getNano()))
53-
.setCategory("MEDIUM_RISK_ONE").build();
47+
Finding finding =
48+
Finding.newBuilder()
49+
.setParent(sourceName.toString())
50+
.setState(State.ACTIVE)
51+
.setResourceName(resourceName)
52+
.setEventTime(
53+
Timestamp.newBuilder()
54+
.setSeconds(eventTime.getEpochSecond())
55+
.setNanos(eventTime.getNano()))
56+
.setCategory("MEDIUM_RISK_ONE")
57+
.build();
5458

5559
// Call the API.
5660
Finding response = client.createFinding(sourceName, findingId, finding);
@@ -91,14 +95,17 @@ static Finding createFindingWithSourceProperties(SourceName sourceName) {
9195
ImmutableMap.of("stringKey", stringValue, "numKey", numValue);
9296

9397
// Start setting up a request to create a finding in a source.
94-
Finding finding = Finding.newBuilder()
95-
.setParent(sourceName.toString())
96-
.setState(State.ACTIVE)
97-
.setResourceName(resourceName)
98-
.setEventTime(Timestamp.newBuilder()
99-
.setSeconds(eventTime.getEpochSecond())
100-
.setNanos(eventTime.getNano()))
101-
.putAllSourceProperties(sourceProperties).build();
98+
Finding finding =
99+
Finding.newBuilder()
100+
.setParent(sourceName.toString())
101+
.setState(State.ACTIVE)
102+
.setResourceName(resourceName)
103+
.setEventTime(
104+
Timestamp.newBuilder()
105+
.setSeconds(eventTime.getEpochSecond())
106+
.setNanos(eventTime.getNano()))
107+
.putAllSourceProperties(sourceProperties)
108+
.build();
102109

103110
// Call the API.
104111
Finding response = client.createFinding(sourceName, findingId, finding);
@@ -127,16 +134,21 @@ static Finding updateFinding(FindingName findingName) {
127134
// Define source properties values as protobuf "Value" objects.
128135
Value stringValue = Value.newBuilder().setStringValue("value").build();
129136

130-
FieldMask updateMask = FieldMask.newBuilder()
131-
.addPaths("event_time")
132-
.addPaths("source_properties.stringKey").build();
133-
134-
Finding finding = Finding.newBuilder()
135-
.setName(findingName.toString())
136-
.setEventTime(Timestamp.newBuilder()
137-
.setSeconds(eventTime.getEpochSecond())
138-
.setNanos(eventTime.getNano()))
139-
.putSourceProperties("stringKey", stringValue).build();
137+
FieldMask updateMask =
138+
FieldMask.newBuilder()
139+
.addPaths("event_time")
140+
.addPaths("source_properties.stringKey")
141+
.build();
142+
143+
Finding finding =
144+
Finding.newBuilder()
145+
.setName(findingName.toString())
146+
.setEventTime(
147+
Timestamp.newBuilder()
148+
.setSeconds(eventTime.getEpochSecond())
149+
.setNanos(eventTime.getNano()))
150+
.putSourceProperties("stringKey", stringValue)
151+
.build();
140152

141153
UpdateFindingRequest.Builder request =
142154
UpdateFindingRequest.newBuilder().setFinding(finding).setUpdateMask(updateMask);
@@ -169,7 +181,8 @@ static ImmutableList<ListFindingsResult> listAllFindings(OrganizationName organi
169181
// Call the API.
170182
ListFindingsPagedResponse response = client.listFindings(request.build());
171183

172-
// This creates one list for all findings. If your organization has a large number of findings
184+
// This creates one list for all findings. If your organization has a large number of
185+
// findings
173186
// this can cause out of memory issues. You can process them batches by returning
174187
// the Iterable returned response.iterateAll() directly.
175188
ImmutableList<ListFindingsResult> results = ImmutableList.copyOf(response.iterateAll());
@@ -201,7 +214,8 @@ static ImmutableList<ListFindingsResult> listFilteredFindings(SourceName sourceN
201214
// Call the API.
202215
ListFindingsPagedResponse response = client.listFindings(request.build());
203216

204-
// This creates one list for all findings in the filter.If your organization has a large number of
217+
// This creates one list for all findings in the filter.If your organization has a large
218+
// number of
205219
// findings this can cause out of memory issues. You can process them batches by returning
206220
// the Iterable returned response.iterateAll() directly.
207221
ImmutableList<ListFindingsResult> results = ImmutableList.copyOf(response.iterateAll());
@@ -225,19 +239,21 @@ static ImmutableList<ListFindingsResult> listFindingsAtTime(SourceName sourceNam
225239
// SourceName sourceName = SourceName.of("123234324", "423432321");
226240

227241
// 5 days ago
228-
Instant fiveDaysAgo = Instant.now().minusSeconds(60*60*24*5);
242+
Instant fiveDaysAgo = Instant.now().minusSeconds(60 * 60 * 24 * 5);
229243

230244
ListFindingsRequest.Builder request =
231245
ListFindingsRequest.newBuilder()
232246
.setParent(sourceName.toString())
233-
.setReadTime(Timestamp.newBuilder()
234-
.setSeconds(fiveDaysAgo.getEpochSecond())
235-
.setNanos(fiveDaysAgo.getNano()));
247+
.setReadTime(
248+
Timestamp.newBuilder()
249+
.setSeconds(fiveDaysAgo.getEpochSecond())
250+
.setNanos(fiveDaysAgo.getNano()));
236251

237252
// Call the API.
238253
ListFindingsPagedResponse response = client.listFindings(request.build());
239254

240-
// This creates one list for all findings in the filter.If your organization has a large number of
255+
// This creates one list for all findings in the filter.If your organization has a large
256+
// number of
241257
// findings this can cause out of memory issues. You can process them batches by returning
242258
// the Iterable returned response.iterateAll() directly.
243259
ImmutableList<ListFindingsResult> results = ImmutableList.copyOf(response.iterateAll());
@@ -261,7 +277,7 @@ static TestIamPermissionsResponse testIamPermissions(SourceName sourceName) {
261277
try (SecurityCenterClient client = SecurityCenterClient.create()) {
262278
// SourceName sourceName = SourceName.of("123234324", "423432321");
263279

264-
//Iam permission to test.
280+
// Iam permission to test.
265281
ArrayList permissionsToTest = new ArrayList<>();
266282
permissionsToTest.add("securitycenter.findings.update");
267283

google-cloud-examples/src/main/java/com/google/cloud/examples/securitycenter/snippets/OrganizationSnippets.java

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,11 @@ static OrganizationSettings updateOrganizationSettings(OrganizationName organiza
4949
try (SecurityCenterClient client = SecurityCenterClient.create()) {
5050
// Start setting up a request to update OrganizationSettings for.
5151
// OrganizationName organizationName = OrganizationName.of("123234324");
52-
OrganizationSettings organizationSettings = OrganizationSettings.newBuilder()
53-
.setName(organizationName.toString() + "/organizationSettings")
54-
.setEnableAssetDiscovery(true).build();
52+
OrganizationSettings organizationSettings =
53+
OrganizationSettings.newBuilder()
54+
.setName(organizationName.toString() + "/organizationSettings")
55+
.setEnableAssetDiscovery(true)
56+
.build();
5557
FieldMask updateMask = FieldMask.newBuilder().addPaths("enable_asset_discovery").build();
5658

5759
UpdateOrganizationSettingsRequest.Builder request =

google-cloud-examples/src/main/java/com/google/cloud/examples/securitycenter/snippets/SecurityMarkSnippets.java

Lines changed: 52 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -34,16 +34,19 @@ static SecurityMarks addToAsset(Asset asset) {
3434
ImmutableMap markMap = ImmutableMap.of("key_a", "value_a", "key_b", "value_b");
3535

3636
// Add security marks and field mask for security marks.
37-
SecurityMarks securityMarks = SecurityMarks.newBuilder()
38-
.setName(asset.getName() + "/securityMarks")
39-
.putAllMarks(markMap).build();
40-
FieldMask updateMask = FieldMask.newBuilder()
41-
.addPaths("marks.key_a")
42-
.addPaths("marks.key_b").build();
37+
SecurityMarks securityMarks =
38+
SecurityMarks.newBuilder()
39+
.setName(asset.getName() + "/securityMarks")
40+
.putAllMarks(markMap)
41+
.build();
42+
FieldMask updateMask =
43+
FieldMask.newBuilder().addPaths("marks.key_a").addPaths("marks.key_b").build();
4344

44-
UpdateSecurityMarksRequest request = UpdateSecurityMarksRequest.newBuilder()
45-
.setSecurityMarks(securityMarks)
46-
.setUpdateMask(updateMask).build();
45+
UpdateSecurityMarksRequest request =
46+
UpdateSecurityMarksRequest.newBuilder()
47+
.setSecurityMarks(securityMarks)
48+
.setUpdateMask(updateMask)
49+
.build();
4750

4851
// Call the API.
4952
SecurityMarks response = client.updateSecurityMarks(request);
@@ -67,14 +70,16 @@ static SecurityMarks clearFromAsset(Asset asset) {
6770
try (SecurityCenterClient client = SecurityCenterClient.create()) {
6871
// Start setting up a request to clear security marks for an asset.
6972
// Create security mark and field mask for clearing security marks.
70-
SecurityMarks securityMarks = SecurityMarks.newBuilder()
71-
.setName(asset.getName() + "/securityMarks").build();
72-
FieldMask updateMask = FieldMask.newBuilder()
73-
.addPaths("marks.key_a").addPaths("marks.key_b").build();
74-
75-
UpdateSecurityMarksRequest request = UpdateSecurityMarksRequest.newBuilder()
76-
.setSecurityMarks(securityMarks)
77-
.setUpdateMask(updateMask).build();
73+
SecurityMarks securityMarks =
74+
SecurityMarks.newBuilder().setName(asset.getName() + "/securityMarks").build();
75+
FieldMask updateMask =
76+
FieldMask.newBuilder().addPaths("marks.key_a").addPaths("marks.key_b").build();
77+
78+
UpdateSecurityMarksRequest request =
79+
UpdateSecurityMarksRequest.newBuilder()
80+
.setSecurityMarks(securityMarks)
81+
.setUpdateMask(updateMask)
82+
.build();
7883

7984
// Call the API.
8085
SecurityMarks response = client.updateSecurityMarks(request);
@@ -98,15 +103,19 @@ static SecurityMarks deleteAndUpdateMarks(Asset asset) {
98103
try (SecurityCenterClient client = SecurityCenterClient.create()) {
99104
// Start setting up a request to clear and update security marks for an asset.
100105
// Create security mark and field mask for clearing security marks.
101-
SecurityMarks securityMarks = SecurityMarks.newBuilder()
102-
.setName(asset.getName() + "/securityMarks")
103-
.putMarks("key_a", "new_value_for_a").build();
104-
FieldMask updateMask = FieldMask.newBuilder()
105-
.addPaths("marks.key_a").addPaths("marks.key_b").build();
106+
SecurityMarks securityMarks =
107+
SecurityMarks.newBuilder()
108+
.setName(asset.getName() + "/securityMarks")
109+
.putMarks("key_a", "new_value_for_a")
110+
.build();
111+
FieldMask updateMask =
112+
FieldMask.newBuilder().addPaths("marks.key_a").addPaths("marks.key_b").build();
106113

107-
UpdateSecurityMarksRequest request = UpdateSecurityMarksRequest.newBuilder()
108-
.setSecurityMarks(securityMarks)
109-
.setUpdateMask(updateMask).build();
114+
UpdateSecurityMarksRequest request =
115+
UpdateSecurityMarksRequest.newBuilder()
116+
.setSecurityMarks(securityMarks)
117+
.setUpdateMask(updateMask)
118+
.build();
110119

111120
// Call the API.
112121
SecurityMarks response = client.updateSecurityMarks(request);
@@ -129,18 +138,22 @@ static SecurityMarks deleteAndUpdateMarks(Asset asset) {
129138
static SecurityMarks addToFinding(Finding finding) {
130139
try (SecurityCenterClient client = SecurityCenterClient.create()) {
131140
// Start setting up a request to add security marks for a finding.
132-
ImmutableMap markMap =
133-
ImmutableMap.of("key_a", "value_a", "key_b", "value_b");
141+
ImmutableMap markMap = ImmutableMap.of("key_a", "value_a", "key_b", "value_b");
134142

135143
// Add security marks and field mask for security marks.
136-
SecurityMarks securityMarks = SecurityMarks.newBuilder()
137-
.setName(finding.getName() + "/securityMarks")
138-
.putAllMarks(markMap).build();
139-
FieldMask updateMask = FieldMask.newBuilder()
140-
.addPaths("marks.key_a").addPaths("marks.key_b").build();
144+
SecurityMarks securityMarks =
145+
SecurityMarks.newBuilder()
146+
.setName(finding.getName() + "/securityMarks")
147+
.putAllMarks(markMap)
148+
.build();
149+
FieldMask updateMask =
150+
FieldMask.newBuilder().addPaths("marks.key_a").addPaths("marks.key_b").build();
141151

142-
UpdateSecurityMarksRequest request = UpdateSecurityMarksRequest.newBuilder()
143-
.setSecurityMarks(securityMarks).setUpdateMask(updateMask).build();
152+
UpdateSecurityMarksRequest request =
153+
UpdateSecurityMarksRequest.newBuilder()
154+
.setSecurityMarks(securityMarks)
155+
.setUpdateMask(updateMask)
156+
.build();
144157

145158
// Call the API.
146159
SecurityMarks response = client.updateSecurityMarks(request);
@@ -160,15 +173,15 @@ static SecurityMarks addToFinding(Finding finding) {
160173
* @param organizationName The organization to list assets for.
161174
*/
162175
// [START list_assets_with_filter]
163-
static ImmutableList<ListAssetsResult> listAssetsWithQueryMarks(OrganizationName organizationName) {
176+
static ImmutableList<ListAssetsResult> listAssetsWithQueryMarks(
177+
OrganizationName organizationName) {
164178
try (SecurityCenterClient client = SecurityCenterClient.create()) {
165179
// Start setting up a request for to list all assets filtered by a specific security mark.
166180
// OrganizationName organizationName = OrganizationName.of("123234324");
167181
ListAssetsRequest request =
168182
ListAssetsRequest.newBuilder()
169183
.setParent(organizationName.toString())
170-
.setFilter(
171-
"security_marks.marks.key_a = \"value_a\"")
184+
.setFilter("security_marks.marks.key_a = \"value_a\"")
172185
.build();
173186

174187
// Call the API.
@@ -206,7 +219,8 @@ static ImmutableList<ListFindingsResult> listFindingsWithQueryMarks(SourceName s
206219
// Call the API.
207220
ListFindingsPagedResponse response = client.listFindings(request.build());
208221

209-
// This creates one list for all findings in the filter.If your organization has a large number of
222+
// This creates one list for all findings in the filter.If your organization has a large
223+
// number of
210224
// findings this can cause out of memory issues. You can process them batches by returning
211225
// the Iterable returned response.iterateAll() directly.
212226
ImmutableList<ListFindingsResult> results = ImmutableList.copyOf(response.iterateAll());

google-cloud-examples/src/main/java/com/google/cloud/examples/securitycenter/snippets/SourceSnippets.java

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,11 @@ static Source createSource(OrganizationName organizationName) {
3131
try (SecurityCenterClient client = SecurityCenterClient.create()) {
3232
// Start setting up a request to create a source in an organization.
3333
// OrganizationName organizationName = OrganizationName.of("123234324");
34-
Source source = Source.newBuilder().setDisplayName("Customized Display Name")
35-
.setDescription("A new custom source that does X").build();
34+
Source source =
35+
Source.newBuilder()
36+
.setDisplayName("Customized Display Name")
37+
.setDescription("A new custom source that does X")
38+
.build();
3639

3740
CreateSourceRequest.Builder request =
3841
CreateSourceRequest.newBuilder().setParent(organizationName.toString()).setSource(source);
@@ -87,8 +90,11 @@ static Source updateSource(SourceName sourceName) {
8790
try (SecurityCenterClient client = SecurityCenterClient.create()) {
8891
// Start setting up a request to update a source.
8992
// SourceName sourceName = SourceName.of("123234324", "423432321");
90-
Source source = Source.newBuilder().setDisplayName("Updated Display Name")
91-
.setName(sourceName.toString()).build();
93+
Source source =
94+
Source.newBuilder()
95+
.setDisplayName("Updated Display Name")
96+
.setName(sourceName.toString())
97+
.build();
9298
FieldMask updateMask = FieldMask.newBuilder().addPaths("display_name").build();
9399

94100
UpdateSourceRequest.Builder request =
@@ -140,9 +146,11 @@ static Policy setIamPolicySource(SourceName sourceName) {
140146
// Set up IAM Policy for the user [email protected] to use the role findingsEditor.
141147
// The user must be a valid google account.
142148
Policy oldPolicy = client.getIamPolicy(sourceName.toString());
143-
Binding bindings = Binding.newBuilder()
144-
.setRole("roles/securitycenter.findingsEditor")
145-
.addMembers("user:[email protected]").build();
149+
Binding bindings =
150+
Binding.newBuilder()
151+
.setRole("roles/securitycenter.findingsEditor")
152+
.addMembers("user:[email protected]")
153+
.build();
146154
Policy policy = oldPolicy.toBuilder().addBindings(bindings).build();
147155

148156
// Start setting up a request to set IAM policy for a source.
@@ -185,4 +193,4 @@ static Policy getIamPolicySource(SourceName sourceName) {
185193
}
186194
// [END get_source_iam_policy]
187195

188-
}
196+
}

google-cloud-examples/src/test/java/com/google/cloud/examples/securitycenter/snippets/ITFindingSnippets.java

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@ public class ITFindingSnippets {
2020
@BeforeClass
2121
public static void setUp() throws IOException {
2222
SOURCE_NAME = SourceName.parse(SourceSnippets.createSource(getOrganizationId()).getName());
23-
FINDING_NAME = FindingName.parse(FindingSnippets.createFinding(SOURCE_NAME, "testfindingid").getName());
23+
FINDING_NAME =
24+
FindingName.parse(FindingSnippets.createFinding(SOURCE_NAME, "testfindingid").getName());
2425
}
2526

2627
@Test
@@ -36,7 +37,11 @@ public void testCreateFindingWithSourceProperties() throws IOException {
3637
@Test
3738
public void testUpdateFinding() throws IOException {
3839
Value stringValue = Value.newBuilder().setStringValue("value").build();
39-
assertTrue(FindingSnippets.updateFinding(FINDING_NAME).getSourcePropertiesMap().get("stringKey").equals(stringValue));
40+
assertTrue(
41+
FindingSnippets.updateFinding(FINDING_NAME)
42+
.getSourcePropertiesMap()
43+
.get("stringKey")
44+
.equals(stringValue));
4045
}
4146

4247
@Test
@@ -56,11 +61,13 @@ public void testListFindingsAtTime() throws IOException {
5661

5762
@Test
5863
public void testTestIamPermissions() throws IOException {
59-
assertTrue(FindingSnippets.testIamPermissions(SOURCE_NAME).getPermissions(0).equals("securitycenter.findings.update"));
64+
assertTrue(
65+
FindingSnippets.testIamPermissions(SOURCE_NAME)
66+
.getPermissions(0)
67+
.equals("securitycenter.findings.update"));
6068
}
6169

6270
private static OrganizationName getOrganizationId() {
6371
return OrganizationName.of(System.getenv("GCLOUD_ORGANIZATION"));
6472
}
65-
6673
}

0 commit comments

Comments
 (0)