Add tests for Driver