test(runtime-config): Add some tests to make sure data is cached

getsentry · Nov 20, 2024 · b94afcb · b94afcb
1 parent feffb16
commit b94afcb
Showing 1 changed file with 44 additions and 0 deletions.
diff --git a/rust_snuba/src/runtime_config.rs b/rust_snuba/src/runtime_config.rs
@@ -40,12 +40,56 @@ pub fn get_str_config(key: &str) -> Result<Option<String>, Error> {
 
 #[cfg(test)]
 mod tests {
+
     use super::*;
 
     #[test]
     fn test_runtime_config() {
         crate::testutils::initialize_python();
         let config = get_str_config("test");
         assert_eq!(config.unwrap(), None);
+        CONFIG.write().clear();
+    }
+
+    #[test]
+    fn test_runtime_config_is_stored_in_cache() {
+        crate::testutils::initialize_python();
+        assert!(!CONFIG.read().contains_key("test"));
+        let config = get_str_config("test");
+        assert_eq!(config.unwrap(), None);
+        assert!(CONFIG.read().contains_key("test"));
+        CONFIG.write().clear();
+    }
+
+    #[test]
+    fn test_runtime_config_retrieves_from_cache_within_deadline() {
+        {
+            CONFIG.write().insert(
+                "key".to_string(),
+                (
+                    Some("value".to_string()),
+                    Deadline::new(Duration::from_secs(10)),
+                ),
+            );
+        }
+        let config = get_str_config("key");
+        assert_eq!(config.unwrap(), Some("value".to_string()));
+        CONFIG.write().clear();
+    }
+
+    #[test]
+    fn test_runtime_config_invalidates_cache_outside_deadline() {
+        {
+            CONFIG.write().insert(
+                "key".to_string(),
+                (
+                    Some("value".to_string()),
+                    Deadline::new(Duration::from_secs(0)),
+                ),
+            );
+        }
+        let config = get_str_config("key");
+        assert_eq!(config.unwrap(), None);
+        CONFIG.write().clear();
     }
 }