Fix repair step and test it

Signed-off-by: Sean Molenaar <sean@seanmolenaar.eu>

2 files changed, 6 insertions, 5 deletions

diff --git a/tests/Integration/IntegrationTest.php b/tests/Integration/IntegrationTest.php
index 276c6c394..1135bbd2e 100644
--- a/tests/Integration/IntegrationTest.php
+++ b/tests/Integration/IntegrationTest.php
@@ -30,6 +30,7 @@ use OCA\News\Tests\Integration\Fixtures\FolderFixture;
 use OCA\News\Db\FeedMapper;
 use OCA\News\Db\ItemMapper;
 use OCA\News\Db\FolderMapper;
+use Psr\Container\ContainerInterface;
 
 
 abstract class IntegrationTest extends \Test\TestCase
@@ -67,9 +68,9 @@ abstract class IntegrationTest extends \Test\TestCase
         $this->setupUser($this->user, $this->userPassword);
 
         // set up database layers
-        $this->itemMapper = $this->container->query(ItemMapper::class);
-        $this->feedMapper = $this->container->query(FeedMapper::class);
-        $this->folderMapper = $this->container->query(FolderMapper::class);
+        $this->itemMapper = $this->container->get(ItemMapper::class);
+        $this->feedMapper = $this->container->get(FeedMapper::class);
+        $this->folderMapper = $this->container->get(FolderMapper::class);
     }
 
     protected function findItemByTitle($title)
diff --git a/tests/Unit/Controller/JSONHttpErrorTest.php b/tests/Unit/Controller/JSONHttpErrorTest.php
index c02b60c9e..3b48bc36e 100644
--- a/tests/Unit/Controller/JSONHttpErrorTest.php
+++ b/tests/Unit/Controller/JSONHttpErrorTest.php
@@ -13,13 +13,13 @@
 
 namespace OCA\News\Tests\Unit\Controller;
 
-use OCA\News\Controller\JSONHttpError;
+use OCA\News\Controller\JSONHttpErrorTrait;
 
 use PHPUnit\Framework\TestCase;
 
 class Test
 {
-    use JSONHttpError;
+    use JSONHttpErrorTrait;
 }
 
 class JSONHttpErrorTest extends TestCase