diff --git a/pillar/api/blender_cloud/texture_libs.py b/pillar/api/blender_cloud/texture_libs.py index 34ce67f9..be193300 100644 --- a/pillar/api/blender_cloud/texture_libs.py +++ b/pillar/api/blender_cloud/texture_libs.py @@ -104,7 +104,7 @@ def has_texture_node(proj, return_hdri=True): if return_hdri: node_types.append('group_hdri') - count = nodes_collection.count( + count = nodes_collection.count_documents( {'node_type': {'$in': node_types}, 'project': proj['_id'], 'parent': None}) diff --git a/tests/test_orphan_files.py b/tests/test_orphan_files.py index 09d31a07..e3c6d165 100644 --- a/tests/test_orphan_files.py +++ b/tests/test_orphan_files.py @@ -19,7 +19,7 @@ class OrphanFilesTest(AbstractPillarTest): 24 * 'c', project_overrides={'_id': ObjectId(), 'is_private': True}) private2, _ = self.create_project_with_admin( 24 * 'd', project_overrides={'_id': ObjectId(), 'is_private': None}) - self.assertEqual(4, self.app.db('projects').count()) + self.assertEqual(4, self.app.db('projects').count_documents({})) # Create files, some orphan and some used. project_ids = (public1, public2, private1, private2)