Merge pull request #106 from DonCallisto/patch-2

Update PageObjectArgumentResolver.php
sensiolabs · Feb 20, 2018 · 3e695fb · 3e695fb
2 parents d259c8b + e2ae6a8
commit 3e695fb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/Context/Argument/PageObjectArgumentResolver.php b/src/Context/Argument/PageObjectArgumentResolver.php
@@ -3,7 +3,9 @@
 namespace SensioLabs\Behat\PageObjectExtension\Context\Argument;
 
 use Behat\Behat\Context\Argument\ArgumentResolver;
+use SensioLabs\Behat\PageObjectExtension\PageObject\Element;
 use SensioLabs\Behat\PageObjectExtension\PageObject\Factory;
+use SensioLabs\Behat\PageObjectExtension\PageObject\Page;
 
 class PageObjectArgumentResolver implements ArgumentResolver
 {
@@ -55,7 +57,7 @@ private function isPageOrElement($className)
      */
     private function isPage($className)
     {
-        return is_subclass_of($className, 'SensioLabs\Behat\PageObjectExtension\PageObject\Page');
+        return is_subclass_of($className, Page::class);
     }
 
     /**
@@ -65,7 +67,7 @@ private function isPage($className)
      */
     private function isElement($className)
     {
-        return is_subclass_of($className, 'SensioLabs\Behat\PageObjectExtension\PageObject\Element');
+        return is_subclass_of($className, Element::class);
     }
 
     /**