@@ -179,7 +179,7 @@ def test_create_lookup_key_index_preserves_structure(setup):
179179 fields = ["a" , "b" , "c" ]
180180
181181 expected_headers = (
182- '[{"col1":{"type":"str","key":true}},' + '{"col2":{"type":"str"}},{"col3":{"type":"str"}}]'
182+ '[{"col1":{"type":"str","key":true}},' + '{"col2":{"type":"str"}},{"col3":{"type":"str"}}]'
183183 )
184184 with mock .patch .object (lookup , "send_control" , wraps = lookup .send_control ) as lookup_spy :
185185 lookup .send_headers (headers = headers , key_index = 0 , event = "START" , action = "FULL" )
@@ -204,8 +204,8 @@ def test_send_headers_with_type_of_key(setup):
204204 headers = ["col1" , "col2" , "col3" ]
205205
206206 expected_headers = (
207- '[{"col1":{"type":"int4","key":true}},'
208- + '{"col2":{"type":"str"}},{"col3":{"type":"str"}}]'
207+ '[{"col1":{"type":"int4","key":true}},'
208+ + '{"col2":{"type":"str"}},{"col3":{"type":"str"}}]'
209209 )
210210 with mock .patch .object (lookup , "send_control" , wraps = lookup .send_control ) as lookup_spy :
211211 lookup .send_headers (
@@ -309,7 +309,7 @@ def test_escape_quotes_in_send_data_line_key(setup):
309309
310310 with mock .patch .object (Lookup , "clean_field" , wraps = Lookup .clean_field ) as clean_field :
311311 lookup .send_data_line (key_index = 0 , fields = ["11" , 'Double quotes"' ])
312- clean_field .assert_called_with ('Double quotes"' , True )
312+ clean_field .assert_called_with ('Double quotes"' , True , False )
313313
314314
315315# Test to make sure escape_quotes is propagated correctly
@@ -325,7 +325,7 @@ def test_escape_quotes_in_send_data_line(setup):
325325
326326 with mock .patch .object (Lookup , "clean_field" , wraps = Lookup .clean_field ) as clean_field :
327327 lookup .send_data_line (key_index = 0 , fields = ["11" , 'Double quotes"' ])
328- clean_field .assert_called_with ('Double quotes"' , True )
328+ clean_field .assert_called_with ('Double quotes"' , True , False )
329329
330330 # Test to make sure escape_quotes is propagated correctly
331331
@@ -342,7 +342,7 @@ def test_escape_quotes_in_send_csv(setup):
342342
343343 with mock .patch .object (Lookup , "clean_field" , wraps = Lookup .clean_field ) as clean_field :
344344 lookup .send_csv (path = setup .lookup_file , has_header = True , key = setup .lookup_key )
345- clean_field .assert_called_with ("ffffff" , True )
345+ clean_field .assert_called_with ("ffffff" , True , False )
346346
347347 # Test to make sure escape_quotes is propagated correctly
348348
@@ -364,7 +364,7 @@ def test_escape_quotes_in_send_csv_delete_index(setup):
364364 key = setup .lookup_key ,
365365 delete_field = "Green" ,
366366 )
367- clean_field .assert_called_with ("ffffff" , True )
367+ clean_field .assert_called_with ("ffffff" , True , False )
368368
369369
370370if __name__ == "__main__" :
0 commit comments