From 595716afc55b204f31df41a15406d46722df0aee Mon Sep 17 00:00:00 2001
From: kozabrada123 <“kozabrada123@users.noreply.github.com”>
Date: Sat, 13 May 2023 15:36:29 +0200
Subject: [PATCH] Small unit test update

---
 src/gateway.rs | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/gateway.rs b/src/gateway.rs
index e0728e1..1996ff1 100644
--- a/src/gateway.rs
+++ b/src/gateway.rs
@@ -542,23 +542,33 @@ mod example {
         };
 
         let consumer = Consumer;
+        let arc_mut_consumer = Arc::new(Mutex::new(consumer));
 
-        event.subscribe(Arc::new(Mutex::new(consumer)));
+        event.subscribe(arc_mut_consumer.clone());
 
         event.notify().await;
 
         event.update_data(new_data).await;
 
         let second_consumer = Consumer;
+        let arc_mut_second_consumer = Arc::new(Mutex::new(second_consumer));
 
-        match event.subscribe(Arc::new(Mutex::new(second_consumer))) {
+        match event.subscribe(arc_mut_second_consumer.clone()) {
             None => assert!(false),
             Some(err) => println!("You cannot subscribe twice: {}", err),
         }
+
+        event.unsubscribe(arc_mut_consumer.clone());
+
+        match event.subscribe(arc_mut_second_consumer.clone()) {
+            None => assert!(true),
+            Some(err) => assert!(false),
+        }
+    
     }
 
     #[tokio::test]
-    async fn test_gateway() {
+    async fn test_gateway_establish() {
         let _gateway = Gateway::new("ws://localhost:3001/".to_string())
             .await
             .unwrap();