Code review updates

Author: aclark4life

django_mongodb_backend/fields/encryption.py

@@ -99,11 +99,11 @@ class EncryptedSmallIntegerField(EncryptedFieldMixin, models.SmallIntegerField):
     pass
 
 
-class EncryptedTimeField(EncryptedFieldMixin, models.TimeField):
+class EncryptedTextField(EncryptedFieldMixin, models.TextField):
     pass
 
 
-class EncryptedTextField(EncryptedFieldMixin, models.TextField):
+class EncryptedTimeField(EncryptedFieldMixin, models.TimeField):
     pass
 
 

tests/encryption_/test_fields.py

@@ -57,9 +57,10 @@ def setUp(self):
         )
 
     def test_array(self):
-        self.assertEqual(len(self.movie.cast), 2)
-        self.assertEqual(self.movie.cast[0].name, "Actor One")
-        self.assertEqual(self.movie.cast[1].name, "Actor Two")
+        movie = Movie.objects.get(id=self.movie.id)
+        self.assertEqual(len(movie.cast), 2)
+        self.assertEqual(movie.cast[0].name, "Actor One")
+        self.assertEqual(movie.cast[1].name, "Actor Two")
 
         # ======================================================================
         # ERROR: test_array (encryption_.test_fields.EmbeddedModelArrayTests.test_array)