Change document_class back to dict

cms/djangoapps/contentstore/tests/test_course_listing.py

@@ -265,7 +265,7 @@ class TestCourseListing(ModuleStoreTestCase):
         course_db_record = modulestore()._find_one(course.location)
         course_db_record.setdefault('metadata', {}).get('tabs', []).append({"type": "wiko", "name": "Wiki" })
         modulestore().collection.update(
-            {'_id': course_db_record['_id']},
+            {'_id': course.location.to_deprecated_son()},
             {'$set': {
                 'metadata.tabs': course_db_record['metadata']['tabs'],
             }},

common/djangoapps/student/tests/test_course_listing.py

@@ -107,7 +107,7 @@ class TestCourseListing(ModuleStoreTestCase):
         course_db_record = modulestore('direct')._find_one(course.location)
         course_db_record.setdefault('metadata', {}).get('tabs', []).append({"type": "wiko", "name": "Wiki" })
         modulestore('direct').collection.update(
-            {'_id': course_db_record['_id']},
+            {'_id': course.location.to_deprecated_son()},
             {'$set': {
                 'metadata.tabs': course_db_record['metadata']['tabs'],
             }},

common/lib/xmodule/xmodule/contentstore/mongo.py

@@ -29,7 +29,7 @@ class MongoContentStore(ContentStore):
             pymongo.MongoClient(
                 host=host,
                 port=port,
-                document_class=SON,
+                document_class=dict,
                 **kwargs
             ),
             db

common/lib/xmodule/xmodule/modulestore/mongo/base.py

@@ -295,8 +295,7 @@ class MongoModuleStore(ModuleStoreWriteBase):
                     host=host,
                     port=port,
                     tz_aware=tz_aware,
-                    # deserialize dicts as SONs
-                    document_class=SON,
+                    document_class=dict,
                     **kwargs
                 ),
                 db

common/lib/xmodule/xmodule/modulestore/tests/test_mixed_modulestore.py

@@ -99,7 +99,6 @@ class TestMixedModuleStore(LocMapperSetupSansDjango):
             host=self.HOST,
             port=self.PORT,
             tz_aware=True,
-            document_class=bson.son.SON,
         )
         self.connection.drop_database(self.DB)
         self.addCleanup(self.connection.drop_database, self.DB)

common/lib/xmodule/xmodule/modulestore/tests/test_mongo.py

@@ -66,7 +66,7 @@ class TestMongoModuleStore(unittest.TestCase):
             host=HOST,
             port=PORT,
             tz_aware=True,
-            document_class=bson.son.SON,
+            document_class=dict,
         )
         cls.connection.drop_database(DB)