add more tests for mysql, postgres

Author: hrach

tests/cases/MySqlMultiQueryParserTest.phpt

@@ -17,39 +17,27 @@ require_once __DIR__ . '/../bootstrap.php';
 class MySqlMultiQueryParserTest extends TestCase
 {
 	/**
-	 * @dataProvider provideLoadFileData
+	 * @dataProvider provideDelimitersData
 	 */
-	public function testLoadFile($content, array $expectedQueries)
+	public function testDelimiter($content, array $expectedQueries)
 	{
 		$parser = new MySqlMultiQueryParser();
-		$actualQueries = iterator_to_array($parser->parseFile(FileMock::create($content)));
-		Assert::same($expectedQueries, $actualQueries);
+		$queries = iterator_to_array($parser->parseFile(FileMock::create($content)));
+		Assert::same($expectedQueries, $queries);
 	}
 
 
-	protected function provideLoadFileData()
+	public function testFile()
+	{
+		$parser = new MySqlMultiQueryParser();
+		$queries = iterator_to_array($parser->parseFile(__DIR__ . '/data/mysql.sql'));
+		Assert::count(58, $queries);
+	}
+
+
+	protected function provideDelimitersData()
 	{
 		return [
-			[
-				'SELECT 1',
-				[
-					'SELECT 1',
-				],
-			],
-			[
-				'SELECT 1; ',
-				[
-					'SELECT 1',
-				],
-			],
-			[
-				'SELECT 1; SELECT 2;    SELECT 3; ',
-				[
-					'SELECT 1',
-					'SELECT 2',
-					'SELECT 3',
-				],
-			],
 			[
 				implode("\n", [
 					'SELECT 1;',

tests/cases/PgSqlMultiQueryParserTest.phpt

@@ -0,0 +1,37 @@
+<?php declare(strict_types = 1);
+
+/**
+ * @testCase
+ */
+
+namespace Nextras\MultiQueryParser;
+
+use Tester\Assert;
+use Tester\TestCase;
+
+
+require_once __DIR__ . '/../bootstrap.php';
+
+
+class PostgreSqlMultiQueryParserTest extends TestCase
+{
+	public function testFile()
+	{
+		$parser = new PostgreSqlMultiQueryParser();
+		$queries = iterator_to_array($parser->parseFile(__DIR__ . '/data/postgres.sql'));
+		Assert::count(66, $queries);
+		Assert::same(<<<SQL
+		CREATE FUNCTION "book_collections_before"() RETURNS TRIGGER AS
+		\$BODY$
+		BEGIN
+		    NEW."updated_at" = NOW();
+		    return NEW;
+		END;
+		\$BODY$
+		    LANGUAGE 'plpgsql' VOLATILE
+		SQL, $queries[16]);
+	}
+}
+
+
+(new PostgreSqlMultiQueryParserTest())->run();