Implement Clone for MockStore

nrxus · Feb 14, 2022 · d4b9fe4 · d4b9fe4
1 parent 43027e4
commit d4b9fe4
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/src/mock_store.rs b/src/mock_store.rs
@@ -26,7 +26,7 @@ use std::{
 /// implements_default(faux::MaybeFaux::Real(3));
 /// ```
 
-#[derive(Debug)]
+#[derive(Clone, Debug)]
 pub enum MaybeFaux<T> {
     Real(T),
     Faux(MockStore),
@@ -38,15 +38,6 @@ impl<T: Default> Default for MaybeFaux<T> {
     }
 }
 
-impl<T: Clone> Clone for MaybeFaux<T> {
-    fn clone(&self) -> Self {
-        match self {
-            MaybeFaux::Real(r) => MaybeFaux::Real(r.clone()),
-            MaybeFaux::Faux(_) => panic!("cannot clone a mock"),
-        }
-    }
-}
-
 impl<T> MaybeFaux<T> {
     pub fn faux() -> Self {
         MaybeFaux::Faux(MockStore::new())
@@ -124,3 +115,12 @@ impl MockStore {
         Err(errors.join("\n\n"))
     }
 }
+
+impl Clone for MockStore {
+    fn clone(&self) -> Self {
+        let stubs = self.stubs.lock().unwrap();
+        Self {
+            stubs: Mutex::new(stubs.clone()),
+        }
+    }
+}