From b6873cf66106e0129661c21ebfdc642716f44567 Mon Sep 17 00:00:00 2001
From: realaravinth <realaravinth@batsense.net>
Date: Wed, 9 Jun 2021 14:30:41 +0530
Subject: [PATCH] race test

---
 src/master/embedded/master.rs | 10 +++++---
 src/master/mod.rs             |  2 +-
 src/master/redis/master.rs    | 45 ++++++++++++++++++++++++++++++++---
 3 files changed, 50 insertions(+), 7 deletions(-)

diff --git a/src/master/embedded/master.rs b/src/master/embedded/master.rs
index 39af6de..a6d3f66 100644
--- a/src/master/embedded/master.rs
+++ b/src/master/embedded/master.rs
@@ -92,14 +92,18 @@ impl Handler<AddVisitor> for Master {
     fn handle(&mut self, m: AddVisitor, ctx: &mut Self::Context) -> Self::Result {
         let (tx, rx) = channel();
         match self.get_site(&m.0) {
-            None => tx.send(Ok(None)).unwrap(),
+            None => {
+                let _ = tx.send(Ok(None));
+            }
             Some(addr) => {
                 let fut = async move {
                     match addr.send(super::counter::AddVisitor).await {
-                        Ok(val) => tx.send(Ok(Some(val))).unwrap(),
+                        Ok(val) => {
+                            let _ = tx.send(Ok(Some(val)));
+                        }
                         Err(e) => {
                             let err: CaptchaError = e.into();
-                            tx.send(Err(err)).unwrap();
+                            let _ = tx.send(Err(err));
                         }
                     }
                 }
diff --git a/src/master/mod.rs b/src/master/mod.rs
index eb8b3c0..0825a0c 100644
--- a/src/master/mod.rs
+++ b/src/master/mod.rs
@@ -71,7 +71,7 @@ pub mod messages {
     use crate::mcaptcha::MCaptcha;
 
     /// Message to add visitor to an [MCaptcha] actor
-    #[derive(Message)]
+    #[derive(Message, Clone)]
     #[rtype(result = "Receiver<CaptchaResult<Option<super::AddVisitorResult>>>")]
     pub struct AddVisitor(pub String);
 
diff --git a/src/master/redis/master.rs b/src/master/redis/master.rs
index 99a0060..9d1df46 100644
--- a/src/master/redis/master.rs
+++ b/src/master/redis/master.rs
@@ -52,7 +52,7 @@ impl Handler<AddVisitor> for Master {
         let con = self.redis.get_client();
         let fut = async move {
             let res = con.add_visitor(m).await;
-            tx.send(res).unwrap()
+            let _ = tx.send(res);
         }
         .into_actor(self);
         ctx.wait(fut);
@@ -82,12 +82,11 @@ mod tests {
     use crate::master::redis::master::Master;
     use crate::redis::RedisConfig;
 
-    const CAPTCHA_NAME: &str = "REDIS_MASTER_CAPTCHA_TEST";
-
     const REDIS_URL: &str = "redis://127.0.1.1/";
 
     #[actix_rt::test]
     async fn redis_master_works() {
+        const CAPTCHA_NAME: &str = "REDIS_MASTER_CAPTCHA_TEST";
         let master = Master::new(RedisConfig::Single(REDIS_URL.into())).await;
         let sec_master = Master::new(RedisConfig::Single(REDIS_URL.into())).await;
         let r = sec_master.unwrap().redis.get_client();
@@ -116,5 +115,45 @@ mod tests {
 
         let timer_expire = std::time::Duration::new(duration, 0);
         actix::clock::delay_for(timer_expire).await;
+        let visitors = r.get_visitors(CAPTCHA_NAME).await.unwrap();
+        assert_eq!(visitors, 0);
+    }
+
+    #[actix_rt::test]
+    async fn race_redis_master() {
+        const CAPTCHA_NAME: &str = "REDIS_MASTER_CAPTCHA_RACE";
+
+        let master = Master::new(RedisConfig::Single(REDIS_URL.into())).await;
+        let sec_master = Master::new(RedisConfig::Single(REDIS_URL.into())).await;
+        let r = sec_master.unwrap().redis.get_client();
+
+        assert!(master.is_ok());
+        let master = master.unwrap();
+        {
+            let _ = master.redis.get_client().delete_captcha(CAPTCHA_NAME).await;
+        }
+
+        let addr = master.start();
+
+        let mcaptcha = get_mcaptcha();
+        let duration = mcaptcha.get_duration();
+
+        let add_mcaptcha_msg = AddSite {
+            id: CAPTCHA_NAME.into(),
+            mcaptcha,
+        };
+        addr.send(add_mcaptcha_msg).await.unwrap();
+
+        let add_visitor_msg = AddVisitor(CAPTCHA_NAME.into());
+        for _ in 0..500 {
+            addr.send(add_visitor_msg.clone()).await.unwrap();
+        }
+        let visitors = r.get_visitors(CAPTCHA_NAME).await.unwrap();
+        assert_eq!(visitors, 500);
+
+        let timer_expire = std::time::Duration::new(duration, 0);
+        actix::clock::delay_for(timer_expire).await;
+        let visitors = r.get_visitors(CAPTCHA_NAME).await.unwrap();
+        assert_eq!(visitors, 0);
     }
 }