def test_change_key_correctly_change_the_key(): # Given key_old = generate_key_32() key_new = generate_key_32() # When udp_socket = UDPSocket(key=key_old, socket_port=50015).start() udp_socket.change_key(key_new) # Then assert udp_socket.get_key() == key_new udp_socket.stop() time.sleep(.5)
def test_change_key_correctly_change_the_fernet_encoder(): # Given key_old = generate_key_32() key_new = generate_key_32() msg = b"tests" # When udp_socket = UDPSocket(key=key_old, socket_port=50016).start() msg_crypt = udp_socket._encrypt(msg) msg_decrypt_old = udp_socket._decrypt(msg_crypt) udp_socket.change_key(key_new) # Then assert msg_decrypt_old == msg udp_socket.stop() time.sleep(.5)