def testLoadSecretsFromFile (self):
self.__create_test_secrets_path()
generatedSecrets = generate_secrets_to_file(self.tmpFile)
secrets = load_secrets_from_file(self.tmpFile)
self.assertEquals(
secrets,
generatedSecrets,
"Loaded secrets are same as generated ones"
)
self.__remove_test_secrets_path()
def testValidatetokenFromGeneratedSecrets (self):
self.__create_test_secrets_path()
generatedSecrets = generate_secrets_to_file(self.tmpFile)
sigmund = Sigmund()
sigmund.secret = load_secrets_from_file(self.tmpFile);
testData = {"hello": "world"}
token = sigmund.generate(testData)
self.assertTrue(
sigmund.validate(token, testData),
"Can validate a token with secrets loaded from file"
)
self.__remove_test_secrets_path()
def testGenerateSecretsToFile (self):
self.__create_test_secrets_path()
secrets = generate_secrets_to_file(self.tmpFile)
self.assertTrue(
os.stat(self.tmpFile),
"Secrets file has been created"
)
generatedFile = open(self.tmpFile, 'r')
self.assertEquals(
generatedFile.read(),
",".join(secrets),
"Secrets have been written to the file"
)
self.__remove_test_secrets_path()
def testGenerateSecretsToFileNmberOfSecretsParameter (self):
self.__create_test_secrets_path()
secrets = generate_secrets_to_file(self.tmpFile, 23)
self.assertEquals(
23,
len(secrets),
"Expected number of secrets returned"
)
generatedFile = open(self.tmpFile, 'r')
self.assertEquals(
22,
generatedFile.read().count(","),
"Expected numbr of secrets have been written to the file"
)
self.__remove_test_secrets_path()
