From 4a501dab85ab579d5174e7630a8850a74b11af87 Mon Sep 17 00:00:00 2001
From: Isaac Murchie <isaac@saucelabs.com>
Date: Mon, 28 Sep 2015 08:35:44 -0700
Subject: [PATCH] Add command to get driver for a session

---
 lib/driver.js        | 6 ++++++
 test/driver-tests.js | 5 +++++
 2 files changed, 11 insertions(+)

diff --git a/lib/driver.js b/lib/driver.js
index 40df27d59..5ab7318b3 100644
--- a/lib/driver.js
+++ b/lib/driver.js
@@ -79,6 +79,12 @@ class BaseDriver extends MobileJsonWireProtocol {
     return sessionId === this.sessionId;
   }
 
+  // method required by MJSONWP in order to determine if the command should
+  // be proxied directly to the driver
+  driverForSession (/*sessionId*/) {
+    return this;
+  }
+
   logExtraCaps (caps) {
     let extraCaps = _.difference(_.keys(caps),
                                  _.keys(this._constraints));
diff --git a/test/driver-tests.js b/test/driver-tests.js
index 77878ea32..701d816f1 100644
--- a/test/driver-tests.js
+++ b/test/driver-tests.js
@@ -103,6 +103,11 @@ function baseDriverUnitTests (DriverClass, defaultCaps = {}) {
       await d.deleteSession();
     });
 
+    it('should have a method to get driver for a session', async () => {
+      let [sessId] = await d.createSession(defaultCaps);
+      d.driverForSession(sessId).should.eql(d);
+    });
+
     describe('command queue', () => {
       let d = new DriverClass();