Merge b4c341e into bbb4abe

schemasafe · Oct 19, 2016 · dc13ef5 · dc13ef5
2 parents bbb4abe + b4c341e
commit dc13ef5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/troy-schema/src/main/scala/troy/schema/ColumnOps.scala b/troy-schema/src/main/scala/troy/schema/ColumnOps.scala
@@ -8,9 +8,9 @@ object ColumnOps {
     import Operator._
     import DataType._
 
-    // TODO: Add NotEquals, In
+    // TODO: Add In
     def operandType(operator: Operator): Result[DataType] = operator match {
-      case Equals | LessThan | GreaterThan | LessThanOrEqual | GreaterThanOrEqual =>
+      case Equals | LessThan | GreaterThan | LessThanOrEqual | GreaterThanOrEqual | NotEquals =>
         V.Success(column.dataType).collect(operatorNotSupported(operator)) {
           case n: Native => n
         }

diff --git a/troy-schema/src/test/scala/troy/schema/UpdateSpec.scala b/troy-schema/src/test/scala/troy/schema/UpdateSpec.scala
@@ -58,6 +58,14 @@ class UpdateSpec extends FlatSpec with Matchers {
     variableTypes.isEmpty shouldBe true
   }
 
+  it should "support simple update statement with NotEquals ops" in {
+    val statement = parse("UPDATE test.post_details SET title = 'Testing' WHERE author_id != ?;")
+    val (rowType, variableTypes) = schema(statement).get
+    rowType.asInstanceOf[SchemaEngine.Columns].types.isEmpty shouldBe true
+    variableTypes.size shouldBe 1
+    variableTypes(0) shouldBe DataType.Uuid
+  }
+
   it should "update statement with variables" in {
     val statement = parse("UPDATE test.posts SET post_title = 'Testing' WHERE author_id = ?; ")
     val (rowType, variableTypes) = schema(statement).get