@@ -47,6 +47,25 @@ def test_with_data_row_object_buffered(self, client, data_row,
47
47
assert export_task .get_total_lines (stream_type = StreamType .RESULT ) == 1
48
48
assert list (export_task .get_buffered_stream ())[0 ].json ["data_row" ]["id" ] == data_row .uid
49
49
50
+ def test_with_data_row_object_buffered_result (self , client , data_row ,
51
+ wait_for_data_row_processing ):
52
+ data_row = wait_for_data_row_processing (client , data_row )
53
+ time .sleep (7 ) # temp fix for ES indexing delay
54
+ export_task = DataRow .export (
55
+ client = client ,
56
+ data_rows = [data_row ],
57
+ task_name = "TestExportDataRow:test_with_data_row_object_buffered_result" ,
58
+ )
59
+ export_task .wait_till_done ()
60
+ assert export_task .status == "COMPLETE"
61
+ assert isinstance (export_task , ExportTask )
62
+ assert export_task .has_result ()
63
+ assert export_task .has_errors () is False
64
+ assert export_task .get_total_file_size (
65
+ stream_type = StreamType .RESULT ) > 0
66
+ assert export_task .get_total_lines (stream_type = StreamType .RESULT ) == 1
67
+ assert export_task .result [0 ]["data_row" ]["id" ] == data_row .uid
68
+
50
69
def test_with_id (self , client , data_row , wait_for_data_row_processing ):
51
70
data_row = wait_for_data_row_processing (client , data_row )
52
71
time .sleep (7 ) # temp fix for ES indexing delay
0 commit comments