Repository: camel Updated Branches: refs/heads/master 66600aa81 -> c45c0b3c2
Added Spark Hive collecting test. Project: http://git-wip-us.apache.org/repos/asf/camel/repo Commit: http://git-wip-us.apache.org/repos/asf/camel/commit/c45c0b3c Tree: http://git-wip-us.apache.org/repos/asf/camel/tree/c45c0b3c Diff: http://git-wip-us.apache.org/repos/asf/camel/diff/c45c0b3c Branch: refs/heads/master Commit: c45c0b3c2463d9902cc12f1a38630e0a482bbaa2 Parents: 66600aa Author: Henryk Konsek <hekon...@gmail.com> Authored: Mon Dec 14 21:34:53 2015 +0100 Committer: Henryk Konsek <hekon...@gmail.com> Committed: Mon Dec 14 21:34:53 2015 +0100 ---------------------------------------------------------------------- .../camel/component/spark/SparkProducerTest.java | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/camel/blob/c45c0b3c/components/camel-spark/src/test/java/org/apache/camel/component/spark/SparkProducerTest.java ---------------------------------------------------------------------- diff --git a/components/camel-spark/src/test/java/org/apache/camel/component/spark/SparkProducerTest.java b/components/camel-spark/src/test/java/org/apache/camel/component/spark/SparkProducerTest.java index 311d641..11b6451 100644 --- a/components/camel-spark/src/test/java/org/apache/camel/component/spark/SparkProducerTest.java +++ b/components/camel-spark/src/test/java/org/apache/camel/component/spark/SparkProducerTest.java @@ -18,6 +18,7 @@ package org.apache.camel.component.spark; import java.io.File; import java.io.IOException; +import java.util.List; import static java.lang.Boolean.parseBoolean; import static java.util.Arrays.asList; @@ -30,6 +31,7 @@ import org.apache.spark.api.java.AbstractJavaRDDLike; import org.apache.spark.api.java.JavaRDD; import org.apache.spark.api.java.JavaSparkContext; import org.apache.spark.sql.DataFrame; +import org.apache.spark.sql.Row; import org.apache.spark.sql.hive.HiveContext; import org.junit.BeforeClass; import org.junit.Test; @@ -221,8 +223,15 @@ public class SparkProducerTest extends CamelTestSupport { @Test public void shouldExecuteHiveQuery() { assumeTrue(shouldRunHive); - long tablesCount = template.requestBody(sparkHiveUri + "?collect=false", "SELECT * FROM cars", Long.class); - Truth.assertThat(tablesCount).isEqualTo(2); + List<Row> cars = template.requestBody(sparkHiveUri, "SELECT * FROM cars", List.class); + Truth.assertThat(cars.get(0).getString(1)).isEqualTo("X-trail"); + } + + @Test + public void shouldExecuteHiveCountQuery() { + assumeTrue(shouldRunHive); + long carsCount = template.requestBody(sparkHiveUri + "?collect=false", "SELECT * FROM cars", Long.class); + Truth.assertThat(carsCount).isEqualTo(2); } // Data frames tests
