diff --git a/src/collection.rs b/src/collection.rs
index a39b049e..52fc2fc2 100644
--- a/src/collection.rs
+++ b/src/collection.rs
@@ -619,3 +619,51 @@ where
         .ok_or_else(|| std::io::Error::other("key not found"))
         .and_then(|val| serde_json::from_value(val).map_err(|err| err.into()))
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_data_collection() {
+        let mut collector = DataCollector::new(DataFilters {
+            disk_filter: None,
+            mount_filter: None,
+            temp_filter: None,
+            net_filter: None,
+        });
+
+        // #[cfg(feature = "battery")]
+        // {
+        //     collector.widgets_to_harvest.use_battery = true;
+        // }
+
+        // #[cfg(feature = "zfs")]
+        // {
+        //     collector.widgets_to_harvest.use_cache = true;
+        // }
+
+        // #[cfg(feature = "gpu")]
+        // {
+        //     collector.widgets_to_harvest.use_gpu = true;
+        // }
+
+        collector.widgets_to_harvest.use_cpu = true;
+        collector.widgets_to_harvest.use_disk = true;
+        collector.widgets_to_harvest.use_mem = true;
+        collector.widgets_to_harvest.use_net = true;
+        collector.widgets_to_harvest.use_proc = true;
+        collector.widgets_to_harvest.use_temp = true;
+
+        collector.update_data();
+
+        let data = collector.data;
+
+        assert!(!data.cpu.unwrap().is_empty());
+        assert!(!data.disks.unwrap().is_empty());
+        assert!(data.memory.is_some());
+        assert!(data.network.is_some());
+        assert!(!data.list_of_processes.unwrap().is_empty());
+        assert!(data.temperature_sensors.is_some());
+    }
+}