fix CQL ArrayExpression

packaged · Jan 22, 2015 · d2ec911 · d2ec911
1 parent 76ef2c0
commit d2ec911
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/Assembler/CQL/CqlExpressionAssembler.php b/src/Assembler/CQL/CqlExpressionAssembler.php
@@ -2,6 +2,7 @@
 namespace Packaged\QueryBuilder\Assembler\CQL;
 
 use Packaged\QueryBuilder\Assembler\Segments\ExpressionAssembler;
+use Packaged\QueryBuilder\Expression\ArrayExpression;
 use Packaged\QueryBuilder\Expression\StringExpression;
 
 class CqlExpressionAssembler extends ExpressionAssembler
@@ -10,4 +11,9 @@ public function assembleStringExpression(StringExpression $expr)
   {
     return '\'' . $expr->getValue() . '\'';
   }
+
+  public function assembleArrayExpression(ArrayExpression $expr)
+  {
+    return '(\'' . implode('\',\'', $expr->getValue()) . '\')';
+  }
 }
diff --git a/tests/Assembler/CQL/CqlAssemblerTest.php b/tests/Assembler/CQL/CqlAssemblerTest.php
@@ -8,6 +8,7 @@
 use Packaged\QueryBuilder\Expression\StringExpression;
 use Packaged\QueryBuilder\Expression\TableExpression;
 use Packaged\QueryBuilder\Predicate\BetweenPredicate;
+use Packaged\QueryBuilder\Predicate\InPredicate;
 use Packaged\QueryBuilder\Predicate\PredicateSet;
 use Packaged\QueryBuilder\SelectExpression\AllSelectExpression;
 use Packaged\QueryBuilder\SelectExpression\CountSelectExpression;
@@ -103,6 +104,15 @@ public function testCountExpression()
     $this->assertEquals('COUNT(*)', CqlAssembler::stringify($predicate));
   }
 
+  public function testArrayExpression()
+  {
+    $predicate = InPredicate::create('field', [1, 2, 3]);
+    $this->assertEquals(
+      "\"field\" IN ('1','2','3')",
+      CqlAssembler::stringify($predicate)
+    );
+  }
+
   public function testTableExpression()
   {
     $this->assertEquals(