Magic call Switchto and Test Magic SwitchTo

Author: LuisRBarreras

SeleniumClient/Options.php

@@ -45,20 +45,14 @@ public function timeouts()
 	 */
 	public function window()
 	{
-		if(!$this->_window)
-		{
-			$this->setWindow(new Window($this->_driver));
-		}
+        isset($this->_window) ? : $this->setWindow(new Window($this->_driver));
 		return $this->_window;
    	}
 
     /** Sets Window
      * @param $window
      */
-    public function setWindow($window)
-    {
-        $this->_window = $window;
-    }
+    public function setWindow($window) { $this->_window = $window; }
 
 	/**
 	 * Sets cookie

SeleniumClient/WebDriver.php

@@ -57,6 +57,8 @@ public function __construct(DesiredCapabilities $desiredCapabilities = null, $ho
      * Example: navigationRefresh, navigationBack.
      * Enables window's methods be invoked
      * Example: windowMaximize, windowGetPosition.
+     * Enables TargetLocator's methods
+     * Examples: switchToWindow, switchToFrame
      * Enables findElement and findElements methods be invoked through method missing.
      * The methods should be invoked with the format 'findElementBy<strategy>'.
      * Arguments should match those required by findElement and findElements methods.
@@ -78,6 +80,11 @@ public function __call( $name, array $args )
           return $values;
         }
 
+        else if ( strpos($name, "switchTo") === 0 ) {
+            $values = $this->callSwitchTo($name, $args);
+            return $values;
+        }
+
         else if( strpos($name, 'findElement') === 0 ){
             return $this->callFindElement($name, $args);
         }
@@ -136,6 +143,26 @@ private function callWindowMethods($name, $args)
         return $values;
     }
 
+    /**
+     * Call Target Locator Methods
+     * @param $name
+     * @param array $args
+     * @return mixed
+     * @throws \Exception
+     */
+    private function callSwitchTo($name, array $args)
+    {
+        $method = lcfirst(substr($name, 8));
+        switch($method){
+            case 'window':
+            case 'frame':
+                $values = call_user_func( array($this->switchTo(), $method),$args[0]);
+                break;
+            default: throw new \Exception ('Invalid magic call:'. $name);
+        }
+        return $values;
+    }
+
     /**Call findElement and findElement methods
      * @param $name
      * @param array $args
@@ -238,19 +265,14 @@ public function getHubUrl() { return $this->_hubUrl; }
      */
     public function navigate()
     {
-        if(!$this->_navigate) {
-            $this->setNavigate(new Navigation($this));
-        }
+        isset($this->_navigate) ? : $this->setNavigate(new Navigation($this));
         return $this->_navigate;
     }
 
     /**Set Navigation
      * @param $navigate
      */
-    public function setNavigate($navigate)
-    {
-        $this->_navigate = $navigate;
-    }
+    public function setNavigate($navigate) { $this->_navigate = $navigate; }
 
 	/**
 	 * Get assigned session id
@@ -277,10 +299,6 @@ public function getScreenShotsDirectory() { return $this->_screenshotsDirectory;
 	 */
 	public function manage()
 	{
-
-		if(!$this->_options) {
-			$this->_options = new Options($this);
-		}
 		isset($this->_options) ? : $this->_options = new Options($this);
 		return $this->_options;
 	}
@@ -291,10 +309,18 @@ public function manage()
 	 */
 	public function switchTo()
 	{
-		isset($this->_targetLocator) ? : $this->_targetLocator = new TargetLocator($this);
+		isset($this->_targetLocator) ? : $this->setSwitchTo(new TargetLocator($this));
 		return $this->_targetLocator;
 	}
 
+    /**
+     * Set Target Locator
+     * @param $targetLocator
+     */
+    public function setSwitchTo($targetLocator) { $this->_targetLocator = $targetLocator; }
+
+
+
 	/**
 	 * Starts new Selenium session
 	 * @param DesiredCapabilities $desiredCapabilities

SeleniumClientTest/Tests/WebDriverTest.php

@@ -11,16 +11,21 @@ class WebDriverTest extends AbstractTest
 {
     private $_navigationMock = null;
     private $_windowMock = null;
+    private $_targetLocatorMock = null;
+
     private $_originalNavigate = null;
     private $_originalWindow = null;
+    private $_originalTargetLocator = null;
 
     public function setUp()
     {
         parent::setUp();
         $this->_navigationMock = $this->getMock('Navigation', array('refresh','to','back','forward'));
         $this->_windowMock = $this->getMock('Window', array('maximize','close' ,'setSize', 'getSize', 'setPosition', 'getPosition'));
+        $this->_targetLocatorMock = $this->getMock('TargetLocator', array('window','frame'));
         $this->_originalNavigate = $this->_driver->navigate();
         $this->_originalWindow = $this->_driver->manage()->window();
+        $this->_originalTargetLocator = $this->_driver->switchTo();
 
     }
 
@@ -398,4 +403,32 @@ public function testMagicWindowSetPositionShouldCallMethodSetPosition()
         $this->_driver->manage()->setWindow($this->_originalWindow);
 
     }
+
+    public function testMagicSwitchToWindowShouldCallMethodWindow()
+    {
+        $this->_targetLocatorMock->expects($this->exactly(1))
+            ->method('window');
+
+        $this->_driver->setSwitchTo($this->_targetLocatorMock);
+
+        $this->_driver->switchToWindow('popup1');
+
+        $this->_driver->setSwitchTo($this->_originalTargetLocator);
+
+
+    }
+
+    public function testMagicSwitchToFrameShouldCallMethodFrame()
+    {
+        $this->_targetLocatorMock->expects($this->exactly(1))
+            ->method('frame');
+
+        $this->_driver->setSwitchTo($this->_targetLocatorMock);
+
+        $this->_driver->switchToFrame(null);
+
+        $this->_driver->setSwitchTo($this->_originalTargetLocator);
+
+    }
+
 }