Assert results variable instead of dataFrame. (elastic#1311)

williamrandolph · Jul 2, 2019 · 2b9db87 · 2b9db87
1 parent ae77e1c
commit 2b9db87
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...sql-13/src/itest/java/org/elasticsearch/spark/integration/AbstractJavaEsSparkSQLTest.java b/...sql-13/src/itest/java/org/elasticsearch/spark/integration/AbstractJavaEsSparkSQLTest.java
@@ -95,7 +95,8 @@ public void testBasicRead() throws Exception {
 		assertEquals(5, dataFrame.take(5).length);
 		DataFrame results = sqc
 				.sql("SELECT name FROM datfile WHERE id >=1 AND id <=10");
-		assertEquals(10, dataFrame.take(10).length);
+		assertEquals(10, results.take(10).length);
+		assertEquals(10L, results.count());
 	}
 
 	@Test