1 files changed, 111 insertions, 0 deletions
diff --git a/tests/Builder/LoadStatementTest.php b/tests/Builder/LoadStatementTest.php
new file mode 100644
index 0000000..2ef6236
--- /dev/null
+++ b/tests/Builder/LoadStatementTest.php
@@ -0,0 +1,111 @@
+<?php
+
+namespace PhpMyAdmin\SqlParser\Tests\Builder;
+
+use PhpMyAdmin\SqlParser\Parser;
+use PhpMyAdmin\SqlParser\Tests\TestCase;
+
+class LoadStatementTest extends TestCase
+{
+    public function testBuilder()
+    {
+        /* Assertion 1 */
+        $query = 'LOAD DATA CONCURRENT INFILE '
+            . '\'employee1.txt\' INTO TABLE employee';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'LOAD DATA CONCURRENT INFILE '
+            . '\'employee1.txt\' INTO TABLE employee',
+            $stmt->build()
+        );
+
+        /* Assertion 2 */
+        $query = 'LOAD DATA INFILE \'/tmp/test.txt\' '
+            . 'INTO TABLE test FIELDS TERMINATED BY '
+            . '\',\' IGNORE 1 LINES';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'LOAD DATA  INFILE \'/tmp/test.txt\' '
+            . 'INTO TABLE test FIELDS TERMINATED BY '
+            . '\',\' IGNORE 1 LINES',
+            $stmt->build()
+        );
+
+
+        /* Assertion 3 */
+        $query = 'LOAD DATA INFILE \'employee3.txt\' '
+            . 'INTO TABLE employee FIELDS TERMINATED BY '
+            . '\',\' ENCLOSED BY \'"\'';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'LOAD DATA  INFILE \'employee3.txt\' '
+            . 'INTO TABLE employee FIELDS TERMINATED BY '
+            . '\',\' ENCLOSED BY \'"\'',
+            $stmt->build()
+        );
+
+
+        /* Assertion 4 */
+        $query = 'LOAD DATA INFILE \'/tmp/test.txt\' IGNORE '
+            . 'INTO TABLE test '
+            . 'CHARACTER SET \'utf8\' '
+            . 'COLUMNS TERMINATED BY \',\' '
+            . 'LINES TERMINATED BY \';\' '
+            . 'IGNORE 1 LINES (col1, col2) SET @a = 1';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'LOAD DATA  INFILE \'/tmp/test.txt\' IGNORE '
+            . 'INTO TABLE test '
+            . 'CHARACTER SET \'utf8\' '
+            . 'COLUMNS TERMINATED BY \',\' '
+            . 'LINES TERMINATED BY \';\' '
+            . 'IGNORE 1 LINES (col1, col2) SET @a = 1',
+            $stmt->build()
+        );
+
+
+        /* Assertion 5 */
+        $query = 'LOAD DATA INFILE \'/tmp/test.txt\' REPLACE '
+            . 'INTO TABLE test COLUMNS TERMINATED BY \',\' IGNORE 1 ROWS';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'LOAD DATA  INFILE \'/tmp/test.txt\' REPLACE '
+            . 'INTO TABLE test COLUMNS TERMINATED BY \',\' IGNORE 1 ROWS',
+            $stmt->build()
+        );
+
+
+        /* Assertion 6 */
+        $query = 'LOAD DATA INFILE \'/tmp/test.txt\' IGNORE '
+            . 'INTO TABLE test PARTITION (p0, p1, p2) CHARACTER SET \'utf8\' '
+            . 'COLUMNS TERMINATED BY \',\' LINES TERMINATED BY \';\' '
+            . 'IGNORE 1 LINES (col1, col2) SET @a = 1';
+
+        $parser = new Parser($query);
+        $stmt = $parser->statements[0];
+
+        $this->assertEquals(
+            'LOAD DATA  INFILE \'/tmp/test.txt\' IGNORE '
+            . 'INTO TABLE test PARTITION (p0, p1, p2) CHARACTER SET \'utf8\' '
+            . 'COLUMNS TERMINATED BY \',\' LINES TERMINATED BY \';\' '
+            . 'IGNORE 1 LINES (col1, col2) SET @a = 1',
+            $stmt->build()
+        );
+    }
+
+}