Refactor search integration tests