def _send(url: str, data: dict, headers: dict, cookies: dict, method: str): url = f"{ENV.base_url()}{url}" if headers is None: headers = {} if cookies is None: cookies = {} additional_header = {'X-THIS_IS_TEST': 'True'} headers.update(additional_header) Logger.get_instance().add_request(url, data, headers, cookies, method) if method == 'GET': response = requests.get(url, params=data, headers=headers, cookies=cookies) elif method == 'POST': response = requests.post(url, data=data, headers=headers, cookies=cookies) elif method == 'PUT': response = requests.put(url, data=data, headers=headers, cookies=cookies) elif method == 'DELETE': response = requests.delete(url, data=data, headers=headers, cookies=cookies) else: raise Exception(f'Bad HTTP method "{method}" was received') Logger.get_instance().add_response(response) return response
def teardown(self): Logger.get_instance().write_log_to_file()