class TestVerificationCommand(unittest.TestCase): """ Test that the verification has happened """ def setUp(self): self.psk = PreSharedSecrets() self.generate_server = Generate(self.psk.serverKey) self.generate_client = Generate(self.psk.clientKey) self.verification_command = VerificationCommand(self.psk.serverKey, self.psk.clientKey) def test_verification_passes(self): assert self.verification_command.perform_action(self.generate_server.generate()) == self.generate_client.generate()
def setUp(self): self.psk = PreSharedSecrets() self.generate_server = Generate(self.psk.serverKey) self.generate_client = Generate(self.psk.clientKey) self.verification_command = VerificationCommand(self.psk.serverKey, self.psk.clientKey)