Dot-Notation for Users.status

[mediagoblin.git] / mediagoblin / tests / test_auth.py

diff --git a/mediagoblin/tests/test_auth.py b/mediagoblin/tests/test_auth.py
index ad9a5bca9c4f5e0914cd0fa85b86981e0721b431..bd79a40750ad6c859dfde9672c79f3e902f00147 100644 (file)
--- a/mediagoblin/tests/test_auth.py
+++ b/mediagoblin/tests/test_auth.py
@@ -162,7 +162,7 @@ def test_register_views(test_app):
     new_user = mg_globals.database.User.find_one(
         {'username': 'happygirl'})
     assert new_user
-    assert new_user['status'] == u'needs_email_verification'
+    assert new_user.status == u'needs_email_verification'
     assert new_user.email_verified == False
 
     ## Make sure user is logged in
@@ -202,7 +202,7 @@ def test_register_views(test_app):
     new_user = mg_globals.database.User.find_one(
         {'username': 'happygirl'})
     assert new_user
-    assert new_user['status'] == u'needs_email_verification'
+    assert new_user.status == u'needs_email_verification'
     assert new_user.email_verified == False
 
     ## Verify the email activation works
@@ -216,7 +216,7 @@ def test_register_views(test_app):
     new_user = mg_globals.database.User.find_one(
         {'username': 'happygirl'})
     assert new_user
-    assert new_user['status'] == u'active'
+    assert new_user.status == u'active'
     assert new_user.email_verified == True
 
     # Uniqueness checks