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()