Tear down client's connection states when dropping test clients

Max Brunsfeld created 3 years ago

Change summary

crates/client/src/client.rs | 2 ++
1 file changed, 2 insertions(+)

Detailed changes

crates/client/src/client.rs

@@ -230,10 +230,12 @@ impl Client {
     #[cfg(any(test, feature = "test-support"))]
     pub fn tear_down(&self) {
         let mut state = self.state.write();
+        state._maintain_connection.take();
         state.message_handlers.clear();
         state.models_by_message_type.clear();
         state.models_by_entity_type_and_remote_id.clear();
         state.entity_id_extractors.clear();
+        self.peer.reset();
     }
 
     #[cfg(any(test, feature = "test-support"))]