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)