@@ -49,28 +49,28 @@ def test_excel_template(self, app):
4949 id = self .org ['id' ],
5050 include_datasets = False )
5151
52- expected_records = [
53- {'_id' : 1 , ' reference_number' : 'sheet_test_1' , 'year' : 2026 },
54- {'_id' : 2 , ' reference_number' : 'sheet_test_2' , 'year' : 2025 },
55- {'_id' : 3 , ' reference_number' : 'sheet_test_3' , 'year' : 2024 },
52+ records = [
53+ {'reference_number' : 'sheet_test_1' , 'year' : 2026 },
54+ {'reference_number' : 'sheet_test_2' , 'year' : 2025 },
55+ {'reference_number' : 'sheet_test_3' , 'year' : 2024 },
5656 ]
5757
5858 # setup sample ds data
5959 self .lc .action .datastore_upsert (
6060 resource_id = dataset ['resources' ][0 ]['id' ],
6161 force = True ,
6262 method = 'insert' ,
63- records = expected_records )
63+ records = records )
6464
6565 # reference_number is primary key in sample, can update year
66- for r in expected_records :
66+ for r in records :
6767 r ['year' ] = 2001
68- expected_records .append ({'_id' : 4 , 'reference_number' : 'sheet_test_new' , 'year' : 2026 })
68+ records .append ({'reference_number' : 'sheet_test_new' , 'year' : 2026 })
6969
7070 # write excel file, should not raise any exceptions
7171 chromo = get_chromo (dataset ['resources' ][0 ]['name' ])
7272 book = excel_template (dataset ['type' ], org )
73- append_data (book , expected_records , chromo )
73+ append_data (book , records , chromo )
7474 blob = BytesIO ()
7575 book .save (blob )
7676
@@ -83,5 +83,12 @@ def test_excel_template(self, app):
8383 result = self .lc .action .datastore_search (
8484 resource_id = dataset ['resources' ][0 ]['id' ])
8585
86+ expected_records = [
87+ {'_id' : 1 , 'reference_number' : 'sheet_test_1' , 'year' : 2001 },
88+ {'_id' : 2 , 'reference_number' : 'sheet_test_2' , 'year' : 2001 },
89+ {'_id' : 3 , 'reference_number' : 'sheet_test_3' , 'year' : 2001 },
90+ {'_id' : 4 , 'reference_number' : 'sheet_test_new' , 'year' : 2026 },
91+ ]
92+
8693 assert result ['total' ] == 4
8794 assert result ['records' ] == expected_records
0 commit comments