Add tests with different operators

cycle · Nov 29, 2023 · cc23e1e · cc23e1e
1 parent 32537f4
commit cc23e1e
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 1 deletion.
diff --git a/src/Driver/Compiler.php b/src/Driver/Compiler.php
@@ -475,7 +475,7 @@ protected function condition(QueryParameters $params, Quoter $q, array $context)
 
         $placeholder = '?';
         if ($value->isArray()) {
-            return $this->arrayToInOperator($params, $q, $value->getValue(), $operator === 'IN');
+            return $this->arrayToInOperator($params, $q, $value->getValue(), $operator === 'IN' || $operator === '=');
         }
 
         if ($value->isNull()) {

diff --git a/tests/Database/Functional/Driver/Common/Query/SelectQueryTest.php b/tests/Database/Functional/Driver/Common/Query/SelectQueryTest.php
@@ -2166,6 +2166,62 @@ public function testSelectWithFragmentedColumns(): void
         );
     }
 
+    public function testWhereInWithoutSpecifiedOperator(): void
+    {
+        $select = $this->database
+            ->select()
+            ->from(['users'])
+            ->where(
+                'uuid',
+                new Parameter(['12345678-1234-1234-1234-123456789012', '12345678-1234-1234-1234-123456789013'])
+            );
+
+        $this->assertSameQuery('SELECT * FROM {users} WHERE {uuid} IN (?, ?)', $select);
+
+        $this->assertSameParameters(
+            ['12345678-1234-1234-1234-123456789012', '12345678-1234-1234-1234-123456789013'],
+            $select,
+        );
+    }
+
+    public function testWhereInWithEqualSpecifiedOperator(): void
+    {
+        $select = $this->database
+            ->select()
+            ->from(['users'])
+            ->where(
+                'uuid',
+                '=',
+                new Parameter(['12345678-1234-1234-1234-123456789012', '12345678-1234-1234-1234-123456789013'])
+            );
+
+        $this->assertSameQuery('SELECT * FROM {users} WHERE {uuid} IN (?, ?)', $select);
+
+        $this->assertSameParameters(
+            ['12345678-1234-1234-1234-123456789012', '12345678-1234-1234-1234-123456789013'],
+            $select,
+        );
+    }
+
+    public function testWhereInWithNotEqualSpecifiedOperator(): void
+    {
+        $select = $this->database
+            ->select()
+            ->from(['users'])
+            ->where(
+                'uuid',
+                '!=',
+                new Parameter(['12345678-1234-1234-1234-123456789012', '12345678-1234-1234-1234-123456789013'])
+            );
+
+        $this->assertSameQuery('SELECT * FROM {users} WHERE {uuid} NOT IN (?, ?)', $select);
+
+        $this->assertSameParameters(
+            ['12345678-1234-1234-1234-123456789012', '12345678-1234-1234-1234-123456789013'],
+            $select,
+        );
+    }
+
     public function testFragmentInWhereInClause(): void
     {
         $select = $this->database