From 2f57cfc780a6b5d608af493b7bd0ebc2514c56d1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Aljoscha=20P=C3=B6rtner?= <aljoscha.poertner@proton.me>
Date: Sun, 1 Sep 2024 14:14:58 +0200
Subject: [PATCH] test(labels): add tests

---
 src/labels.rs | 54 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)

diff --git a/src/labels.rs b/src/labels.rs
index 65f2a6f..3a42cff 100644
--- a/src/labels.rs
+++ b/src/labels.rs
@@ -18,3 +18,57 @@ pub fn managed_by_us(obj: &Arc<Secret>) -> bool {
 pub fn get_managed_label() -> String {
     "v1.secret.runo.rocks/managed".to_string()
 }
+
+#[cfg(test)]
+mod tests {
+    use super::{get_managed_label, managed_by_us};
+    use k8s_openapi::api::core::v1::Secret;
+    use rstest::*;
+    use std::{collections::BTreeMap, sync::Arc};
+
+    #[fixture]
+    pub fn secret() -> Secret {
+        Secret {
+            metadata: kube::api::ObjectMeta {
+                ..Default::default()
+            },
+            ..Default::default()
+        }
+    }
+
+    #[fixture]
+    pub fn managed_secret(mut secret: Secret) -> Arc<Secret> {
+        let mut labels = BTreeMap::new();
+        labels.insert(get_managed_label(), "true".to_string());
+        secret.metadata.labels = Some(labels);
+        Arc::new(secret)
+    }
+
+    #[fixture]
+    pub fn unmanaged_secret_explicitly(mut secret: Secret) -> Arc<Secret> {
+        let mut labels = BTreeMap::new();
+        labels.insert(get_managed_label(), "false".to_string());
+        secret.metadata.labels = Some(labels);
+        Arc::new(secret)
+    }
+
+    #[fixture]
+    pub fn unmanaged_secret_implicitly(secret: Secret) -> Arc<Secret> {
+        Arc::new(secret)
+    }
+
+    #[rstest]
+    fn v1_managed_secret(managed_secret: Arc<Secret>) {
+        assert!(managed_by_us(&managed_secret));
+    }
+
+    #[rstest]
+    fn v1_unmanaged_secret_explicitly(unmanaged_secret_explicitly: Arc<Secret>) {
+        assert_eq!(managed_by_us(&unmanaged_secret_explicitly), false);
+    }
+
+    #[rstest]
+    fn v1_unmanaged_secret_implicitly(unmanaged_secret_explicitly: Arc<Secret>) {
+        assert_eq!(managed_by_us(&unmanaged_secret_explicitly), false);
+    }
+}