{"id":649,"date":"2023-07-09T17:38:11","date_gmt":"2023-07-09T09:38:11","guid":{"rendered":"https:\/\/www.wennroy.com\/?p=649"},"modified":"2023-07-09T17:40:11","modified_gmt":"2023-07-09T09:40:11","slug":"python-someways_to_mock_requests_and_argparse_input","status":"publish","type":"post","link":"https:\/\/wennroy.com\/index.php\/2023\/07\/09\/python-someways_to_mock_requests_and_argparse_input\/","title":{"rendered":"Python \u4e00\u4e9b\u6a21\u62df\u8bf7\u6c42\u548c\u8f93\u5165\u7684\u65b9\u6cd5"},"content":{"rendered":"\n<h2 class=\"wp-block-heading\">requests_mock<\/h2>\n\n\n\n<p class=\"wp-block-paragraph\">requests_mock\u662f\u4e00\u4e2a\u53ef\u4ee5\u6a21\u62dfrequests\u8bf7\u6c42\u7684\u4e00\u4e2a\u5305\uff0c\u901a\u5e38\u7528\u4e8e\u6d4b\u8bd5\u9636\u6bb5\u3002<\/p>\n\n\n\n<p class=\"wp-block-paragraph\"><a href=\"https:\/\/requests-mock.readthedocs.io\/en\/latest\/\">requests_mock documentation<\/a><\/p>\n\n\n\n<p class=\"wp-block-paragraph\">\u5b83\u65e2\u53ef\u4ee5\u652f\u6301real requests\u4e5f\u53ef\u4ee5\u4e0d\u8fdb\u884creal requests\u800c\u662f\u901a\u8fc7\u8bbe\u7f6e\u597d\u7684\u5185\u5bb9\u8fdb\u884crequests\u3002<\/p>\n\n\n\n<p class=\"wp-block-paragraph\">\u4e00\u4e9b\u6bd4\u8f83\u5e38\u7528\u7684\u65b9\u6cd5\u5305\u62ec\u4f7f\u7528<\/p>\n\n\n\n<h3 class=\"wp-block-heading\">Context Mangaer<\/h3>\n\n\n\n<pre class=\"EnlighterJSRAW\" data-enlighter-language=\"python\" data-enlighter-theme=\"\" data-enlighter-highlight=\"\" data-enlighter-linenumbers=\"\" data-enlighter-lineoffset=\"\" data-enlighter-title=\"\" data-enlighter-group=\"\">import requests\nimport requests_mock\n\nexample_url = 'https:\/\/www.example.com'\nwith requests_mock.Mocker() as m:\n    m.get(example_url, text=\"response\")\n    print(requests.get(example_url).text)\n\n# response<\/pre>\n\n\n\n<h3 class=\"wp-block-heading\">Decorator<\/h3>\n\n\n\n<p class=\"wp-block-paragraph\">\u5f53\u7136\u4e5f\u53ef\u4ee5\u7528\u88c5\u9970\u5668\u6765\u88c5\u9970\uff0c\u5982\u679c\u51fd\u6570\u6240\u9700\u8981\u7684\u53c2\u6570\u4e0e<code>requests_mock<\/code>\u4f20\u9012\u7684\u53c2\u6570\u4ea7\u751f\u51b2\u7a81\uff0c\u90a3\u4e48\u53ef\u4ee5\u8003\u8651\u7528<code>kwargs<\/code>\u7684\u65b9\u5f0f\u4f20\u5165\uff0c\u6dfb\u52a0<code>kw<\/code>\u6765\u6307\u5b9a\u7d22\u5f15\u3002<\/p>\n\n\n\n<pre class=\"EnlighterJSRAW\" data-enlighter-language=\"python\" data-enlighter-theme=\"\" data-enlighter-highlight=\"\" data-enlighter-linenumbers=\"\" data-enlighter-lineoffset=\"\" data-enlighter-title=\"\" data-enlighter-group=\"\">example_url = 'https:\/\/www.example.com'\n@requests_mock.Mocker()\ndef test_function(m):\n    m.post(example_url, request_headers={\"username\": \"user\", \"password\": \"pw\"}, text=\"sessionid1\")\n    return requests.post(example_url, headers={\"username\": \"user\", \"password\": \"pw\"}).text\n\nsessionid = test_function()\nprint(sessionid)\n#sessionid1\n\n# If the position of the mock is likely to conflict with other arguments you can pass the kw argument to the Mocker to have the mocker object passed as that keyword argument instead.\n\n@requests_mock.Mocker(kw='mock')\ndef test_kw_function(sessionid, **kwargs):\n    kwargs['mock'].delete(f\"{example_url}\/{sessionid}\", text=\"delete success\", status_code=200)\n    return requests.delete(f\"{example_url}\/{sessionid}\").text\n\nprint(test_kw_function(sessionid))\n#delete success\n<\/pre>\n\n\n\n<h2 class=\"wp-block-heading\">unittest.mock<\/h2>\n\n\n\n<p class=\"wp-block-paragraph\"><a href=\"https:\/\/docs.python.org\/3\/library\/unittest.mock.html#module-unittest.mock\"><code>unittest.mock<\/code><\/a>&nbsp;is a library for testing in Python. It allows you to replace parts of your system under test with mock objects and make assertions about how they have been used.<\/p>\n\n\n\n<p class=\"wp-block-paragraph\">\u4e4b\u524d\u6211\u6240\u9047\u5230\u7684\u95ee\u9898\u4e3b\u8981\u662f\u5173\u4e8e\u5982\u4f55\u6a21\u62df<code>argparse<\/code>\uff0c\u56e0\u6b64\u6211\u8fd9\u91cc\u4e3b\u8981\u6d89\u53ca\u5982\u4f55\u5229\u7528<code>unittest.mock<\/code>\u6765\u6a21\u62df\u7c7b\u4f3c\u4e8e<code>argparse<\/code>\u7684\u8f93\u5165\u3002\u6211\u4eec\u8fd9\u91cc\u4e3b\u8981\u7528\u5230\u7684\u662f<code>patch<\/code>\u8fd9\u4e2a\u51fd\u6570<\/p>\n\n\n\n<pre class=\"EnlighterJSRAW\" data-enlighter-language=\"python\" data-enlighter-theme=\"\" data-enlighter-highlight=\"\" data-enlighter-linenumbers=\"\" data-enlighter-lineoffset=\"\" data-enlighter-title=\"\" data-enlighter-group=\"\">from unittest import mock\nimport argparse\n\nwith mock.patch('argparse._sys.argv', ['python', '--user', 'username', '--password', 'pw']):\n    parser = argparse.ArgumentParser()\n    parser.add_argument(\"--user\", type=str)\n    parser.add_argument(\"--password\", type=str)\n    args, unknown = parser.parse_known_args()\n    assert vars(args) == {\"user\": \"username\", \"password\": \"pw\"}\n    print(vars(args), unknown)\n\n# {'user': 'username', 'password': 'pw'} []<\/pre>\n\n\n\n<p class=\"wp-block-paragraph\">\u8fd9\u6837\u6211\u4eec\u5c31\u80fd\u76f4\u63a5\u6a21\u62df<code>argparse<\/code>\u7684\u8f93\u5165\u4e86\u3002<\/p>\n\n\n\n<p class=\"wp-block-paragraph\"><\/p>\n","protected":false},"excerpt":{"rendered":"<p>requests_mock requests_mo &hellip;<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"_import_markdown_pro_load_document_selector":0,"_import_markdown_pro_submit_text_textarea":"","footnotes":""},"categories":[41,56,55],"tags":[61,51,60,59,58],"class_list":["post-649","post","type-post","status-publish","format-standard","hentry","category-python","category-weekly-update","category-55","tag-argparse","tag-python","tag-requests","tag-unittest","tag-weekly-update"],"_links":{"self":[{"href":"https:\/\/wennroy.com\/index.php\/wp-json\/wp\/v2\/posts\/649","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/wennroy.com\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/wennroy.com\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/wennroy.com\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/wennroy.com\/index.php\/wp-json\/wp\/v2\/comments?post=649"}],"version-history":[{"count":3,"href":"https:\/\/wennroy.com\/index.php\/wp-json\/wp\/v2\/posts\/649\/revisions"}],"predecessor-version":[{"id":652,"href":"https:\/\/wennroy.com\/index.php\/wp-json\/wp\/v2\/posts\/649\/revisions\/652"}],"wp:attachment":[{"href":"https:\/\/wennroy.com\/index.php\/wp-json\/wp\/v2\/media?parent=649"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/wennroy.com\/index.php\/wp-json\/wp\/v2\/categories?post=649"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/wennroy.com\/index.php\/wp-json\/wp\/v2\/tags?post=649"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}