def test_simple_create_one(self, mock): """Ensure that simple CREATE one method will add a new row to our existing a spreadsheet""" mock.return_value = MagicMock(status_code=200, content=json.dumps([{ 'id': '1', 'name': 'Peter', 'score': '42' }, { 'id': '2', 'name': 'Lois', 'score': '89' }])) client = SheetsuClient(**self.kwargs) response = client.create_one(id=2, name="Lois", score=89) self.assertEqual(response, [{ 'id': '1', 'name': 'Peter', 'score': '42' }, { 'id': '2', 'name': 'Lois', 'score': '89' }])
def test_simple_create_one_different_sheet(self, mock): """Ensure that simple CREATE one method will add a new row to our specified sheet""" mock.return_value = MagicMock(status_code=200, content=json.dumps([ { 'id': '1', 'name': 'Peter', 'score': '42' }, ])) client = SheetsuClient(**self.kwargs) response = client.create_one(sheet="Sheet2", id=1, name="Peter", score=42) self.assertEqual(response, [ { 'id': '1', 'name': 'Peter', 'score': '42' }, ])
def test_create_one_or_many_fail(self, mock): """Ensure that if request fails, that proper response is given""" mock.return_value = MagicMock(status_code=400) client = SheetsuClient(**self.kwargs) response = client.create_one() self.assertIsNone(response) response = client.create_many() self.assertIsNone(response)