From 37ed9aff985ae809600d24f631d4786dde63e938 Mon Sep 17 00:00:00 2001
From: Marco Ziccardi <marco.ziccard@gmail.com>
Date: Wed, 21 Sep 2016 16:47:35 +0200
Subject: [PATCH] Add of() method to DatasetInfo

---
 .../google/cloud/bigquery/DatasetInfo.java    | 14 ++++++++++
 .../cloud/bigquery/DatasetInfoTest.java       | 28 +++++++++++++++++++
 2 files changed, 42 insertions(+)

diff --git a/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/DatasetInfo.java b/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/DatasetInfo.java
index 3e7da7cc6c87..0ea3f9a36147 100644
--- a/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/DatasetInfo.java
+++ b/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/DatasetInfo.java
@@ -460,6 +460,20 @@ public static Builder builder(String datasetId) {
     return builder(DatasetId.of(datasetId));
   }
 
+  /**
+   * Returns a {@code DatasetInfo} object given it's identity.
+   */
+  public static DatasetInfo of(DatasetId datasetId) {
+    return builder(datasetId).build();
+  }
+
+  /**
+   * Returns a {@code DatasetInfo} object given it's user-defined id.
+   */
+  public static DatasetInfo of(String datasetId) {
+    return builder(datasetId).build();
+  }
+
   /**
    * Returns a builder for the DatasetInfo object given it's user-defined project and dataset ids.
    */
diff --git a/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/DatasetInfoTest.java b/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/DatasetInfoTest.java
index 6e9c961f207f..21d75cb28592 100644
--- a/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/DatasetInfoTest.java
+++ b/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/DatasetInfoTest.java
@@ -107,6 +107,34 @@ public void testBuilder() {
     assertEquals(SELF_LINK, DATASET_INFO_COMPLETE.selfLink());
   }
 
+  @Test
+  public void testOf() {
+    DatasetInfo datasetInfo = DatasetInfo.of(DATASET_ID.dataset());
+    assertEquals(DATASET_ID, datasetInfo.datasetId());
+    assertNull(datasetInfo.acl());
+    assertNull(datasetInfo.creationTime());
+    assertNull(datasetInfo.defaultTableLifetime());
+    assertNull(datasetInfo.description());
+    assertNull(datasetInfo.etag());
+    assertNull(datasetInfo.friendlyName());
+    assertNull(datasetInfo.generatedId());
+    assertNull(datasetInfo.lastModified());
+    assertNull(datasetInfo.location());
+    assertNull(datasetInfo.selfLink());
+    datasetInfo = DatasetInfo.of(DATASET_ID);
+    assertEquals(DATASET_ID, datasetInfo.datasetId());
+    assertNull(datasetInfo.acl());
+    assertNull(datasetInfo.creationTime());
+    assertNull(datasetInfo.defaultTableLifetime());
+    assertNull(datasetInfo.description());
+    assertNull(datasetInfo.etag());
+    assertNull(datasetInfo.friendlyName());
+    assertNull(datasetInfo.generatedId());
+    assertNull(datasetInfo.lastModified());
+    assertNull(datasetInfo.location());
+    assertNull(datasetInfo.selfLink());
+  }
+
   @Test
   public void testToPbAndFromPb() {
     compareDatasets(DATASET_INFO_COMPLETE, DatasetInfo.fromPb(DATASET_INFO_COMPLETE.toPb()));