MongoDB: fixed deprecation warnings

- collection.count() → either counting the result or using count_documents()
- collection.update() → replaced by update_one()
This commit is contained in:
2019-05-28 16:35:28 +02:00
parent 049d71dc77
commit 577bf8b964
3 changed files with 8 additions and 8 deletions

View File

@@ -52,7 +52,7 @@ class RandomFeaturedNodeTest(AbstractCloudTest):
with self.app.app_context():
proj_col = self.app.db('projects')
proj_col.update(
proj_col.update_one(
{'_id': self.pid},
{'$set': {
'nodes_featured': all_asset_ids,
@@ -99,7 +99,7 @@ class RandomFeaturedNodeTest(AbstractCloudTest):
# Featured but project is private, should be ignored
with self.app.app_context():
proj_col = self.app.db('projects')
proj_col.update(
proj_col.update_one(
{'_id': self.pid},
{'$set': {
'nodes_featured': [node_id],
@@ -110,7 +110,7 @@ class RandomFeaturedNodeTest(AbstractCloudTest):
# Featured but node is deleted, should be ignored
with self.app.app_context():
proj_col = self.app.db('projects')
proj_col.update(
proj_col.update_one(
{'_id': self.pid},
{'$set': {
'nodes_featured': [node_id],
@@ -118,7 +118,7 @@ class RandomFeaturedNodeTest(AbstractCloudTest):
}})
node_col = self.app.db('nodes')
node_col.update(
node_col.update_one(
{'_id': node_id},
{'$set': {
'_deleted': True,
@@ -149,7 +149,7 @@ class RandomFeaturedNodeTest(AbstractCloudTest):
with self.app.app_context():
proj_col = self.app.db('projects')
proj_col.update(
proj_col.update_one(
{'_id': self.pid},
{'$set': {
'nodes_featured': [node_id],

View File

@@ -21,7 +21,7 @@ class StatsTest(AbstractCloudTest):
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 asset node
self.assertEqual('asset', ctd.EXAMPLE_NODE['node_type'])