Started to create a small FLASK restful api for handling requests

v1/flask/lib/python3.4/site-packages/wheel/test/test_keys.py

@@ -0,0 +1,98 @@
+import tempfile
+import os.path
+import unittest
+import json
+
+from wheel.signatures import keys
+
+wheel_json = """
+{
+  "verifiers": [
+    {
+      "scope": "+", 
+      "vk": "bp-bjK2fFgtA-8DhKKAAPm9-eAZcX_u03oBv2RlKOBc"
+    }, 
+    {
+      "scope": "+", 
+      "vk": "KAHZBfyqFW3OcFDbLSG4nPCjXxUPy72phP9I4Rn9MAo"
+    },
+    {
+      "scope": "+", 
+      "vk": "tmAYCrSfj8gtJ10v3VkvW7jOndKmQIYE12hgnFu3cvk"
+    } 
+  ], 
+  "signers": [
+    {
+      "scope": "+", 
+      "vk": "tmAYCrSfj8gtJ10v3VkvW7jOndKmQIYE12hgnFu3cvk"
+    }, 
+    {
+      "scope": "+", 
+      "vk": "KAHZBfyqFW3OcFDbLSG4nPCjXxUPy72phP9I4Rn9MAo"
+    }
+  ], 
+  "schema": 1
+}
+"""
+
+class TestWheelKeys(unittest.TestCase):
+    def setUp(self):
+        self.config = tempfile.NamedTemporaryFile(suffix='.json')
+        self.config.close()
+        
+        self.config_path, self.config_filename = os.path.split(self.config.name) 
+        def load(*args):
+            return [self.config_path]
+        def save(*args):
+            return self.config_path
+        keys.load_config_paths = load
+        keys.save_config_path = save
+        self.wk = keys.WheelKeys()
+        self.wk.CONFIG_NAME = self.config_filename
+        
+    def tearDown(self):
+        os.unlink(self.config.name)
+        
+    def test_load_save(self):
+        self.wk.data = json.loads(wheel_json)
+        
+        self.wk.add_signer('+', '67890')
+        self.wk.add_signer('scope', 'abcdefg')
+        
+        self.wk.trust('epocs', 'gfedcba')
+        self.wk.trust('+', '12345')
+        
+        self.wk.save()
+        
+        del self.wk.data
+        self.wk.load()
+        
+        signers = self.wk.signers('scope')
+        self.assertTrue(signers[0] == ('scope', 'abcdefg'), self.wk.data['signers'])
+        self.assertTrue(signers[1][0] == '+', self.wk.data['signers'])
+        
+        trusted = self.wk.trusted('epocs')
+        self.assertTrue(trusted[0] == ('epocs', 'gfedcba'))
+        self.assertTrue(trusted[1][0] == '+')
+        
+        self.wk.untrust('epocs', 'gfedcba')
+        trusted = self.wk.trusted('epocs')
+        self.assertTrue(('epocs', 'gfedcba') not in trusted)
+        
+    def test_load_save_incomplete(self):
+        self.wk.data = json.loads(wheel_json)
+        del self.wk.data['signers']
+        self.wk.data['schema'] = self.wk.SCHEMA+1
+        self.wk.save()
+        try:
+            self.wk.load()
+        except ValueError:
+            pass
+        else:
+            raise Exception("Expected ValueError")
+        
+        del self.wk.data['schema']
+        self.wk.save()
+        self.wk.load()
+    
+