diff --git a/tests/unit/test_oneclick_gen_p0_fixes.py b/tests/unit/test_oneclick_gen_p0_fixes.py index 7bfd58ff0..efde21b16 100644 --- a/tests/unit/test_oneclick_gen_p0_fixes.py +++ b/tests/unit/test_oneclick_gen_p0_fixes.py @@ -132,10 +132,8 @@ class TestDownloadLibraryAssets: session.query.return_value = query filter_result = MagicMock() query.filter.return_value = filter_result - in_filter = MagicMock() - filter_result.filter.return_value = in_filter id_filter = MagicMock() - in_filter.filter.return_value = id_filter + filter_result.filter.return_value = id_filter assets = [self._make_asset("a1", "video/a1.mp4")] id_filter.order_by.return_value.all.return_value = assets