Implement unit tests for driver