test: update tests for response passed in