diff --git a/castle/test/client_test.py b/castle/test/client_test.py index fd82089..7fdb922 100644 --- a/castle/test/client_test.py +++ b/castle/test/client_test.py @@ -60,7 +60,7 @@ def test_impersonate(self): status=200 ) client = Client.from_request(request(), {}) - options = {'impersonator': 'admin', 'user_id': '1234'} + options = {'properties': {'impersonator': 'admin'}, 'user_id': '1234'} self.assertEqual(client.impersonate(options), response_text) @responses.activate @@ -73,7 +73,7 @@ def test_impersonate_failed(self): status=200 ) client = Client.from_request(request(), {}) - options = {'impersonator': 'admin', 'user_id': '1234'} + options = {'properties': {'impersonator': 'admin'}, 'user_id': '1234'} with self.assertRaises(ImpersonationFailed): client.impersonate(options) diff --git a/castle/test/commands/impersonate_test.py b/castle/test/commands/impersonate_test.py index d112549..9647cb8 100644 --- a/castle/test/commands/impersonate_test.py +++ b/castle/test/commands/impersonate_test.py @@ -7,7 +7,7 @@ def default_options(): """Default options include all required fields.""" - return {'impersonator': 'admin', 'user_id': '1234', + return {'properties': {'impersonator': 'admin'}, 'user_id': '1234', 'context': {'ip': '127.0.0.1', 'user_agent': 'Chrome'}}