@@ -108,7 +108,7 @@ def test_handle_partition_done_no_other_streams_to_generate_partitions_for(self)
108
108
messages = list (handler .on_partition_generation_completed (sentinel ))
109
109
110
110
expected_messages = []
111
- assert expected_messages == messages
111
+ assert messages == expected_messages
112
112
113
113
@freezegun .freeze_time ("2020-01-01T00:00:00" )
114
114
def test_handle_last_stream_partition_done (self ):
@@ -146,7 +146,7 @@ def test_handle_last_stream_partition_done(self):
146
146
),
147
147
),
148
148
]
149
- assert expected_messages == messages
149
+ assert messages == expected_messages
150
150
assert in_order_validation_mock .mock_calls .index (
151
151
call ._another_stream .cursor .ensure_at_least_one_state_emitted
152
152
) < in_order_validation_mock .mock_calls .index (call ._message_repository .consume_queue )
@@ -223,7 +223,7 @@ def test_handle_on_partition_complete_sentinel_with_messages_from_repository(sel
223
223
expected_messages = [
224
224
AirbyteMessage (type = MessageType .LOG , log = AirbyteLogMessage (level = LogLevel .INFO , message = "message emitted from the repository" ))
225
225
]
226
- assert expected_messages == messages
226
+ assert messages == expected_messages
227
227
228
228
partition .close .assert_called_once ()
229
229
@@ -267,7 +267,7 @@ def test_handle_on_partition_complete_sentinel_yields_status_message_if_the_stre
267
267
),
268
268
)
269
269
]
270
- assert expected_messages == messages
270
+ assert messages == expected_messages
271
271
self ._a_closed_partition .close .assert_called_once ()
272
272
273
273
@freezegun .freeze_time ("2020-01-01T00:00:00" )
@@ -294,7 +294,7 @@ def test_handle_on_partition_complete_sentinel_yields_no_status_message_if_the_s
294
294
messages = list (handler .on_partition_complete_sentinel (sentinel ))
295
295
296
296
expected_messages = []
297
- assert expected_messages == messages
297
+ assert messages == expected_messages
298
298
partition .close .assert_called_once ()
299
299
300
300
@freezegun .freeze_time ("2020-01-01T00:00:00" )
@@ -331,7 +331,7 @@ def test_on_record_no_status_message_no_repository_messge(self):
331
331
),
332
332
)
333
333
]
334
- assert expected_messages == messages
334
+ assert messages == expected_messages
335
335
336
336
@freezegun .freeze_time ("2020-01-01T00:00:00" )
337
337
def test_on_record_with_repository_messge (self ):
@@ -381,7 +381,7 @@ def test_on_record_with_repository_messge(self):
381
381
),
382
382
AirbyteMessage (type = MessageType .LOG , log = AirbyteLogMessage (level = LogLevel .INFO , message = "message emitted from the repository" )),
383
383
]
384
- assert expected_messages == messages
384
+ assert messages == expected_messages
385
385
assert handler ._record_counter [_STREAM_NAME ] == 2
386
386
387
387
@freezegun .freeze_time ("2020-01-01T00:00:00" )
@@ -423,7 +423,7 @@ def test_on_record_emits_status_message_on_first_record_no_repository_message(se
423
423
),
424
424
),
425
425
]
426
- assert expected_messages == messages
426
+ assert messages == expected_messages
427
427
428
428
@freezegun .freeze_time ("2020-01-01T00:00:00" )
429
429
def test_on_record_emits_status_message_on_first_record_with_repository_message (self ):
@@ -477,7 +477,7 @@ def test_on_record_emits_status_message_on_first_record_with_repository_message(
477
477
),
478
478
AirbyteMessage (type = MessageType .LOG , log = AirbyteLogMessage (level = LogLevel .INFO , message = "message emitted from the repository" )),
479
479
]
480
- assert expected_messages == messages
480
+ assert messages == expected_messages
481
481
482
482
@freezegun .freeze_time ("2020-01-01T00:00:00" )
483
483
def test_on_exception_return_trace_message_and_on_stream_complete_return_stream_status (self ):
0 commit comments