Update UrlTesterExtension.php

Author: stepapo

src/DI/UrlTesterExtension.php

@@ -8,6 +8,7 @@
 use Nette\DI\CompilerExtension;
 use Nette\DI\Definitions\Statement;
 use Nette\Http\UrlScript;
+use Stepapo\UrlTester\Mock\User;
 use Stepapo\UrlTester\UrlTester;
 
 
@@ -33,6 +34,11 @@ public function loadConfiguration()
 				->setFactory(Session::class);
 		}
 
+		if ($builder->hasDefinition('security.user')) {
+			$builder->getDefinition('security.user')
+				->setFactory(User::class);
+		}
+
 		$builder->addDefinition($this->prefix('helper'))
 			->setFactory(Helper::class);