Update unit_tests/sources/declarative/test_concurrent_declarative_source.py

aaronsteers · web-flow · commit c70821e8acc6 · 2025-09-03T10:48:25.000-07:00
diff --git a/unit_tests/sources/declarative/test_concurrent_declarative_source.py b/unit_tests/sources/declarative/test_concurrent_declarative_source.py
@@ -3885,7 +3885,7 @@ def test_read_concurrent_declarative_source(
         output_data = [
             message.record.data for message in _run_read(manifest, _stream_name) if message.record
         ]
-        sort_key = lambda x: (x.get("partition", 0), x.get("ABC", 0), x.get("AED", 0))
+        sort_key = lambda x: (x.get("partition", 0), x.get("ABC", 0), x.get("AED", 0), x.get("USD", 0))
         assert sorted(output_data, key=sort_key) == sorted(expected_records, key=sort_key)
         mock_retriever.assert_has_calls(expected_calls)
 

Original file line number	Diff line number	Diff line change
`@@ -3885,7 +3885,7 @@ def test_read_concurrent_declarative_source(`
`3885`	`3885`	`output_data = [`
`3886`	`3886`	`message.record.data for message in _run_read(manifest, _stream_name) if message.record`
`3887`	`3887`	`]`
`3888`		`- sort_key = lambda x: (x.get("partition", 0), x.get("ABC", 0), x.get("AED", 0))`
	`3888`	`+ sort_key = lambda x: (x.get("partition", 0), x.get("ABC", 0), x.get("AED", 0), x.get("USD", 0))`
`3889`	`3889`	`assert sorted(output_data, key=sort_key) == sorted(expected_records, key=sort_key)`
`3890`	`3890`	`mock_retriever.assert_has_calls(expected_calls)`
`3891`	`3891`