devices: code cleanup

* Removed unused functions
* Moved test code to `tests` module
* Removed unnecessary pub

Signed-off-by: Andreea Florescu <[email protected]>

Author: andreeaflorescu

devices/src/virtio/block.rs

@@ -237,7 +237,7 @@ impl Request {
     }
 }
 
-pub struct BlockEpollHandler {
+struct BlockEpollHandler {
     queues: Vec<Queue>,
     mem: GuestMemory,
     disk_image: File,
@@ -333,21 +333,6 @@ impl BlockEpollHandler {
         }
     }
 
-    #[cfg(test)]
-    fn set_queue(&mut self, idx: usize, q: Queue) {
-        self.queues[idx] = q;
-    }
-
-    #[cfg(test)]
-    fn get_rate_limiter(&self) -> &RateLimiter {
-        &self.rate_limiter
-    }
-
-    #[cfg(test)]
-    fn set_rate_limiter(&mut self, rate_limiter: RateLimiter) {
-        self.rate_limiter = rate_limiter;
-    }
-
     fn update_disk_image(&mut self, disk_image: File) {
         self.disk_image = disk_image;
         self.disk_image_id = build_disk_image_id(&self.disk_image);
@@ -641,6 +626,20 @@ mod tests {
         }};
     }
 
+    impl BlockEpollHandler {
+        fn set_queue(&mut self, idx: usize, q: Queue) {
+            self.queues[idx] = q;
+        }
+
+        fn get_rate_limiter(&self) -> &RateLimiter {
+            &self.rate_limiter
+        }
+
+        fn set_rate_limiter(&mut self, rate_limiter: RateLimiter) {
+            self.rate_limiter = rate_limiter;
+        }
+    }
+
     struct DummyBlock {
         block: Block,
         epoll_raw_fd: i32,