Added initial scheduler implementation

1 files changed, 3 insertions, 3 deletions

diff --git a/src/mock.rs b/src/mock.rs
index 4ee9ed4..3b3e3cf 100644
--- a/src/mock.rs
+++ b/src/mock.rs
@@ -2,13 +2,13 @@ use iocuddle::*;
 use libc::rand;
 use rand::Rng;
 
-trait KernelModule {
+pub trait KernelModule {
     fn start_trace(&mut self, _pid: u64) {}
     fn stop_trace(&mut self, _pid: u64) {}
     fn read_consumption(&mut self, pid: u64) -> u64;
 }
 
-struct MockModule;
+pub struct MockModule;
 
 impl KernelModule for MockModule {
     fn start_trace(&mut self, pid: u64) {
@@ -25,7 +25,7 @@ impl KernelModule for MockModule {
 }
 
 #[derive(Debug)]
-struct KernelDriver {
+pub struct KernelDriver {
     file: std::fs::File,
 }