优化单测

Gemini-D · Jul 15, 2024 · 9edc530 · 9edc530
1 parent 098718a
commit 9edc530
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/Importer.php b/src/Importer.php
@@ -43,7 +43,7 @@ public function import(array $config, string $sql, bool $once = true): Result
         $result = new Result(true);
         foreach ($sqls as $line) {
             try {
-                if (str_starts_with($line, '--') || $line == '') {
+                if (str_starts_with($line, '#') || str_starts_with($line, '--') || $line == '') {
                     continue;
                 }
 

diff --git a/tests/Cases/ImportTest.php b/tests/Cases/ImportTest.php
@@ -46,6 +46,9 @@ public function testLoadConfig()
         $import = new Importer($this->getContainer());
         $res = $import->import($config, file_get_contents(__DIR__ . '/../sql/init.sql'));
         $this->assertTrue($res->isSuccess);
+
+        $res = $import->import($config, file_get_contents(__DIR__ . '/../sql/init.sql'), false);
+        $this->assertTrue($res->isSuccess);
     }
 
     public function testLoadPath()
@@ -65,6 +68,9 @@ public function testLoadPath()
         $import = new Importer($this->getContainer());
         $res = $import->importPath($config, __DIR__ . '/../sql/init.sql');
         $this->assertTrue($res->isSuccess);
+
+        $res = $import->importPath($config, __DIR__ . '/../sql/init.sql', true);
+        $this->assertTrue($res->isSuccess);
     }
 
     public function testPrivilegeCreateUser()