diff --git a/tests/test_api.py b/tests/test_api.py index 273f689a..9453b21b 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -210,7 +210,7 @@ def test_raw_exceptions_possible_sizes_not_in_list(self, grep): target = b"size must be in one of these:" assert target in resp.data, f"{target!r} not in {resp.data!r}" - @patch("datagrepper.app.dm.Message.query", autospec=True) + @patch("datagrepper.app.dm.Message.query") def test_id(self, query): msg = query.filter_by.return_value.first.return_value msg.as_dict = MagicMock(return_value={"key": "value"})