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