Fix #99 - use response's capabilities (if returned)

Author: robocoder

lib/WebDriver/Session.php

@@ -472,6 +472,16 @@ public function execute()
         return new Execute($this->url . '/execute');
     }
 
+    /**
+     * Set capabilities
+     *
+     * @param array $capabilities
+     */
+    public function setCapabilities($capabilities)
+    {
+        $this->capabilities = $capabilities;
+    }
+
     /**
      * {@inheritdoc}
      */

lib/WebDriver/WebDriver.php

@@ -31,6 +31,11 @@
  */
 class WebDriver extends AbstractWebDriver implements WebDriverInterface
 {
+    /**
+     * @var boolean
+     */
+    private $w3c;
+
     /**
      * {@inheritdoc}
      */
@@ -70,7 +75,13 @@ public function session($requiredCapabilities = Browser::FIREFOX, $desiredCapabi
             array(CURLOPT_FOLLOWLOCATION => true)
         );
 
-        return new Session($result['sessionUrl']);
+        $capabilities = isset($result['value']['capabilities']) ? $result['value']['capabilities'] : null;
+        $this->w3c    = !! $capabilities;
+
+        $session = new Session($result['sessionUrl']);
+        $session->setCapabilities($capabilities);
+
+        return $session;
     }
 
     /**
@@ -87,4 +98,14 @@ public function sessions()
 
         return $sessions;
     }
+
+    /**
+     * Is this a W3C driver?
+     *
+     * @return boolean
+     */
+    public function isW3c()
+    {
+        return $this->w3c;
+    }
 }