diff --git a/src/lib.rs b/src/lib.rs
index 9794e87..01c2ae4 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -57,3 +57,13 @@ pub fn handle_client() {
       disk.write_all(&buffer).unwrap();
     }
 }
+
+#[cfg(test)]
+mod tests {
+  use super::*;
+
+  #[test]
+  fn test_disk_locator {
+    assert_eq!(get_block_devices(), vec!["vda".to_string()])
+  }
+}