Adds test coverage for APIDriver