def test_add_keys(self):
keys = LicenseKeys()
self.assertFalse(keys.has_key("KEY1"))
keys.add_key("KEY1", "VALUE1")
self.assertTrue(keys.has_key("KEY1"))
self.assertEqual("VALUE1", keys.get_key("KEY1"))
keys.add_key("KEY1", "VALUE2")
self.assertTrue(keys.has_key("KEY1"))
self.assertEqual("VALUE2", keys.get_key("KEY1"))
def test_add_keys(self):
keys = LicenseKeys()
self.assertFalse(keys.has_key("KEY1"))
keys.add_key("KEY1", "VALUE1")
self.assertTrue(keys.has_key("KEY1"))
self.assertEqual("VALUE1", keys.get_key("KEY1"))
keys.add_key("KEY1", "VALUE2")
self.assertTrue(keys.has_key("KEY1"))
self.assertEqual("VALUE2", keys.get_key("KEY1"))
def test_load_license_keys_file(self):
keys = LicenseKeys()
keys.load_license_key_file(os.path.dirname(__file__)+ os.sep + "test.keys")
self.assertTrue(keys.has_key('KEY1'))
self.assertEquals("Key1Data", keys.get_key("KEY1"))
self.assertTrue(keys.has_key('KEY2'))
self.assertEquals("Key2Data", keys.get_key("KEY2"))
self.assertTrue(keys.has_key('KEY3'))
self.assertEquals("KEY3 Data With Spaces", keys.get_key("KEY3"))
with self.assertRaises(Exception):
keys.get_key("KEY5")
def test_load_license_keys_file(self):
keys = LicenseKeys()
keys.load_license_key_file(os.path.dirname(__file__) + "/test.keys")
self.assertTrue(keys.has_key('KEY1'))
self.assertEquals("Key1Data", keys.get_key("KEY1"))
self.assertTrue(keys.has_key('KEY2'))
self.assertEquals("Key2Data", keys.get_key("KEY2"))
self.assertTrue(keys.has_key('KEY3'))
self.assertEquals("KEY3 Data With Spaces", keys.get_key("KEY3"))
with self.assertRaises(Exception):
keys.get_key("KEY5")
def test_load_license_keys_file(self):
keys = LicenseKeys()
keys.add_key("KEY1", "Key1Data")
keys.add_key("KEY2", "Key2Data")
keys.add_key("KEY3", "KEY3 Data With Spaces")
self.assertTrue(keys.has_key('KEY1'))
self.assertEqual("Key1Data", keys.get_key("KEY1"))
self.assertTrue(keys.has_key('KEY2'))
self.assertEqual("Key2Data", keys.get_key("KEY2"))
self.assertTrue(keys.has_key('KEY3'))
self.assertEqual("KEY3 Data With Spaces", keys.get_key("KEY3"))
with self.assertRaises(Exception):
keys.get_key("KEY5")
def assert_upload_license_keys_from_file(self, store):
store.empty()
store.upload_from_file(
os.path.dirname(__file__) + os.sep + "data" + os.sep + "licenses" +
os.sep + "test_license.keys")
license_keys = LicenseKeys()
store.load(license_keys)
self.assertTrue(license_keys.has_key("TESTKEY1"))
self.assertEqual("VALUE1", license_keys.get_key("TESTKEY1"))
self.assertTrue(license_keys.has_key("TESTKEY2"))
self.assertEqual("VERY LONG VALUE 2", license_keys.get_key("TESTKEY2"))
def test_load_license_key(self):
config = FileStorageConfiguration()
config._license_storage = FileStoreConfiguration(file=os.path.dirname(__file__) + os.sep + "data" + os.sep + "licenses" + os.sep + "test_license.keys", format="text", encoding="utf-8", delete_on_start=False)
engine = FileStorageEngine(config)
engine.initialise()
store = FileLicenseStore(engine)
store.empty()
license_keys = LicenseKeys()
store.load(license_keys)
self.assertTrue(license_keys.has_key("TESTKEY1"))
self.assertEqual("VALUE1", license_keys.get_key("TESTKEY1"))
self.assertTrue(license_keys.has_key("TESTKEY2"))
self.assertEqual("VERY LONG VALUE 2", license_keys.get_key("TESTKEY2"))
def test_load_license_keys_data(self):
keys = LicenseKeys()
keys.load_license_key_data("""
KEY1=Key1Data
KEY2=Key2Data
# This is a comment
KEY3=KEY3 Data With Spaces
This is not a key
""")
self.assertTrue(keys.has_key('KEY1'))
self.assertEquals("Key1Data", keys.get_key("KEY1"))
self.assertTrue(keys.has_key('KEY2'))
self.assertEquals("Key2Data", keys.get_key("KEY2"))
self.assertTrue(keys.has_key('KEY3'))
self.assertEquals("KEY3 Data With Spaces", keys.get_key("KEY3"))
with self.assertRaises(Exception):
keys.get_key("KEY5")
def test_load_license_keys_data(self):
keys = LicenseKeys()
keys.load_license_key_data("""
KEY1=Key1Data
KEY2=Key2Data
# This is a comment
KEY3=KEY3 Data With Spaces
This is not a key
""")
self.assertTrue(keys.has_key('KEY1'))
self.assertEquals("Key1Data", keys.get_key("KEY1"))
self.assertTrue(keys.has_key('KEY2'))
self.assertEquals("Key2Data", keys.get_key("KEY2"))
self.assertTrue(keys.has_key('KEY3'))
self.assertEquals("KEY3 Data With Spaces", keys.get_key("KEY3"))
with self.assertRaises(Exception):
keys.get_key("KEY5")