class TestVerify(unittest.TestCase):
"""
Class for testing the Generate class
"""
def setUp(self):
"""
Setup
"""
self.generate = Generate(PSK)
self.verify = Verify(PSK)
def test_verify(self):
"""
Generate a test
"""
assert self.verify.verify(self.generate.generate())
def test_verify_where_shared_secret_is_wrong(self):
"""
Set Generation to be a different key
"""
self.generate = Generate("notthekey")
assert not self.verify.verify(self.generate.generate())
def test_verify_where_time_is_wrong(self):
thetime = "201511031002"
thehash = hashlib.sha256(thetime + PSK)
assert not self.verify.verify(thehash.hexdigest())
def perform_action(self, data):
"""
Perform the verification
:param data - the data handed to the perform action command
:return a hash based on the client key, or Statics.ERROR message
"""
return_value = Statics.ERROR
verify = Verify(self.server_key)
print self.server_key
if verify.verify(data):
generate = Generate(self.client_key)
return_value = generate.generate()
return return_value
def setUp(self):
"""
Setup
"""
self.generate = Generate(PSK)
self.verify = Verify(PSK)
