Fix type hints

Author: enumag

src/Module/Doctrine.php

@@ -12,13 +12,13 @@
 
 use Codeception\Exception\ModuleConfigException;
 use Codeception\Module;
-use Codeception\TestCase;
+use Codeception\TestInterface;
 use Doctrine\ORM\EntityManagerInterface;
 use PDOException;
 
 class Doctrine extends Module
 {
-    public function _before(TestCase $test)
+    public function _before(TestInterface $test)
     {
         if ($this->config['dump']) {
             $em = $this->getModule(Nette::class)->grabService(EntityManagerInterface::class);

src/Module/Nette.php

@@ -13,7 +13,7 @@
 use Arachne\Codeception\Connector\Nette as NetteConnector;
 use Arachne\Codeception\Module\Container as ContainerModule;
 use Codeception\Lib\Framework;
-use Codeception\TestCase;
+use Codeception\TestInterface;
 use Nette\DI\Container;
 use Nette\DI\MissingServiceException;
 
@@ -51,7 +51,7 @@ public function _beforeSuite($settings = [])
         $this->path = $settings['path'];
     }
 
-    public function _before(TestCase $test)
+    public function _before(TestInterface $test)
     {
         $this->configFiles = null;
         $this->container = null;
@@ -78,7 +78,7 @@ public function useConfigFiles(array $configFiles)
         $this->configFiles = $configFiles;
     }
 
-    public function _after(TestCase $test)
+    public function _after(TestInterface $test)
     {
         parent::_after($test);