From 86ac8ac9664b4406a1aa51cd24381377156347a2 Mon Sep 17 00:00:00 2001 From: Munyoki Kilyungi Date: Wed, 25 Jun 2025 20:19:22 +0300 Subject: Store edit status in enum class. Signed-off-by: Munyoki Kilyungi --- tests/unit/db/test_case_attributes.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'tests/unit/db/test_case_attributes.py') diff --git a/tests/unit/db/test_case_attributes.py b/tests/unit/db/test_case_attributes.py index 097a96f..18de4d3 100644 --- a/tests/unit/db/test_case_attributes.py +++ b/tests/unit/db/test_case_attributes.py @@ -5,7 +5,10 @@ import tempfile import os from pytest_mock import MockFixture from gn3.db.case_attributes import queue_edit -from gn3.db.case_attributes import CaseAttributeEdit +from gn3.db.case_attributes import ( + CaseAttributeEdit, + EditStatus +) @pytest.mark.unit_test @@ -14,8 +17,9 @@ def test_queue_edit(mocker: MockFixture) -> None: with mock_conn.cursor() as cursor: type(cursor).lastrowid = 28 TMPDIR = os.environ.get("TMPDIR", tempfile.gettempdir()) - review_ids = queue_edit( + caseattr_id = queue_edit( cursor, + status=EditStatus.review directory=TMPDIR, edit=CaseAttributeEdit(inbredset_id=1, user_id="xxxx", changes={"a": 1, "b": 2})) cursor.execute.assert_called_once_with( -- cgit 1.4.1