当 content_type 是字典时，单元测试很热门

Question

我正在尝试测试返回字典的函数。我收到错误 AttributeError: 'dict' object has no attribute 'content_type'。如何测试响应是否是字典？

def response_get(url):
    try:
        response = requests.get(url)
    except requests.exceptions.RequestException as e:
        raise SystemExit(e)
    data = response.json()
    return data

def test_response_get(self):
    response = response_get('https://ghibliapi.herokuapp.com/films/58611129-2dbc-4a81-a72f-77ddfc1b1b49')
    self.assertEqual(response.content_type, 'application/dict')

Answer 1

你已经返回了一个字典并且字典上没有content_type。 相反，您可以使用isinstance 和assertTrue。

import requests
import unittest

def response_get(url):
    try:
        response = requests.get(url)
    except requests.exceptions.RequestException as e:
        raise SystemExit(e)
    data = response.json()
    return data

class SimpleTest(unittest.TestCase): 

    def test_response_get(self):
        response = response_get('https://ghibliapi.herokuapp.com/films/58611129-2dbc-4a81-a72f-77ddfc1b1b49')
        self.assertTrue(isinstance(response, dict))
    
if __name__ == '__main__': 
    unittest.main()

输出：

.
----------------------------------------------------------------------
Ran 1 test in 0.966s

OK