Skip to content

Commit 33037c3

Browse files
rename one argument
1 parent 8e014a9 commit 33037c3

File tree

3 files changed

+3
-64
lines changed

3 files changed

+3
-64
lines changed

cads_broker/factory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,9 +76,9 @@ def register_functions():
7676
)
7777
expressions.FunctionFactory.FunctionFactory.register_function(
7878
"user_last_completed_request",
79-
lambda context, max_time=24 * 60 * 60: database.user_last_completed_request(
79+
lambda context, interval=24 * 60 * 60: database.user_last_completed_request(
8080
user_uid=context.request.user_uid,
81-
interval=max_time,
81+
interval=interval,
8282
session=context.environment.session,
8383
),
8484
)

tests/test_02_database.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -807,7 +807,7 @@ def test_users_last_finished_at(session_obj: sa.orm.sessionmaker) -> None:
807807
session.add(request_2)
808808
session.add(request_3)
809809
session.commit()
810-
users_last_finished_at = db.users_last_finished_at(session=session, max_time=now - datetime.timedelta(hours=24))
810+
users_last_finished_at = db.users_last_finished_at(session=session, after=now - datetime.timedelta(hours=24))
811811
assert finished_at == users_last_finished_at["user1"]
812812
assert "user2" not in users_last_finished_at
813813

tests/test_20_dispatcher.py

Lines changed: 0 additions & 61 deletions
Original file line numberDiff line numberDiff line change
@@ -127,64 +127,3 @@ def mock_get_tasks() -> dict[str, str]:
127127
# with pytest.raises(db.NoResultFound):
128128
# with session_obj() as session:
129129
# 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

Comments
 (0)