def test_header(key, value): response = Response() response.put_header(key, value) assert response.get_header(key) == value assert response.get_headers()[key] == value new_dict = {"headerA": "valueA", "headerB": "valueB"} response.update_headers(new_dict) assert response.get_header(key) == value assert response.get_headers()[key] == value
def test_redirect(url): response = Response() response.set_redirect_found(url) assert response.get_status_code() == SC.REDIRECT_FOUND assert response.get_header("Location") == url