34 lines
963 B
Python
34 lines
963 B
Python
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
|
|
from services.red_blue_service import RedBlueService
|
|
|
|
|
|
def test_submit_prompt_creates_submission(mocker):
|
|
session = mocker.MagicMock()
|
|
sub = RedBlueService.submit_prompt(
|
|
challenge_id="cid",
|
|
tenant_id="tid",
|
|
team="red",
|
|
prompt="attack",
|
|
account_id="aid",
|
|
end_user_id="eid",
|
|
session=session,
|
|
)
|
|
assert sub.team == "red"
|
|
session.add.assert_called_once()
|
|
session.commit.assert_called_once()
|
|
|
|
|
|
def test_select_counterparty_submission_latest_active(mocker):
|
|
c = SimpleNamespace(id="cid")
|
|
session = mocker.MagicMock()
|
|
qs = (
|
|
session.query.return_value.filter.return_value.order_by.return_value
|
|
)
|
|
qs.first.return_value = SimpleNamespace(id="subid", team="blue")
|
|
sub = RedBlueService.select_counterparty_submission(challenge=c, team="red", session=session)
|
|
assert sub.team == "blue"
|
|
|
|
|