diff --git a/tests/test_unit.py b/tests/test_unit.py index be85543..eeea6ea 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -162,12 +162,16 @@ def test__getattr__(self, mock_lib): client.version(3) self.assertEqual(client._version, 3) + # Test Response Headers + r = client.get() + self.assertEqual(r.headers, 'HEADERS') + # Test GET client._url_path += ['test'] r = client.get() self.assertEqual(r.status_code, 200) - # Test POST + # Test PUT r = client.put() self.assertEqual(r.status_code, 200)