Tests: Add lazy-accessor for MongoDB

perftools · Dec 26, 2020 · feec5ad · feec5ad
1 parent 6c8c186
commit feec5ad
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/tests/LazyContainerProperties.php b/tests/LazyContainerProperties.php
@@ -3,6 +3,7 @@
 namespace XHGui\Test;
 
 use LazyProperty\LazyPropertiesTrait;
+use MongoDB;
 use Slim\Slim as App;
 use Slim\View;
 use XHGui\Controller\ImportController;
@@ -24,6 +25,8 @@ trait LazyContainerProperties
     protected $import;
     /** @var MongoSearcher */
     private $mongo;
+    /** @var MongoDB */
+    protected $mongodb;
     /** @var RunController */
     protected $runs;
     /** @var App */
@@ -44,6 +47,7 @@ protected function setupProperties(): void
             'app',
             'import',
             'mongo',
+            'mongodb',
             'runs',
             'saver',
             'searcher',
@@ -90,6 +94,11 @@ protected function getMongo()
         return $this->di['searcher.mongodb'];
     }
 
+    protected function getMongoDb()
+    {
+        return $this->di[MongoDB::class];
+    }
+
     protected function getSearcher()
     {
         return $this->di['searcher'];

diff --git a/tests/Searcher/MongoTest.php b/tests/Searcher/MongoTest.php
@@ -188,7 +188,7 @@ public function testSaveRemove(): void
 
     public function testTruncateResultsPreserveIndexes(): void
     {
-        $mongoDb = $this->getDi()[MongoDB::class];
+        $mongoDb = $this->mongodb;
         $collection = $mongoDb->results;
 
         // dropping "results" collection using raw client
@@ -252,7 +252,7 @@ public function testTruncateResultsPreserveIndexes(): void
 
     public function testTruncateWatchesPreserveIndexes(): void
     {
-        $mongoDb = $this->getDi()[MongoDB::class];
+        $mongoDb = $this->mongodb;
         $collection = $mongoDb->watches;
 
         // dropping "watches" collection using raw client