Fixes tests to now look for to_jwt (#12)

* Fixes tests to now look for to_jwt * Python 3 tweak
TwilioDevEd · Feb 24, 2019 · 607520d · 607520d
1 parent bb7d08e
commit 607520d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/browser_calls_flask/views.py b/browser_calls_flask/views.py
@@ -53,7 +53,7 @@ def get_token():
         capability.allow_client_incoming('customer')
 
     # Generate the capability token
-    token = capability.to_jwt().decode('utf-8')
+    token = capability.to_jwt()
 
     return jsonify({'token': token})
 

diff --git a/tests/views_tests.py b/tests/views_tests.py
@@ -57,7 +57,7 @@ class TwilioTokenTests(BaseTest):
     def test_get_token_for_customer_by_default(self):
         # Given
         mock_capability = MagicMock()
-        mock_capability.generate.return_value = 'abc123'
+        mock_capability.to_jwt.return_value = 'abc123'
 
         # When
         with patch('browser_calls_flask.views.ClientCapabilityToken', return_value=mock_capability) as mock:
@@ -68,13 +68,13 @@ def test_get_token_for_customer_by_default(self):
         # arguments and that the view returned the correct response
         self.assertTrue(mock_capability.allow_client_outgoing.called)
         mock_capability.allow_client_incoming.assert_called_once_with('customer')
-        self.assertTrue(mock_capability.generate.called)
+        self.assertTrue(mock_capability.to_jwt.called)
         self.assertEqual({"token": "abc123"}, json.loads(response.data.decode("utf-8")))
 
     def test_get_token_for_agent_if_referrer_is_dashboard(self):
         # Given
         mock_capability = MagicMock()
-        mock_capability.generate.return_value = 'abc123'
+        mock_capability.to_jwt.return_value = 'abc123'
 
         # When
         with patch('browser_calls_flask.views.ClientCapabilityToken', return_value=mock_capability) as mock:
@@ -83,7 +83,7 @@ def test_get_token_for_agent_if_referrer_is_dashboard(self):
         # Then
         self.assertTrue(mock_capability.allow_client_outgoing.called)
         mock_capability.allow_client_incoming.assert_called_once_with('support_agent')
-        self.assertTrue(mock_capability.generate.called)
+        self.assertTrue(mock_capability.to_jwt.called)
         self.assertEqual({"token": "abc123"}, json.loads(response.data.decode("utf-8")))