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)
