@@ -56,7 +56,7 @@ def test_add_list_of_files(servicex, did_summary_obj, single_file_info):
56
56
57
57
def test_send_on (servicex , did_summary_obj , single_file_info ):
58
58
acc = Accumulator (sx = servicex , sum = did_summary_obj )
59
- acc .add ([single_file_info , single_file_info , single_file_info ])
59
+ acc .add ([single_file_info , single_file_info , single_file_info , single_file_info ])
60
60
acc .send_on (3 )
61
61
servicex .put_file_add_bulk .assert_called_with (
62
62
[single_file_info , single_file_info , single_file_info ]
@@ -65,6 +65,17 @@ def test_send_on(servicex, did_summary_obj, single_file_info):
65
65
assert acc .summary .file_count == 3
66
66
67
67
68
+ def test_send_on_all_files (servicex , did_summary_obj , single_file_info ):
69
+ acc = Accumulator (sx = servicex , sum = did_summary_obj )
70
+ acc .add ([single_file_info , single_file_info , single_file_info ])
71
+ acc .send_on (- 1 )
72
+ servicex .put_file_add_bulk .assert_called_with (
73
+ [single_file_info , single_file_info , single_file_info ]
74
+ )
75
+ assert acc .cache_len == 0
76
+ assert acc .summary .file_count == 3
77
+
78
+
68
79
def test_invalid_constructor_arg (servicex , did_summary_obj ):
69
80
with pytest .raises (ValueError ):
70
81
acc = Accumulator (sx = servicex , sum = did_summary_obj )
0 commit comments