Skip to content

Commit fe732b4

Browse files
committed
Explicit test values for insert_records
1 parent 893d817 commit fe732b4

File tree

1 file changed

+21
-8
lines changed

1 file changed

+21
-8
lines changed

tests/worksheet_test.py

Lines changed: 21 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1723,17 +1723,22 @@ def test_append_records(self):
17231723
)
17241724

17251725
@pytest.mark.vcr()
1726-
def test_insert_rows(self):
1726+
def test_insert_records(self):
17271727
w = self.spreadsheet.sheet1
1728-
values = [
1728+
values_before = [
1729+
["header1", "header2"],
1730+
["value1", "value2"],
1731+
]
1732+
w.update(values_before, "A1:B2")
1733+
values_after = [
17291734
["header1", "header2"],
1735+
["value3", "value4"],
1736+
["", "value5"],
17301737
["value1", "value2"],
17311738
]
1732-
w.update(values, "A1:B2")
1733-
new_values = [values[0], ["value3", "value4"], ["", "value5"], *values[1:]]
17341739
self.assertEqual(
17351740
w.get_all_values(value_render_option=utils.ValueRenderOption.unformatted),
1736-
values,
1741+
values_before,
17371742
)
17381743

17391744
w.insert_records(
@@ -1745,21 +1750,29 @@ def test_insert_rows(self):
17451750

17461751
self.assertEqual(
17471752
w.get_all_values(value_render_option=utils.ValueRenderOption.unformatted),
1748-
new_values,
1753+
values_after,
17491754
)
17501755

17511756
with pytest.raises(GSpreadException):
17521757
w.insert_record({"header1": "error value1", "location": "error value2"})
17531758

17541759
w.insert_record(
17551760
{"header4": "ignore value", "header1": "value6", "header2": "value7"},
1756-
insert_row=5,
1761+
insert_row=4,
17571762
ignore_extra_headers=True,
17581763
)
17591764

1765+
values_after_single_entry = [
1766+
["header1", "header2"],
1767+
["value3", "value4"],
1768+
["", "value5"],
1769+
["value6", "value7"],
1770+
["value1", "value2"],
1771+
]
1772+
17601773
self.assertEqual(
17611774
w.get_all_values(),
1762-
[*new_values, ["value6", "value7"]],
1775+
values_after_single_entry,
17631776
)
17641777

17651778
@pytest.mark.vcr()

0 commit comments

Comments
 (0)