Merge pull request #1032 from dengpanyin/bug-fix

SAMZA-2193: Fix unit-test failure after making rate limiter as a required config
apache · May 15, 2019 · c4afc2e · c4afc2e
2 parents 077af85 + df36b42
commit c4afc2e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/...re/src/test/java/org/apache/samza/table/remote/descriptors/TestRemoteTableDescriptor.java b/...re/src/test/java/org/apache/samza/table/remote/descriptors/TestRemoteTableDescriptor.java
@@ -217,6 +217,7 @@ public void testTablePartToConfig() {
   public void testTableRetryPolicyToConfig() {
     Map<String, String> tableConfig = new RemoteTableDescriptor("1").withReadFunction(createMockTableReadFunction())
         .withReadRetryPolicy(new TableRetryPolicy())
+        .withRateLimiterDisabled()
         .toConfig(new MapConfig());
     Assert.assertEquals(tableConfig.get("tables.1.io.read.retry.policy.TableRetryPolicy"),
         "{\"exponentialFactor\":0.0,\"backoffType\":\"NONE\",\"retryPredicate\":{}}");

diff --git a/samza-test/src/test/java/org/apache/samza/test/table/TestCouchbaseRemoteTableEndToEnd.java b/samza-test/src/test/java/org/apache/samza/test/table/TestCouchbaseRemoteTableEndToEnd.java
@@ -143,11 +143,13 @@ public void testEndToEnd() throws Exception {
               .withBootstrapHttpDirectPort(couchbaseMock.getHttpPort());
 
       RemoteTableDescriptor<String, String> inputTableDesc = new RemoteTableDescriptor<>("input-table");
-      inputTableDesc.withReadFunction(readFunction);
+      inputTableDesc.withReadFunction(readFunction).withRateLimiterDisabled();
       Table<KV<String, String>> inputTable = appDesc.getTable(inputTableDesc);
 
       RemoteTableDescriptor<String, JsonObject> outputTableDesc = new RemoteTableDescriptor<>("output-table");
-      outputTableDesc.withReadFunction(new DummyReadFunction<>()).withWriteFunction(writeFunction);
+      outputTableDesc.withReadFunction(new DummyReadFunction<>())
+          .withWriteFunction(writeFunction)
+          .withRateLimiterDisabled();
       Table<KV<String, JsonObject>> outputTable = appDesc.getTable(outputTableDesc);
 
       appDesc.getInputStream(inputDescriptor)