---

lucasoares · sduskis · commit 94dae17d1a59 · 2019-04-11T16:16:32.000-04:00
yaml --- r: 27627 b: refs/heads/autosynth-datalabeling c: 47884a3 h: refs/heads/master i: 27625: 77b878e 27623: dbb7150
diff --git a/[refs] b/[refs]
@@ -180,7 +180,7 @@ refs/heads/v4support: aa837fd70877f5a0628d8036e88952db035b792c
 refs/tags/v0.82.0: 7b9807d5d0a400c757b8905fee768be4c85eba25
 refs/tags/v0.83.0: 370ec5a1131a86b36db8efce4f1a943607de8a60
 refs/tags/v0.84.0: 71e85198495a39f4524afa2669434b5075c17c3d
-refs/heads/autosynth-datalabeling: f9ffc7a9a6b1683f3e9b41ff265d58e305576645
+refs/heads/autosynth-datalabeling: 47884a3c08d216d4ae7a7a891291af717fb47ef5
 refs/heads/autosynth-webrisk: e006df2927fba3a0af121e0afd18f73da01a8214
 "refs/heads/cscc_examples": 2813feef0c9a80293739b67fd53a4b4a2f856b24
 refs/heads/marktaskga: 9b161cd985cf38ee44a5e2ab6c53fc00e50a9184
diff --git a/branches/autosynth-datalabeling/google-cloud-clients/google-cloud-datastore/src/main/java/com/google/cloud/datastore/testing/LocalDatastoreHelper.java b/branches/autosynth-datalabeling/google-cloud-clients/google-cloud-datastore/src/main/java/com/google/cloud/datastore/testing/LocalDatastoreHelper.java
@@ -78,10 +78,10 @@ public class LocalDatastoreHelper extends BaseEmulatorHelper<DatastoreOptions> {
     }
   }
 
-  private LocalDatastoreHelper(double consistency) {
+  private LocalDatastoreHelper(double consistency, int port) {
     super(
         "datastore",
-        BaseEmulatorHelper.findAvailablePort(DEFAULT_PORT),
+        port > 0 ? port : BaseEmulatorHelper.findAvailablePort(DEFAULT_PORT),
         PROJECT_ID_PREFIX + UUID.randomUUID().toString());
     Path tmpDirectory = null;
     try {
@@ -162,7 +162,32 @@ public double getConsistency() {
    *     consistency of non-ancestor queries; non-ancestor queries are eventually consistent.
    */
   public static LocalDatastoreHelper create(double consistency) {
-    return new LocalDatastoreHelper(consistency);
+    return create(consistency, 0);
+  }
+
+  /**
+   * Creates a local Datastore helper with the specified settings for project ID and consistency.
+   *
+   * @param consistency the fraction of Datastore writes that are immediately visible to global
+   *     queries, with 0.0 meaning no writes are immediately visible and 1.0 meaning all writes are
+   *     immediately visible. Note that setting this to 1.0 may mask incorrect assumptions about the
+   *     consistency of non-ancestor queries; non-ancestor queries are eventually consistent.
+   * @param port the port to be used to start the emulator service. Note that setting this to 0 the
+   *     emulator will search for a free random port.
+   */
+  public static LocalDatastoreHelper create(double consistency, int port) {
+    return new LocalDatastoreHelper(consistency, port);
+  }
+
+  /**
+   * Creates a local Datastore helper with a placeholder project ID and the default consistency
+   * setting of 0.9.
+   *
+   * @param port the port to be used to start the emulator service. Note that setting this to 0 the
+   *     emulator will search for a free random port.
+   */
+  public static LocalDatastoreHelper create(int port) {
+    return new LocalDatastoreHelper(DEFAULT_CONSISTENCY, port);
   }
 
   /**
diff --git a/branches/autosynth-datalabeling/google-cloud-clients/google-cloud-datastore/src/test/java/com/google/cloud/datastore/testing/LocalDatastoreHelperTest.java b/branches/autosynth-datalabeling/google-cloud-clients/google-cloud-datastore/src/test/java/com/google/cloud/datastore/testing/LocalDatastoreHelperTest.java
@@ -17,6 +17,7 @@
 package com.google.cloud.datastore.testing;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertNull;
 import static org.junit.Assert.assertSame;
@@ -56,6 +57,23 @@ public void testCreate() {
     assertTrue(helper.getProjectId().startsWith(PROJECT_ID_PREFIX));
   }
 
+  @Test
+  public void testCreatePort() {
+    LocalDatastoreHelper helper = LocalDatastoreHelper.create(0.75, 8888);
+    DatastoreOptions options = helper.getOptions(NAMESPACE);
+    assertTrue(options.getHost().endsWith("8888"));
+    assertTrue(Math.abs(0.75 - helper.getConsistency()) < TOLERANCE);
+    helper = LocalDatastoreHelper.create();
+    options = helper.getOptions(NAMESPACE);
+    assertTrue(Math.abs(0.9 - helper.getConsistency()) < TOLERANCE);
+    assertFalse(options.getHost().endsWith("8888"));
+    assertFalse(options.getHost().endsWith("8080"));
+    helper = LocalDatastoreHelper.create(9999);
+    options = helper.getOptions(NAMESPACE);
+    assertTrue(Math.abs(0.9 - helper.getConsistency()) < TOLERANCE);
+    assertTrue(options.getHost().endsWith("9999"));
+  }
+
   @Test
   public void testOptions() {
     LocalDatastoreHelper helper = LocalDatastoreHelper.create();
