feat: add challenge and red-blue competitions across API and web
This commit is contained in:
parent
f5161d9add
commit
8fd3c4bb64
77 changed files with 5355 additions and 24 deletions
34
api/tests/unit_tests/services/test_red_blue_service.py
Normal file
34
api/tests/unit_tests/services/test_red_blue_service.py
Normal file
|
|
@ -0,0 +1,34 @@
|
|||
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"
|
||||
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue