Merge pull request #10 from michaeljoseph/adds-where-support-to-matched

Adds where support to matched sampler

src/Sampler/Matched.php

@@ -28,6 +28,10 @@ class Matched extends BaseSampler
      * @var array
      */
     protected $constraints;
+    /**
+     * @var array
+     */
+    private $where;
 
     /**
      * Return a unique name for this sampler for informational purposes
@@ -52,6 +56,7 @@ public function loadConfig($config)
     {
         parent::loadConfig($config);
         $this->constraints = (array)$this->demandParameterValue($config, 'constraints');
+        $this->where = $config->where ?? [];
     }
 
     /**
@@ -94,6 +99,10 @@ public function getRows()
             }
         }
 
+        foreach ($this->where as $where) {
+            $queryBuilder->andWhere($where);
+        }
+
         if ($this->limit) {
             $queryBuilder->setMaxResults($this->limit);
         }

tests/SamplerTest.php

@@ -5,6 +5,7 @@
 use Quidco\DbSampler\ReferenceStore;
 use Quidco\DbSampler\Sampler\CopyAll;
 use Quidco\DbSampler\Sampler\CopyEmpty;
+use Quidco\DbSampler\Sampler\Matched;
 
 class SamplerTest extends SqliteBasedTestCase
 {
@@ -35,4 +36,21 @@ public function testCopyAllWithReferenceStore()
 
         $this->assertCount(4, $referenceStore->getReferencesByName('fruit_ids'));
     }
+
+    public function testMatchedWithWhereClause()
+    {
+        $sampler = new Matched();
+        $sampler->setTableName('fruit_x_basket');
+        $sampler->setSourceConnection($this->sourceConnection);
+        $sampler->setDestConnection($this->destConnection);
+
+        $config = [
+            'constraints' => ['fruit_id' => 1],
+            'where' => ['basket_id > 1']
+        ];
+        $sampler->loadConfig((object)$config);
+        $sampler->execute();
+
+        $this->assertCount(2, $sampler->getRows());
+    }
 }