Assert results variable instead of dataFrame. (#1311)

martijnvg · martijnvg · commit 2b9db8794eb4 · 2019-07-02T06:40:40.000+02:00
diff --git a/spark/sql-13/src/itest/java/org/elasticsearch/spark/integration/AbstractJavaEsSparkSQLTest.java b/spark/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

Original file line number	Diff line number	Diff line change
`@@ -95,7 +95,8 @@ public void testBasicRead() throws Exception {`
`95`	`95`	`assertEquals(5, dataFrame.take(5).length);`
`96`	`96`	`DataFrame results = sqc`
`97`	`97`	`.sql("SELECT name FROM datfile WHERE id >=1 AND id <=10");`
`98`		`- assertEquals(10, dataFrame.take(10).length);`
	`98`	`+ assertEquals(10, results.take(10).length);`
	`99`	`+ assertEquals(10L, results.count());`
`99`	`100`	`}`
`100`	`101`
`101`	`102`	`@Test`