Test aggregation with SQL.

Signed-off-by: Youssef Aouichaoui <[email protected]>

Author: youssef3wi

src/main/java/org/springframework/data/elasticsearch/core/sql/SqlResponse.java

@@ -116,6 +116,11 @@ public Iterator<Map.Entry<Column, JsonValue>> iterator() {
 			return row.entrySet().iterator();
 		}
 
+		@Nullable
+		public JsonValue get(Column column) {
+			return row.get(column);
+		}
+
 		/**
 		 * This method should attempt to convert the SQL response into a document if the columns selected in an SQL query
 		 * correspond to the fields in the document.

src/test/java/org/springframework/data/elasticsearch/core/sql/SqlOperationsIntegrationTests.java

@@ -15,6 +15,7 @@
  */
 package org.springframework.data.elasticsearch.core.sql;
 
+import static org.assertj.core.api.Assertions.assertThat;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
@@ -100,6 +101,20 @@ void when_search_with_an_sql_query_and_txt_format() {
 				"The Elasticsearch Java Client only supports JSON format.");
 	}
 
+	@Test
+	void when_search_with_an_sql_query_that_has_aggregated_column() {
+		// Given
+		SqlQuery query = SqlQuery.builder("SELECT SUM(views) AS TOTAL FROM entity_for_sql").build();
+
+		// When
+
+		// Then
+		SqlResponse response = operations.search(query);
+		assertThat(response.getColumns()).first().extracting(SqlResponse.Column::name).isEqualTo("TOTAL");
+		assertThat(response.getRows()).hasSize(1).first().extracting(row -> row.get(response.getColumns().get(0)))
+				.hasToString("3");
+	}
+
 	// begin region
 	@Document(indexName = "entity_for_sql")
 	static class EntityForSQL {