模仿请求。post[重复]

Question

这是我第一次编写单元测试，尽管我尽了最大的努力，但还是不可避免地给您带来了麻烦。我试图模拟requests.post，但是我的测试函数没有达到预期的效果，它诱导了一个404状态代码，以便测试错误处理。

MyModule.Py

def scrape(data):
    logger.debug(f'\nBeginning scrape function')
    result = {}
    exceptions = {}
    for id, receipts in data.items():
        logger.debug(f'Looking up Id # {id} and receipts: \n{receipts}')
        dispositions = []
        for receipt in receipts:
            logger.debug(f'The length of receipts is:' + str(len(receipts)))
            attempts = 1
            while attempts < 6:
                logger.debug(
                    f'This is attempt number {attempts} to search for {receipt}')
                payload = {'receipt': 'receipt',
                           'button': 'CHECK+STATUS', }
                try:
                    NOW = datetime.today().strftime('%c')
                    logger.debug(NOW)
                    logger.debug(f'Making post request for: {receipt}')
                    response = requests.post(URL, data=payload, headers=HEADERS, timeout=10)
                except Exception as e:
                    logger.debug(f'There was an exception: {e}')
                    exceptions[id] = receipt + f': {e}'
                    time.sleep(3)
                    attempts += 1
                else:
                    logger.debug(f'It worked {response.status_code}')
                    attempts = 6
                    disp = parse(response)
                    dispositions.append(f'{receipt}: {disp}')
        result[id] = dispositions
        logger.debug(f'Here is the result: {result}')
        return result

def test_scrape(self):
    print(f'\ntest_scrape running')

    # mock a 404 in scrape() here
    with patch("mymodule.requests") as patched_post:

        # mock a request response
        patched_post.return_value.status_code = 404
        print('404 mocked')

        # verify the function returns nothing due to 404
        result = scrape(test_data)
        print(f'\n{result}')
        mock_requests.post.assert_called_once()
        self.assertEqual(result, {})

Answer 1

def test_scrape（self）：打印(f'\ntest_scrape running')

# mock a 404 in scrape() here
with patch("mymodule.requests") as patched_post:

    # mock a request response
    patched_post.return_value.status_code = 404
    print('404 mocked')

    # verify the function returns nothing due to 404
    result = scrape(test_data)
    print(f'\n{result}')
    mock_requests.post.assert_called_once()
    self.assertEqual(result, {})