Add tests for code completion