change snippet response and mock static

security-command-center/snippets/src/main/java/vtwo/bigquery/CreateBigQueryExport.java

@@ -55,7 +55,7 @@ public static void main(String[] args) throws IOException {
 
   // Create export configuration to export findings from a project to a BigQuery dataset.
   // Optionally specify filter to export certain findings only.
-  public static void createBigQueryExport(String organizationId, String location,
+  public static BigQueryExport createBigQueryExport(String organizationId, String location,
       String projectId, String filter, String bigQueryDatasetId, String bigQueryExportId)
       throws IOException {
     // Initialize client that will be used to send requests. This client only needs to be created
@@ -84,6 +84,7 @@ public static void createBigQueryExport(String organizationId, String location,
       BigQueryExport response = client.createBigQueryExport(bigQueryExportRequest);
 
       System.out.printf("BigQuery export request created successfully: %s\n", response.getName());
+      return response;
     }
   }
 }

security-command-center/snippets/src/main/java/vtwo/bigquery/GetBigQueryExport.java

@@ -41,7 +41,7 @@ public static void main(String[] args) throws IOException {
   }
 
   // Retrieve an existing BigQuery export.
-  public static void getBigQueryExport(String organizationId, String location,
+  public static BigQueryExport getBigQueryExport(String organizationId, String location,
       String bigQueryExportId) throws IOException {
     // Initialize client that will be used to send requests. This client only needs to be created
     // once, and can be reused for multiple requests.
@@ -57,6 +57,7 @@ public static void getBigQueryExport(String organizationId, String location,
 
       BigQueryExport response = client.getBigQueryExport(bigQueryExportRequest);
       System.out.printf("Retrieved the BigQuery export: %s", response.getName());
+      return response;
     }
   }
 }

security-command-center/snippets/src/main/java/vtwo/bigquery/ListBigQueryExports.java

@@ -39,7 +39,7 @@ public static void main(String[] args) throws IOException {
   }
 
   // List BigQuery exports in the given parent.
-  public static void listBigQueryExports(String organizationId, String location)
+  public static ListBigQueryExportsPagedResponse listBigQueryExports(String organizationId, String location)
       throws IOException {
     // Initialize client that will be used to send requests. This client only needs to be created
     // once, and can be reused for multiple requests.
@@ -57,6 +57,7 @@ public static void listBigQueryExports(String organizationId, String location)
       for (BigQueryExport bigQueryExport : response.iterateAll()) {
         System.out.println(bigQueryExport.getName());
       }
+      return response;
     }
   }
 }

security-command-center/snippets/src/main/java/vtwo/bigquery/UpdateBigQueryExport.java

@@ -50,7 +50,7 @@ public static void main(String[] args) throws IOException {
   }
 
   // Updates an existing BigQuery export.
-  public static void updateBigQueryExport(String organizationId, String location, String filter,
+  public static BigQueryExport updateBigQueryExport(String organizationId, String location, String filter,
       String bigQueryExportId)
       throws IOException {
     // Initialize client that will be used to send requests. This client only needs to be created
@@ -83,11 +83,8 @@ public static void updateBigQueryExport(String organizationId, String location,
               .build();
 
       BigQueryExport response = client.updateBigQueryExport(request);
-      if (!response.getFilter().equalsIgnoreCase(filter)) {
-        System.out.println("Failed to update BigQueryExport!");
-        return;
-      }
       System.out.println("BigQueryExport updated successfully!");
+      return response;
     }
   }
 }