From 5b99b471b2ba9a9c9da801e9a9d6e1801f02068c Mon Sep 17 00:00:00 2001
From: Kegan Dougal <kegan@matrix.org>
Date: Thu, 5 Feb 2015 15:12:36 +0000
Subject: [PATCH] Fix unit tests.

---
 synapse/api/auth.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/synapse/api/auth.py b/synapse/api/auth.py
index 86f963c190..4f116184c9 100644
--- a/synapse/api/auth.py
+++ b/synapse/api/auth.py
@@ -302,6 +302,12 @@ class Auth(object):
 
             # Check for application service tokens with a user_id override
             try:
+                if "user_id" not in request.args:
+                    # This has to be done like this rather than relying on it
+                    # natively throwing because tests use a Mock for the request
+                    # object which doesn't throw :/
+                    raise KeyError
+
                 masquerade_user_id = request.args["user_id"][0]
                 app_service = yield self.store.get_app_service_by_token(
                     access_token