From 514929c3b8ee0d4f540be5eb4bc8c1a10e62095d Mon Sep 17 00:00:00 2001
From: Weicao-CatilGrass <1992414357@qq.com>
Date: Tue, 9 Jun 2026 21:08:20 +0800
Subject: Add unit and integration tests for mingling_core

---
 mingling_core/src/program/single_instance.rs | 46 ++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

(limited to 'mingling_core/src/program/single_instance.rs')

diff --git a/mingling_core/src/program/single_instance.rs b/mingling_core/src/program/single_instance.rs
index d16bcf5..8b165bf 100644
--- a/mingling_core/src/program/single_instance.rs
+++ b/mingling_core/src/program/single_instance.rs
@@ -110,3 +110,49 @@ where
 {
     THIS_PROGRAM.get_raw()?.downcast_ref::<Program<C>>()
 }
+
+#[cfg(test)]
+mod tests {
+    use super::ProgramCell;
+
+    #[test]
+    fn test_program_cell_set_and_get_raw() {
+        let cell = ProgramCell::new();
+        cell.set(Box::new(42_i32));
+        let val = cell.get_raw();
+        assert!(val.is_some());
+        assert_eq!(*val.unwrap().downcast_ref::<i32>().unwrap(), 42);
+    }
+
+    #[test]
+    fn test_program_cell_get_raw_uninitialized() {
+        let cell = ProgramCell::new();
+        assert!(cell.get_raw().is_none());
+    }
+
+    #[test]
+    #[should_panic(expected = "ProgramCell already initialized")]
+    fn test_program_cell_set_twice_panics() {
+        let cell = ProgramCell::new();
+        cell.set(Box::new(1_i32));
+        cell.set(Box::new(2_i32));
+    }
+
+    #[test]
+    fn test_program_cell_take() {
+        let cell = ProgramCell::new();
+        cell.set(Box::new(99_i32));
+
+        // SAFETY: test-local cell, no outstanding references.
+        let taken = unsafe { cell.take() };
+        assert!(taken.is_some());
+        assert_eq!(*taken.unwrap().downcast_ref::<i32>().unwrap(), 99);
+
+        // After take, get_raw returns None.
+        assert!(cell.get_raw().is_none());
+
+        // Calling take again returns None.
+        let taken_again = unsafe { cell.take() };
+        assert!(taken_again.is_none());
+    }
+}
-- 
cgit