@@ -127,64 +127,3 @@ def mock_get_tasks() -> dict[str, str]:
127
127
# with pytest.raises(db.NoResultFound):
128
128
# with session_obj() as session:
129
129
# db.get_request(dismissed_request_uid, session=session)
130
-
131
-
132
- def test_processing_time_priority_algorithm (
133
- mocker : pytest_mock .plugin .MockerFixture , session_obj : sa .orm .sessionmaker
134
- ) -> None :
135
- request1 = mock_system_request (user_uid = "user1" , request_uid = "request1" )
136
- request2 = mock_system_request (user_uid = "user1" , request_uid = "request2" )
137
-
138
- def mock_get_users_queue_from_processing_time ():
139
- return {"user1" : [request1 , request2 ]}
140
-
141
- mocker .patch (
142
- "cads_broker.database.get_users_queue_from_processing_time" ,
143
- return_value = mock_get_users_queue_from_processing_time (),
144
- )
145
-
146
- candidates = [request1 , request2 ]
147
-
148
- submitted_requests = []
149
-
150
- def mock_submit_request (self , request , priority , session ):
151
- submitted_requests .append (request .request_uid )
152
- for candidate in candidates :
153
- if candidate .request_uid == request .request_uid :
154
- candidates .remove (candidate )
155
-
156
- mocker .patch ("cads_broker.dispatcher.Broker.submit_request" , mock_submit_request )
157
-
158
- environment = Environment .Environment ()
159
- qos = QoS .QoS (
160
- rules = Rule .RuleSet (), environment = environment , rules_hash = "" , logger = logger
161
- )
162
- broker = dispatcher .Broker (
163
- client = CLIENT ,
164
- environment = environment ,
165
- qos = qos ,
166
- address = "scheduler-address" ,
167
- session_maker_read = session_obj ,
168
- session_maker_write = session_obj ,
169
- )
170
-
171
- checked_requests = []
172
-
173
- def mock_can_run (request , * args , ** kwargs ):
174
- checked_requests .append (request .request_uid )
175
- return True
176
-
177
- broker .qos .can_run = mock_can_run
178
-
179
- broker .processing_time_priority_algorithm (
180
- session_write = session_obj (), number_of_requests = 5 , candidates = candidates
181
- )
182
-
183
- assert submitted_requests == [request1 .request_uid ]
184
- assert checked_requests == [request1 .request_uid , request2 .request_uid ]
185
-
186
- broker .processing_time_priority_algorithm (
187
- session_write = session_obj (), number_of_requests = 5 , candidates = candidates
188
- )
189
-
190
- assert submitted_requests == [request1 .request_uid , request2 .request_uid ]
0 commit comments