Add tests for PDO exceptions rethrowing [MAILPOET-966]

tests/unit/Config/DatabaseTest.php

@@ -1,5 +1,6 @@
 <?php
 
+use Codeception\Util\Stub;
 use MailPoet\Config\Database;
 use MailPoet\Config\Env;
 
@@ -58,4 +59,26 @@ class DatabaseTestTest extends MailPoetTest {
     // time zone should be set based on WP's time zone
     expect($result->time_zone)->equals(Env::$db_timezone_offset);
   }
-}
+
+  function testItRethrowsPDOExceptions() {
+    $message = 'Error message';
+    $pdo = Stub::make(
+      'PDO',
+      array(
+        'prepare' => function() use ($message) {
+          throw new \PDOException($message);
+        }
+      )
+    );
+    \ORM::setDb($pdo);
+    try {
+      $this->database->setupDriverOptions();
+      $this->fail('Exception was not thrown');
+    } catch(\Exception $e) {
+      expect($e instanceof \PDOException)->false();
+      expect($e->getMessage())->equals($message);
+    }
+    // Remove the DB stub
+    $this->_before();
+  }
+}

tests/unit/Models/ModelTest.php

@@ -0,0 +1,29 @@
+<?php
+
+use Codeception\Util\Stub;
+use MailPoet\Models\Model;
+
+class ModelTest extends MailPoetTest {
+  function testItRethrowsPDOExceptions() {
+    $message = 'Error message';
+    $model = Stub::make('MailPoet\Models\Model');
+    $pdo = Stub::make(
+      'PDO',
+      array(
+        'prepare' => function() use ($message) {
+          throw new \PDOException($message);
+        }
+      )
+    );
+    \ORM::setDb($pdo);
+    try {
+      $model::findMany();
+      $this->fail('Exception was not thrown');
+    } catch(\Exception $e) {
+      expect($e instanceof \PDOException)->false();
+      expect($e->getMessage())->equals($message);
+    }
+    // Remove the DB stub
+    \ORM::setDb(null);
+  }
+}