class TestSessionVerbs(unittest.TestCase):
def setUp(self):
self.mock_response = mock.Mock()
self.mock_session = mock.Mock()
self.session = Session('client', 'client_secret', self.mock_session)
def test_get(self):
self.mock_session.get.return_value = self.mock_response
self.session.get('/test_url')
self.mock_session.get.assert_called_once_with('/test_url')
self.mock_response.json.assert_called_once_with()
def test_post_json(self):
test_data = {'key': 123}
self.mock_session.post.return_value = self.mock_response
self.session.post_json('/new', test_data)
self.mock_session.post.assert_called_once_with('/new', '{"key": 123}')
self.mock_response.json.assert_called_once_with()
def test_put_json(self):
test_data = {'key': 456}
self.mock_session.put.return_value = self.mock_response
self.session.put_json('/update', test_data)
self.mock_session.put.assert_called_once_with('/update', '{"key": 456}')
self.mock_response.json.assert_called_once_with()
