Refactor tests to provide 2 helpers

Author: rayluo

tests/test_token_cache.py

@@ -12,30 +12,57 @@
 
 class TokenCacheTestCase(unittest.TestCase):
 
+    @staticmethod
+    def build_id_token(sub="sub", oid="oid", preferred_username="me", **kwargs):
+        return "header.%s.signature" % base64.b64encode(json.dumps(dict({
+            "sub": sub,
+            "oid": oid,
+            "preferred_username": preferred_username,
+            }, **kwargs)).encode()).decode('utf-8')
+
+    @staticmethod
+    def build_response(  # simulate a response from AAD
+            uid="uid", utid="utid",  # They will form client_info
+            access_token=None, expires_in=3600, token_type="some type",
+            refresh_token=None,
+            foci=None,
+            id_token=None,  # or something generated by build_id_token()
+            error=None,
+            ):
+        response = {
+            "client_info": base64.b64encode(json.dumps({
+                "uid": uid, "utid": utid,
+                }).encode()).decode('utf-8'),
+            }
+        if error:
+            response["error"] = error
+        if access_token:
+            response.update({
+                "access_token": access_token,
+                "expires_in": expires_in,
+                "token_type": token_type,
+                })
+        if refresh_token:
+            response["refresh_token"] = refresh_token
+        if id_token:
+            response["id_token"] = id_token
+        if foci:
+            response["foci"] = foci
+        return response
+
     def setUp(self):
         self.cache = TokenCache()
 
     def testAdd(self):
-        client_info = base64.b64encode(b'''
-            {"uid": "uid", "utid": "utid"}
-            ''').decode('utf-8')
-        id_token = "header.%s.signature" % base64.b64encode(b'''{
-            "sub": "subject",
-            "oid": "object1234",
-            "preferred_username": "John Doe"
-            }''').decode('utf-8')
+        id_token = self.build_id_token(oid="object1234", preferred_username="John Doe")
         self.cache.add({
             "client_id": "my_client_id",
             "scope": ["s2", "s1", "s3"],  # Not in particular order
             "token_endpoint": "https://login.example.com/contoso/v2/token",
-            "response": {
-		"access_token": "an access token",
-		"token_type": "some type",
-		"expires_in": 3600,
-		"refresh_token": "a refresh token",
-                "client_info": client_info,
-                "id_token": id_token,
-		},
+            "response": self.build_response(
+                uid="uid", utid="utid",  # client_info
+                expires_in=3600, access_token="an access token",
+                id_token=id_token, refresh_token="a refresh token"),
             }, now=1000)
         self.assertEqual(
             {