1 files changed, 10 insertions, 1 deletions
diff --git a/tests/unit/upgrade/UpgradeTest.php b/tests/unit/upgrade/UpgradeTest.php
index 936c329ca..d33cf9aff 100644
--- a/tests/unit/upgrade/UpgradeTest.php
+++ b/tests/unit/upgrade/UpgradeTest.php
@@ -22,18 +22,27 @@ class UpgradeTest extends \PHPUnit_Framework_TestCase {
     /** @var  IConfig */
     private $config;
 
+    /** @var  IDBConnection */
+    private $db;
+
     public function setUp() {
         $this->config = $this->getMockBuilder(
             '\OCP\IConfig')
             ->disableOriginalConstructor()
             ->getMock();
 
+        $this->db = $this->getMockBuilder(
+            '\OCP\IDBConnection')
+            ->disableOriginalConstructor()
+            ->getMock();
+
         $this->service = $this->getMockBuilder(
             '\OCA\News\Service\ItemService')
             ->disableOriginalConstructor()
             ->getMock();
 
-        $this->upgrade = new Upgrade($this->config, $this->service, 'news');
+        $this->upgrade = new Upgrade($this->config, $this->service,
+            $this->db, 'news');
     }
 
     public function testUpgrade() {