Ruoyi-Flask-admin/flasky/Lib/site-packages/wheel/test/test_keys.py

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()