@@ -60,7 +60,7 @@ def test_create_progress_reporter_with_time_interval_seconds(
60
60
"""Test that time_interval_seconds gets properly converted to float and passed to PipelineProgressReporter"""
61
61
command = mocker .Mock ()
62
62
command .option .side_effect = lambda opt : {
63
- "time -interval-seconds" : "30.5" ,
63
+ "metrics -interval-in -seconds" : "30.5" ,
64
64
"reporting-frequency" : "1000" ,
65
65
}.get (opt )
66
66
command .has_json_logging_set = False
@@ -75,7 +75,7 @@ def test_create_progress_reporter_with_time_interval_seconds(
75
75
# Verify PipelineProgressReporter was called with correct time_interval_seconds
76
76
mock_progress_reporter .assert_called_once ()
77
77
call_args = mock_progress_reporter .call_args
78
- assert_that (call_args .kwargs ["time_interval_seconds " ], equal_to (30.5 ))
78
+ assert_that (call_args .kwargs ["metrics_interval_in_seconds " ], equal_to (30.5 ))
79
79
assert_that (call_args .kwargs ["reporting_frequency" ], equal_to (1000 ))
80
80
81
81
@@ -85,7 +85,7 @@ def test_create_progress_reporter_without_time_interval_seconds(
85
85
"""Test that time_interval_seconds is None when not provided"""
86
86
command = mocker .Mock ()
87
87
command .option .side_effect = lambda opt : {
88
- "time -interval-seconds" : None ,
88
+ "metrics -interval-in -seconds" : None ,
89
89
"reporting-frequency" : "2000" ,
90
90
}.get (opt )
91
91
command .has_json_logging_set = False
@@ -100,15 +100,15 @@ def test_create_progress_reporter_without_time_interval_seconds(
100
100
# Verify PipelineProgressReporter was called with None for time_interval_seconds
101
101
mock_progress_reporter .assert_called_once ()
102
102
call_args = mock_progress_reporter .call_args
103
- assert_that (call_args .kwargs ["time_interval_seconds " ], equal_to (None ))
103
+ assert_that (call_args .kwargs ["metrics_interval_in_seconds " ], equal_to (None ))
104
104
assert_that (call_args .kwargs ["reporting_frequency" ], equal_to (2000 ))
105
105
106
106
107
107
def test_create_progress_reporter_with_json_indicator (run_pipeline_operation , mocker ):
108
108
"""Test that create_progress_reporter works correctly with JSON progress indicator"""
109
109
command = mocker .Mock ()
110
110
command .option .side_effect = lambda opt : {
111
- "time -interval-seconds" : "15.0" ,
111
+ "metrics -interval-in -seconds" : "15.0" ,
112
112
"reporting-frequency" : "500" ,
113
113
}.get (opt )
114
114
command .has_json_logging_set = True
@@ -123,7 +123,7 @@ def test_create_progress_reporter_with_json_indicator(run_pipeline_operation, mo
123
123
# Verify PipelineProgressReporter was called with correct arguments
124
124
mock_progress_reporter .assert_called_once ()
125
125
call_args = mock_progress_reporter .call_args
126
- assert_that (call_args .kwargs ["time_interval_seconds " ], equal_to (15.0 ))
126
+ assert_that (call_args .kwargs ["metrics_interval_in_seconds " ], equal_to (15.0 ))
127
127
assert_that (call_args .kwargs ["reporting_frequency" ], equal_to (500 ))
128
128
129
129
@@ -142,7 +142,7 @@ def test_make_run_request_with_time_interval_seconds_integration(
142
142
"annotations" : annotations ,
143
143
"step-outbox-size" : "10001" ,
144
144
"target" : targets ,
145
- "time -interval-seconds" : "45.0" ,
145
+ "metrics -interval-in -seconds" : "45.0" ,
146
146
"reporting-frequency" : "5000" ,
147
147
}
148
148
command .option .side_effect = lambda opt : option_responses .get (opt )
@@ -168,7 +168,7 @@ def test_make_run_request_with_time_interval_seconds_integration(
168
168
# Verify the progress reporter was created with correct time_interval_seconds
169
169
mock_progress_reporter .assert_called_once ()
170
170
call_args = mock_progress_reporter .call_args
171
- assert_that (call_args .kwargs ["time_interval_seconds " ], equal_to (45.0 ))
171
+ assert_that (call_args .kwargs ["metrics_interval_in_seconds " ], equal_to (45.0 ))
172
172
assert_that (call_args .kwargs ["reporting_frequency" ], equal_to (5000 ))
173
173
174
174
# Verify other parts of the request are still correct
0 commit comments