diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/unit/db/test_case_attributes.py | 8 |
1 files changed, 6 insertions, 2 deletions
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( |