From d1345f4bb3bdfceee76de884756c32003c942ca7 Mon Sep 17 00:00:00 2001
From: recca0120 <recca0120@gmail.com>
Date: Wed, 21 Dec 2022 12:06:03 +0800
Subject: [PATCH] load schema to in-memory database

---
 .../Database/Schema/SqliteSchemaState.php      |  6 ++++++
 .../Database/DatabaseSqliteSchemaStateTest.php | 18 ++++++++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/src/Illuminate/Database/Schema/SqliteSchemaState.php b/src/Illuminate/Database/Schema/SqliteSchemaState.php
index 9a98b6331cba..10efc7c0aba9 100644
--- a/src/Illuminate/Database/Schema/SqliteSchemaState.php
+++ b/src/Illuminate/Database/Schema/SqliteSchemaState.php
@@ -61,6 +61,12 @@ protected function appendMigrationData(string $path)
      */
     public function load($path)
     {
+        if ($this->connection->getDatabaseName() === ':memory:') {
+            $this->connection->getPdo()->exec($this->files->get($path));
+
+            return;
+        }
+
         $process = $this->makeProcess($this->baseCommand().' < "${:LARAVEL_LOAD_PATH}"');
 
         $process->mustRun(null, array_merge($this->baseVariables($this->connection->getConfig()), [
diff --git a/tests/Database/DatabaseSqliteSchemaStateTest.php b/tests/Database/DatabaseSqliteSchemaStateTest.php
index 404f3bb9023e..2ff551eee714 100644
--- a/tests/Database/DatabaseSqliteSchemaStateTest.php
+++ b/tests/Database/DatabaseSqliteSchemaStateTest.php
@@ -4,7 +4,9 @@
 
 use Illuminate\Database\Schema\SqliteSchemaState;
 use Illuminate\Database\SQLiteConnection;
+use Illuminate\Filesystem\Filesystem;
 use Mockery as m;
+use PDO;
 use PHPUnit\Framework\TestCase;
 use Symfony\Component\Process\Process;
 
@@ -36,4 +38,20 @@ public function testLoadSchemaToDatabase(): void
         ]);
     }
 
+    public function testLoadSchemaToInMemory(): void
+    {
+        $config = ['driver' => 'sqlite', 'database' => ':memory:', 'prefix' => '', 'foreign_key_constraints' => true, 'name' => 'sqlite'];
+        $connection = m::mock(SQLiteConnection::class);
+        $connection->shouldReceive('getConfig')->andReturn($config);
+        $connection->shouldReceive('getDatabaseName')->andReturn($config['database']);
+        $connection->shouldReceive('getPdo')->andReturn($pdo = m::spy(PDO::class));
+
+        $files = m::mock(Filesystem::class);
+        $files->shouldReceive('get')->andReturn('CREATE TABLE IF NOT EXISTS "migrations" ("id" integer not null primary key autoincrement, "migration" varchar not null, "batch" integer not null);');
+
+        $schemaState = new SqliteSchemaState($connection, $files);
+        $schemaState->load('database/schema/sqlite-schema.dump');
+
+        $pdo->shouldHaveReceived('exec')->with('CREATE TABLE IF NOT EXISTS "migrations" ("id" integer not null primary key autoincrement, "migration" varchar not null, "batch" integer not null);');
+    }
 }