add SSO url generation

Author: mikesea

fixtures/create_user.yaml

@@ -5,7 +5,7 @@
       - !!python/tuple [Content-Type, application/json]
     host: localhost
     method: POST
-    path: /api/v1/users.json?client_secret=a62962938bbe76194bbdc4e3cbfcb6708283b47614594610c71240f81c17af6f&auth_token=a-test-api-key&client_id=bd779af52ce7c26330a2c8a9216caf3f84906db6893788bdb4bd27549ca7bbbf
+    path: /api/v1/users.json?client_secret=4441879b5e2a9c3271f5b1a4bc223b715f091e5ed20fe75d1352e1290c7a6dfb&auth_token=a-test-api-key&client_id=3e4fd842fc99aecb4dc50e5b88a186c1e206ddd516cdd336da3622c4afd7e2e9
     port: 3000
     protocol: http
   response:

fixtures/user.yaml

@@ -5,7 +5,7 @@
       - !!python/tuple [Content-Type, application/json]
     host: localhost
     method: GET
-    path: /api/v1/users/12345.json?client_secret=a62962938bbe76194bbdc4e3cbfcb6708283b47614594610c71240f81c17af6f&auth_token=a-test-api-key&client_id=bd779af52ce7c26330a2c8a9216caf3f84906db6893788bdb4bd27549ca7bbbf
+    path: /api/v1/users/12345.json?client_secret=4441879b5e2a9c3271f5b1a4bc223b715f091e5ed20fe75d1352e1290c7a6dfb&auth_token=a-test-api-key&client_id=3e4fd842fc99aecb4dc50e5b88a186c1e206ddd516cdd336da3622c4afd7e2e9
     port: 3000
     protocol: http
   response:

fixtures/users.yaml

@@ -5,7 +5,7 @@
       - !!python/tuple [Content-Type, application/json]
     host: localhost
     method: GET
-    path: /api/v1/users.json?client_secret=a62962938bbe76194bbdc4e3cbfcb6708283b47614594610c71240f81c17af6f&auth_token=a-test-api-key&client_id=bd779af52ce7c26330a2c8a9216caf3f84906db6893788bdb4bd27549ca7bbbf
+    path: /api/v1/users.json?client_secret=4441879b5e2a9c3271f5b1a4bc223b715f091e5ed20fe75d1352e1290c7a6dfb&auth_token=a-test-api-key&client_id=3e4fd842fc99aecb4dc50e5b88a186c1e206ddd516cdd336da3622c4afd7e2e9
     port: 3000
     protocol: http
   response:

fixtures/users_pagination.yaml

@@ -5,7 +5,7 @@
       - !!python/tuple [Content-Type, application/json]
     host: localhost
     method: GET
-    path: /api/v1/users.json?client_secret=a62962938bbe76194bbdc4e3cbfcb6708283b47614594610c71240f81c17af6f&auth_token=a-test-api-key&page=2&client_id=bd779af52ce7c26330a2c8a9216caf3f84906db6893788bdb4bd27549ca7bbbf
+    path: /api/v1/users.json?client_secret=4441879b5e2a9c3271f5b1a4bc223b715f091e5ed20fe75d1352e1290c7a6dfb&auth_token=a-test-api-key&page=2&client_id=3e4fd842fc99aecb4dc50e5b88a186c1e206ddd516cdd336da3622c4afd7e2e9
     port: 3000
     protocol: http
   response:

swiftype/swiftype.py

@@ -3,7 +3,8 @@
 import urllib
 import urlparse
 import base64
-import datetime
+import time
+import hashlib
 
 from version import VERSION
 
@@ -143,6 +144,17 @@ def create_user(self):
     params = {'client_id': self.client_id, 'client_secret': self.client_secret}
     return self.conn._post(self.__users_path(), params)
 
+  def sso_url(self, user_id):
+    timestamp = self._get_timestamp()
+    params = {'user_id': user_id, 'client_id': self.client_id, 'timestamp': timestamp, 'token': self._sso_token(user_id, timestamp)}
+    return urlparse.urlunparse(('https', 'swiftype.com', '/sso', '', urllib.urlencode(params), ''))
+
+  def _sso_token(self, user_id, timestamp):
+    return hashlib.sha1('%s:%s:%s' % (user_id, self.client_secret, timestamp)).hexdigest()
+
+  def _get_timestamp(self):
+    return int(time.time())
+
   def __search_path(self, engine_id): return 'engines/%s/search' % (engine_id)
   def __suggest_path(self, engine_id): return 'engines/%s/suggest' % (engine_id)
   def __engines_path(self): return 'engines'

test_requirements.txt

@@ -1,2 +1,3 @@
 nose==1.3.0
-vcrpy==0.5.0
+vcrpy==0.5.0
+mock>=1.0.1

tests/test_swiftype.py

@@ -3,6 +3,7 @@
 import time
 import unittest
 import vcr
+from mock import Mock
 
 class TestClientFunctions(unittest.TestCase):
 
@@ -275,8 +276,8 @@ def setUp(self):
         except:
             api_key = "a-test-api-key"
 
-        client_id = 'bd779af52ce7c26330a2c8a9216caf3f84906db6893788bdb4bd27549ca7bbbf'
-        client_secret = 'a62962938bbe76194bbdc4e3cbfcb6708283b47614594610c71240f81c17af6f'
+        client_id = '3e4fd842fc99aecb4dc50e5b88a186c1e206ddd516cdd336da3622c4afd7e2e9'
+        client_secret = '4441879b5e2a9c3271f5b1a4bc223b715f091e5ed20fe75d1352e1290c7a6dfb'
         self.client = swiftype.Client(api_key=api_key, client_id=client_id, client_secret=client_secret, host='localhost:3000')
 
     def test_users(self):
@@ -302,6 +303,18 @@ def test_create_user(self):
             response = self.client.create_user()
             self.assertEqual(response['status'], 200)
 
+    def test_sso_token(self):
+        timestamp = 1379382520
+        user_id = '5064a7de2ed960e715000276'
+        token = self.client._sso_token(user_id, timestamp)
+        self.assertEqual(token, '81033d182ad51f231cc9cda9fb24f2298a411437')
+
+    def test_sso_url(self):
+        self.client._get_timestamp = Mock(return_value=1379382520)
+        user_id = '5064a7de2ed960e715000276'
+        url = self.client.sso_url(user_id)
+        self.assertEqual(url, 'https://swiftype.com/sso?timestamp=1379382520&token=81033d182ad51f231cc9cda9fb24f2298a411437&user_id=5064a7de2ed960e715000276&client_id=3e4fd842fc99aecb4dc50e5b88a186c1e206ddd516cdd336da3622c4afd7e2e9')
+
 class TestPlatformResources(unittest.TestCase):
 
     def setUp(self):