schemasafe#127 Add test cases to cover aggregates

Author: OmniaGM

cql-parser/src/test/scala/troy/cql3_3/lexical/DataManipulationParsingSpec.scala

@@ -0,0 +1,46 @@
+package troy.cql3_3.lexical
+
+import org.scalatest.{ FlatSpec, Matchers }
+import troy.cql.ast.dml.Select._
+import troy.cql.parser.ParserTestUtils._
+
+class DataManipulationParsingSpec extends FlatSpec with Matchers {
+  "SELECT parser" should "parse asterisk" in {
+    parseSelect("SELECT * FROM test;").selection shouldBe Asterisk
+  }
+
+  it should "parse DISTINCT partition" in {
+    parseSelect("SELECT DISTINCT * FROM test;").selection shouldBe Asterisk
+  }
+
+  it should "parse JSON partition" in {
+    parseSelect("SELECT JSON * FROM test;").mod.get shouldBe Json
+  }
+
+  //  TODO: https://github.com/cassandra-scala/troy/issues/134
+  "Standard aggregates" should "parse count" in {
+    val selectClause: SelectClause = parseSelect("SELECT COUNT(*) FROM test;").selection.asInstanceOf[SelectClause]
+    selectClause.items(0).asInstanceOf[SelectionClauseItem].selector shouldBe Count
+  }
+
+  it should "parse max" ignore {
+    parseSelect("SELECT MAX(*) FROM test;")
+  }
+
+  it should "parse min" ignore {
+    parseSelect("SELECT MIN(*) FROM test;")
+  }
+
+  it should "parse sum" ignore {
+    parseSelect("SELECT SUM(*) FROM test;")
+  }
+
+  it should "parse avg" ignore {
+    parseSelect("SELECT AVG(*) FROM test;")
+  }
+
+  //  TODO: https://github.com/cassandra-scala/troy/issues/137
+  "User-defined aggregate" should "be parsed" ignore {
+    parseQuery("CREATE AGGREGATE min_value(double) SFUNC state_min_int STYPE double INITCOND null;")
+  }
+}

troy-schema/src/test/scala/troy/cql3_3/lexical/DataManipulationSchemaSpec.scala

@@ -0,0 +1,57 @@
+package troy.cql3_3.lexical
+
+import org.scalatest._
+import troy.cql.ast.DataType
+import troy.schema.SchemaTestUtils._
+
+class DataManipulationSchemaSpec extends FlatSpec with Matchers {
+  implicit lazy val schema = buildSchema(
+    """
+      | CREATE KEYSPACE specs WITH replication = {'class': 'SimpleStrategy' , 'replication_factor': '1'};
+      | CREATE TABLE specs.test (
+      |  foo int PRIMARY KEY,
+      |  bar TEXT,
+      |  barfoo TEXT
+      | );
+    """.stripMargin
+  )
+
+  // TODO: https://github.com/cassandra-scala/troy/issues/136
+  "Schema" should "support select *" ignore {
+    asteriskOf("SELECT * FROM specs.test;").size shouldBe 3
+  }
+
+  it should "support select with DISTINCT partition" in {
+    columnsOf("SELECT DISTINCT foo FROM specs.test;").head shouldBe DataType.Int
+  }
+
+  it should "support select with JSON formate" in {
+    columnsOf("SELECT JSON foo FROM specs.test;").head shouldBe DataType.Int
+  }
+
+  //  TODO: https://github.com/cassandra-scala/troy/issues/134
+  it should "support select with count aggregate function" ignore {
+    columnsOf("SELECT count(foo) FROM specs.test;").head shouldBe DataType.Int
+  }
+
+  it should "support select with min aggregate function" ignore {
+    columnsOf("SELECT min(foo) FROM specs.test;").head shouldBe DataType.Int
+  }
+
+  it should "support select with max aggregate function" ignore {
+    columnsOf("SELECT max(foo) FROM specs.test;").head shouldBe DataType.Int
+  }
+
+  it should "support select with sum aggregate function" ignore {
+    columnsOf("SELECT sum(foo) FROM specs.test;").head shouldBe DataType.Int
+  }
+
+  it should "support select with avg aggregate function" ignore {
+    columnsOf("SELECT avg(foo) FROM specs.test;").head shouldBe DataType.Int
+  }
+
+  //  TODO: https://github.com/cassandra-scala/troy/issues/137
+  it should "support select with User-defined aggregate function" ignore {
+    columnsOf("SELECT UDAggregate(foo) FROM specs.test;").head shouldBe DataType.Int
+  }
+}

troy-schema/src/test/scala/troy/schema/SchemaTestUtils.scala

@@ -1,9 +1,9 @@
 package troy.schema
 
 import org.scalatest.{ FlatSpec, Matchers }
-import troy.cql.parser.ParserTestUtils.{ parseSchema, parseQuery }
+import troy.cql.parser.ParserTestUtils.{ parseQuery, parseSchema }
 import VTestUtils._
-import troy.schema.SchemaEngine.Columns
+import troy.schema.SchemaEngine.{ Asterisk, Columns }
 
 object SchemaTestUtils extends FlatSpec with Matchers {
   def buildSchema(versionedStatements: String*) =
@@ -12,6 +12,9 @@ object SchemaTestUtils extends FlatSpec with Matchers {
   def analyse(query: String)(implicit schema: VersionedSchemaEngine) =
     schema(parseQuery(query))
 
+  def asteriskOf(query: String)(implicit schema: VersionedSchemaEngine) =
+    analyse(query).get._1.asInstanceOf[Asterisk].types
+
   def columnsOf(query: String)(implicit schema: VersionedSchemaEngine) =
     analyse(query).get._1.asInstanceOf[Columns].types