More unit tests for driver shim