From 0eec1bbee2d25f4fa15667376723133013cb8ef2 Mon Sep 17 00:00:00 2001
From: Himanshu Gupta <g.himanshu@gmail.com>
Date: Wed, 15 Jul 2015 13:19:36 -0500
Subject: [PATCH] json serde tests for HadoopTuningConfig

---
 .../druid/indexer/HadoopTuningConfigTest.java | 93 +++++++++++++++++++
 1 file changed, 93 insertions(+)
 create mode 100644 indexing-hadoop/src/test/java/io/druid/indexer/HadoopTuningConfigTest.java

diff --git a/indexing-hadoop/src/test/java/io/druid/indexer/HadoopTuningConfigTest.java b/indexing-hadoop/src/test/java/io/druid/indexer/HadoopTuningConfigTest.java
new file mode 100644
index 00000000000..e7bbd46f428
--- /dev/null
+++ b/indexing-hadoop/src/test/java/io/druid/indexer/HadoopTuningConfigTest.java
@@ -0,0 +1,93 @@
+/*
+* Licensed to Metamarkets Group Inc. (Metamarkets) under one
+* or more contributor license agreements. See the NOTICE file
+* distributed with this work for additional information
+* regarding copyright ownership. Metamarkets licenses this file
+* to you under the Apache License, Version 2.0 (the
+* "License"); you may not use this file except in compliance
+* with the License. You may obtain a copy of the License at
+*
+* http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing,
+* software distributed under the License is distributed on an
+* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+* KIND, either express or implied. See the License for the
+* specific language governing permissions and limitations
+* under the License.
+*/
+
+package io.druid.indexer;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.google.common.base.Throwables;
+import com.google.common.collect.ImmutableMap;
+import io.druid.jackson.DefaultObjectMapper;
+import io.druid.segment.IndexSpec;
+import org.joda.time.DateTime;
+import org.junit.Assert;
+import org.junit.Test;
+
+import java.util.List;
+
+/**
+ */
+public class HadoopTuningConfigTest
+{
+  private static final ObjectMapper jsonMapper = new DefaultObjectMapper();
+
+  @Test
+  public void testSerde() throws Exception
+  {
+    HadoopTuningConfig expected = new HadoopTuningConfig(
+        "/tmp/workingpath",
+        "version",
+        null,
+        null,
+        null,
+        100,
+        true,
+        true,
+        true,
+        true,
+        null,
+        true,
+        true,
+        true,
+        200,
+        0.1f,
+        true
+    );
+
+    HadoopTuningConfig actual = jsonReadWriteRead(jsonMapper.writeValueAsString(expected), HadoopTuningConfig.class);
+
+    Assert.assertEquals("/tmp/workingpath", actual.getWorkingPath());
+    Assert.assertEquals("version", actual.getVersion());
+    Assert.assertNotNull(actual.getPartitionsSpec());
+    Assert.assertEquals(ImmutableMap.<DateTime, List<HadoopyShardSpec>>of(), actual.getShardSpecs());
+    Assert.assertEquals(new IndexSpec(), actual.getIndexSpec());
+    Assert.assertEquals(100, actual.getRowFlushBoundary());
+    Assert.assertEquals(true, actual.isLeaveIntermediate());
+    Assert.assertEquals(true, actual.isCleanupOnFailure());
+    Assert.assertEquals(true, actual.isOverwriteFiles());
+    Assert.assertEquals(true, actual.isIgnoreInvalidRows());
+    Assert.assertEquals(ImmutableMap.<String, String>of(), actual.getJobProperties());
+    Assert.assertEquals(true, actual.isCombineText());
+    Assert.assertEquals(true, actual.isPersistInHeap());
+    Assert.assertEquals(true, actual.isIngestOffheap());
+    Assert.assertEquals(200, actual.getBufferSize());
+    Assert.assertEquals(0.1f, actual.getAggregationBufferRatio(), 0.0001);
+    Assert.assertEquals(true, actual.getUseCombiner());
+
+  }
+
+  public static <T> T jsonReadWriteRead(String s, Class<T> klass)
+  {
+    try {
+      return jsonMapper.readValue(jsonMapper.writeValueAsBytes(jsonMapper.readValue(s, klass)), klass);
+    }
+    catch (Exception e) {
+      throw Throwables.propagate(e);
+    }
+  }
+}