def test_get_first_row(self):
spreadsheet_id = "123"
sheet = "s1"
expected_first_row = ["1", "2", "3", "4"]
fake_response = Spreadsheet(
spreadsheetId=spreadsheet_id,
sheets=[
Sheet(data=[
GridData(rowData=[
RowData(values=[
CellData(formattedValue=v)
for v in expected_first_row
])
])
])
],
)
client = Mock()
client.get.return_value.execute.return_value = fake_response
with patch.object(GoogleSheetsClient, "__init__",
lambda s, credentials, scopes: None):
sheet_client = GoogleSheetsClient({"fake": "credentials"},
["auth_scopes"])
sheet_client.client = client
actual = Helpers.get_first_row(sheet_client, spreadsheet_id, sheet)
self.assertEqual(expected_first_row, actual)
client.get.assert_called_with(spreadsheetId=spreadsheet_id,
includeGridData=True,
ranges=f"{sheet}!1:1")
def test_get_first_row(self):
spreadsheet_id = "123"
sheet = "s1"
expected_first_row = ["1", "2", "3", "4"]
fake_response = Spreadsheet(
spreadsheetId=spreadsheet_id,
sheets=[
Sheet(data=[
GridData(rowData=[
RowData(values=[
CellData(formattedValue=v)
for v in expected_first_row
])
])
])
],
)
client = Mock()
client.get.return_value.execute.return_value = fake_response
actual = Helpers.get_first_row(client, spreadsheet_id, sheet)
self.assertEqual(expected_first_row, actual)
client.get.assert_called_with(spreadsheetId=spreadsheet_id,
includeGridData=True,
ranges=f"{sheet}!1:1")
