Add tests for wfa drivers