X-Git-Url: http://www.aleph1.co.uk/gitweb/?p=yaffs-website;a=blobdiff_plain;f=web%2Fmodules%2Fcontrib%2Fdrupalmoduleupgrader%2Ftests%2Fsrc%2FUnit%2FSQLiteDatabaseTrait.php;fp=web%2Fmodules%2Fcontrib%2Fdrupalmoduleupgrader%2Ftests%2Fsrc%2FUnit%2FSQLiteDatabaseTrait.php;h=4a9b692d7a4e432769691a9f424cb3f31295401f;hp=0000000000000000000000000000000000000000;hb=8acec36f19c470dfcda1ae2336826a782f41874c;hpb=e0411c4e83ba0d079034db83c3f7f55be24a0e35

diff --git a/web/modules/contrib/drupalmoduleupgrader/tests/src/Unit/SQLiteDatabaseTrait.php b/web/modules/contrib/drupalmoduleupgrader/tests/src/Unit/SQLiteDatabaseTrait.php
new file mode 100644
index 000000000..4a9b692d7
--- /dev/null
+++ b/web/modules/contrib/drupalmoduleupgrader/tests/src/Unit/SQLiteDatabaseTrait.php
@@ -0,0 +1,28 @@
+<?php
+
+namespace Drupal\Tests\drupalmoduleupgrader\Unit;
+
+use Drupal\Core\Database\Driver\sqlite\Connection;
+
+/**
+ * A trait for tests that need a database.
+ */
+trait SQLiteDatabaseTrait {
+
+  /**
+   * @var \Drupal\Core\Database\Connection
+   */
+  protected $db;
+
+  protected function initDB() {
+    if (empty($this->db)) {
+      // In-memory databases will cease to exist as soon as the connection
+      // is closed, which is...convenient as hell!
+      $db = new \PDO('sqlite::memory:');
+      // Throw exceptions when things go awry.
+      $db->setAttribute(\PDO::ATTR_ERRMODE, \PDO::ERRMODE_EXCEPTION);
+      $this->db = new Connection($db, []);
+    }
+  }
+
+}