aquatic_ws: keep valid_until up to date in established connections

TODO.md

@@ -33,7 +33,6 @@
   * tungstenite 0.11
 * is 'key' sent in announce request? if so, maybe handle it like in
   aquatic_http (including ip uniqueness part of peer map key)
-* established connections do not get valid_until updated, I think?
 * tests
 * use enum as return type for handshake machine
 

aquatic_ws/src/lib/network/mod.rs

@@ -182,6 +182,11 @@ pub fn run_handshakes_and_read_messages(
 ){
     loop {
         if let Some(established_ws) = connections.get_mut(&poll_token)
+            .map(|c| { // Ugly but works
+                c.valid_until = valid_until;
+
+                c
+            })
             .and_then(Connection::get_established_ws)
         {
             use ::tungstenite::Error::Io;