From fce8e96979b1bdf4fb2357e463b79056b3a583ba Mon Sep 17 00:00:00 2001
From: Christopher Allan Webber <cwebber@dustycloud.org>
Date: Tue, 8 Oct 2013 11:36:18 -0500
Subject: [PATCH] Fixing issues in the tests caused by the
 OPW-Moderation-Update merge

Not that branch's fault!  Just both that branch and master had both done
a lot of changes to overlapping code.

self.test_user switched to self.our_user().  Updating everywhere.

This commit sponsored by Sam Kleinman.  Thank you! :)
---
 mediagoblin/tests/test_submission.py | 62 ++++++++--------------------
 1 file changed, 17 insertions(+), 45 deletions(-)

diff --git a/mediagoblin/tests/test_submission.py b/mediagoblin/tests/test_submission.py
index 5d42c5a5..b5b13ed3 100644
--- a/mediagoblin/tests/test_submission.py
+++ b/mediagoblin/tests/test_submission.py
@@ -119,24 +119,19 @@ class TestSubmission:
         self.test_app.get(url)
 
     def user_upload_limits(self, uploaded=None, upload_limit=None):
+        our_user = self.our_user()
+
         if uploaded:
-            self.test_user.uploaded = uploaded
+            our_user.uploaded = uploaded
         if upload_limit:
-            self.test_user.upload_limit = upload_limit
-
-        self.test_user.save()
+            our_user.upload_limit = upload_limit
 
-        # Reload
-        self.test_user = User.query.filter_by(
-            username=self.test_user.username
-        ).first()
-
-        # ... and detach from session:
-        Session.expunge(self.test_user)
+        our_user.save()
+        Session.expunge(our_user)
 
     def test_normal_jpg(self):
         # User uploaded should be 0
-        assert self.test_user.uploaded == 0
+        assert self.our_user().uploaded == 0
 
         self.check_normal_upload(u'Normal upload 1', GOOD_JPG)
 
@@ -145,10 +140,7 @@ class TestSubmission:
         file_size = float('{0:.2f}'.format(file_size))
 
         # Reload user
-        self.test_user = User.query.filter_by(
-            username=self.test_user.username
-        ).first()
-        assert self.test_user.uploaded == file_size
+        assert self.our_user().uploaded == file_size
 
     def test_normal_png(self):
         self.check_normal_upload(u'Normal upload 2', GOOD_PNG)
@@ -165,47 +157,37 @@ class TestSubmission:
         self.user_upload_limits(uploaded=500)
 
         # User uploaded should be 500
-        assert self.test_user.uploaded == 500
+        assert self.our_user().uploaded == 500
 
         response, context = self.do_post({'title': u'Normal upload 4'},
                                          do_follow=True,
                                          **self.upload_data(GOOD_JPG))
-        self.check_url(response, '/u/{0}/'.format(self.test_user.username))
+        self.check_url(response, '/u/{0}/'.format(self.our_user().username))
         assert 'mediagoblin/user_pages/user.html' in context
 
-        # Reload user
-        self.test_user = User.query.filter_by(
-            username=self.test_user.username
-        ).first()
-
         # Shouldn't have uploaded
-        assert self.test_user.uploaded == 500
+        assert self.our_user().uploaded == 500
 
     def test_user_upload_limit(self):
         self.user_upload_limits(uploaded=25, upload_limit=25)
 
         # User uploaded should be 25
-        assert self.test_user.uploaded == 25
+        assert self.our_user().uploaded == 25
 
         response, context = self.do_post({'title': u'Normal upload 5'},
                                          do_follow=True,
                                          **self.upload_data(GOOD_JPG))
-        self.check_url(response, '/u/{0}/'.format(self.test_user.username))
+        self.check_url(response, '/u/{0}/'.format(self.our_user().username))
         assert 'mediagoblin/user_pages/user.html' in context
 
-        # Reload user
-        self.test_user = User.query.filter_by(
-            username=self.test_user.username
-        ).first()
-
         # Shouldn't have uploaded
-        assert self.test_user.uploaded == 25
+        assert self.our_user().uploaded == 25
 
     def test_user_under_limit(self):
         self.user_upload_limits(uploaded=499)
 
         # User uploaded should be 499
-        assert self.test_user.uploaded == 499
+        assert self.our_user().uploaded == 499
 
         response, context = self.do_post({'title': u'Normal upload 6'},
                                          do_follow=False,
@@ -214,13 +196,8 @@ class TestSubmission:
         assert form.file.errors == [u'Sorry, uploading this file will put you'
                                     ' over your upload limit.']
 
-        # Reload user
-        self.test_user = User.query.filter_by(
-            username=self.test_user.username
-        ).first()
-
         # Shouldn't have uploaded
-        assert self.test_user.uploaded == 499
+        assert self.our_user().uploaded == 499
 
     def test_big_file(self):
         response, context = self.do_post({'title': u'Normal upload 7'},
@@ -309,13 +286,8 @@ class TestSubmission:
         self.check_media(request, {'id': media_id}, 0)
         self.check_comments(request, media_id, 0)
 
-        # Reload user
-        self.test_user = User.query.filter_by(
-            username = self.test_user.username
-        ).first()
-
         # Check that user.uploaded is the same as before the upload
-        assert self.test_user.uploaded == 50
+        assert self.our_user().uploaded == 50
 
     def test_evil_file(self):
         # Test non-suppoerted file with non-supported extension
-- 
2.25.1