InsertUpdateStatement::setMask has not worked. Added corresponding test

Author: rkrx

src/Builder/InsertUpdateStatement.php

@@ -35,6 +35,9 @@ protected function buildFieldList(array $fields, array $query = array()) {
 			if($fieldValue instanceof DefaultValue) {
 				$fieldValue = 'DEFAULT';
 			}
+			if(is_array($this->mask) && !in_array($fieldName, $this->mask)) {
+				continue;
+			}
 			if (is_int($fieldName)) {
 				$query[] = "\t{$fieldValue}";
 			} else {

tests/Builder/InsertTest.php

@@ -129,4 +129,14 @@ public function testAddOrUpdateAll() {
 		->asString();
 		$this->assertEquals("INSERT INTO\n\ttravis_test.test1\nSET\n\t`field1`='123'\nON DUPLICATE KEY UPDATE\n\t`field1`='123'\n;\n", $query);
 	}
+
+	public function testMask() {
+		$sql = TestInsert::create()
+		->into('test')
+		->addOrUpdate('field1', 1)
+		->addOrUpdate('field2', 2)
+		->setMask(['field1'])
+		->asString();
+		$this->assertEquals("INSERT INTO\n\ttest\nSET\n\t`field1`='1'\nON DUPLICATE KEY UPDATE\n\t`field1`='1'\n;\n", $sql);
+	}
 }

tests/Builder/UpdateTest.php

@@ -111,5 +111,16 @@ public function testLimit() {
 		->asString();
 		$this->assertEquals("UPDATE\n\ttest1\nSET\n\t`field1`='1'\nLIMIT\n\t10\nOFFSET\n\t10\n;\n", $sql);
 	}
+
+	public function testMask() {
+		$sql = TestUpdate::create()
+		->table('test1')
+		->set('field1', 1)
+		->set('field2', 2)
+		->setMask(['field1'])
+		->limit(10)
+		->asString();
+		$this->assertEquals("UPDATE\n\ttest1\nSET\n\t`field1`='1'\nLIMIT\n\t10\n;\n", $sql);
+	}
 }