1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS.
4 # This program is free software: you can redistribute it and/or modify
5 # it under the terms of the GNU Affero General Public License as published by
6 # the Free Software Foundation, either version 3 of the License, or
7 # (at your option) any later version.
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU Affero General Public License for more details.
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
23 from paste
.deploy
import loadapp
24 from webtest
import TestApp
26 from mediagoblin
import mg_globals
27 from mediagoblin
.db
.models
import User
, MediaEntry
, Collection
, MediaComment
, \
28 CommentSubscription
, CommentNotification
, Privilege
, CommentReport
29 from mediagoblin
.tools
import testing
30 from mediagoblin
.init
.config
import read_mediagoblin_config
31 from mediagoblin
.db
.base
import Session
32 from mediagoblin
.meddleware
import BaseMeddleware
33 from mediagoblin
.auth
import gen_password_hash
34 from mediagoblin
.gmg_commands
.dbupdate
import run_dbupdate
36 from datetime
import datetime
39 MEDIAGOBLIN_TEST_DB_NAME
= u
'__mediagoblin_tests__'
40 TEST_SERVER_CONFIG
= pkg_resources
.resource_filename(
41 'mediagoblin.tests', 'test_paste.ini')
42 TEST_APP_CONFIG
= pkg_resources
.resource_filename(
43 'mediagoblin.tests', 'test_mgoblin_app.ini')
46 USER_DEV_DIRECTORIES_TO_SETUP
= ['media/public', 'media/queue']
49 class TestingMeddleware(BaseMeddleware
):
51 Meddleware for the Unit tests
53 It might make sense to perform some tests on all
54 requests/responses. Or prepare them in a special
55 manner. For example all html responses could be tested
56 for being valid html *after* being rendered.
58 This module is getting inserted at the front of the
59 meddleware list, which means: requests are handed here
60 first, responses last. So this wraps up the "normal"
63 If you need to add a test, either add it directly to
64 the appropiate process_request or process_response, or
65 create a new method and call it from process_*.
68 def process_response(self
, request
, response
):
69 # All following tests should be for html only!
70 if getattr(response
, 'content_type', None) != "text/html":
74 # If the template contains a reference to
75 # /mgoblin_static/ instead of using
76 # /request.staticdirect(), error out here.
77 # This could probably be implemented as a grep on
78 # the shipped templates easier...
79 if response
.text
.find("/mgoblin_static/") >= 0:
81 "Response HTML contains reference to /mgoblin_static/ "
82 "instead of staticdirect. Request was for: "
88 def get_app(request
, paste_config
=None, mgoblin_config
=None):
89 """Create a MediaGoblin app for testing.
92 - request: Not an http request, but a pytest fixture request. We
93 use this to make temporary directories that pytest
94 automatically cleans up as needed.
95 - paste_config: particular paste config used by this application.
96 - mgoblin_config: particular mediagoblin config used by this
99 paste_config
= paste_config
or TEST_SERVER_CONFIG
100 mgoblin_config
= mgoblin_config
or TEST_APP_CONFIG
102 # This is the directory we're copying the paste/mgoblin config stuff into
103 run_dir
= request
.config
._tmpdirhandler
.mktemp(
104 'mgoblin_app', numbered
=True)
105 user_dev_dir
= run_dir
.mkdir('user_dev').strpath
107 new_paste_config
= run_dir
.join('paste.ini').strpath
108 new_mgoblin_config
= run_dir
.join('mediagoblin.ini').strpath
109 shutil
.copyfile(paste_config
, new_paste_config
)
110 shutil
.copyfile(mgoblin_config
, new_mgoblin_config
)
115 # install user_dev directories
116 for directory
in USER_DEV_DIRECTORIES_TO_SETUP
:
117 full_dir
= os
.path
.join(user_dev_dir
, directory
)
118 os
.makedirs(full_dir
)
121 global_config
, validation_result
= read_mediagoblin_config(new_mgoblin_config
)
122 app_config
= global_config
['mediagoblin']
124 # Run database setup/migrations
125 run_dbupdate(app_config
, global_config
)
127 # setup app and return
129 'config:' + new_paste_config
)
131 # Insert the TestingMeddleware, which can do some
132 # sanity checks on every request/response.
133 # Doing it this way is probably not the cleanest way.
134 # We'll fix it, when we have plugins!
135 mg_globals
.app
.meddleware
.insert(0, TestingMeddleware(mg_globals
.app
))
137 app
= TestApp(test_app
)
141 def install_fixtures_simple(db
, fixtures
):
143 Very simply install fixtures in the database
145 for collection_name
, collection_fixtures
in fixtures
.iteritems():
146 collection
= db
[collection_name
]
147 for fixture
in collection_fixtures
:
148 collection
.insert(fixture
)
151 def assert_db_meets_expected(db
, expected
):
153 Assert a database contains the things we expect it to.
155 Objects are found via 'id', so you should make sure your document
159 - db: pymongo or mongokit database connection
160 - expected: the data we expect. Formatted like:
161 {'collection_name': [
163 'some_field': 'some_value'},]}
165 for collection_name
, collection_data
in expected
.iteritems():
166 collection
= db
[collection_name
]
167 for expected_document
in collection_data
:
168 document
= collection
.query
.filter_by(id=expected_document
['id']).first()
169 assert document
is not None # make sure it exists
170 assert document
== expected_document
# make sure it matches
173 def fixture_add_user(username
=u
'chris', password
=u
'toast',
174 privileges
=[], wants_comment_notification
=True):
175 # Reuse existing user or create a new one
176 test_user
= User
.query
.filter_by(username
=username
).first()
177 if test_user
is None:
179 test_user
.username
= username
180 test_user
.email
= username
+ u
'@example.com'
181 if password
is not None:
182 test_user
.pw_hash
= gen_password_hash(password
)
183 test_user
.wants_comment_notification
= wants_comment_notification
184 for privilege
in privileges
:
185 query
= Privilege
.query
.filter(Privilege
.privilege_name
==privilege
)
187 test_user
.all_privileges
.append(query
.one())
191 test_user
= User
.query
.filter_by(username
=username
).first()
193 # ... and detach from session:
194 Session
.expunge(test_user
)
199 def fixture_comment_subscription(entry
, notify
=True, send_email
=None):
200 if send_email
is None:
201 uploader
= User
.query
.filter_by(id=entry
.uploader
).first()
202 send_email
= uploader
.wants_comment_notification
204 cs
= CommentSubscription(
205 media_entry_id
=entry
.id,
206 user_id
=entry
.uploader
,
208 send_email
=send_email
)
212 cs
= CommentSubscription
.query
.filter_by(id=cs
.id).first()
219 def fixture_add_comment_notification(entry_id
, subject_id
, user_id
,
221 cn
= CommentNotification(user_id
=user_id
,
223 subject_id
=subject_id
)
226 cn
= CommentNotification
.query
.filter_by(id=cn
.id).first()
233 def fixture_media_entry(title
=u
"Some title", slug
=None,
234 uploader
=None, save
=True, gen_slug
=True,
235 state
=u
'unprocessed', fake_upload
=True,
238 Add a media entry for testing purposes.
240 Caution: if you're adding multiple entries with fake_upload=True,
241 make sure you save between them... otherwise you'll hit an
242 IntegrityError from multiple newly-added-MediaEntries adding
243 FileKeynames at once. :)
246 uploader
= fixture_add_user().id
251 entry
.uploader
= uploader
252 entry
.media_type
= u
'image'
256 entry
.media_files
= {'thumb': ['a', 'b', 'c.jpg'],
257 'medium': ['d', 'e', 'f.png'],
258 'original': ['g', 'h', 'i.png']}
259 entry
.media_type
= u
'mediagoblin.media_types.image'
262 entry
.generate_slug()
268 entry
= MediaEntry
.query
.filter_by(id=entry
.id).first()
270 Session
.expunge(entry
)
275 def fixture_add_collection(name
=u
"My first Collection", user
=None):
277 user
= fixture_add_user()
278 coll
= Collection
.query
.filter_by(creator
=user
.id, title
=name
).first()
282 coll
.creator
= user
.id
288 Session
.refresh(coll
)
290 # ... and detach from session:
291 Session
.expunge(coll
)
295 def fixture_add_comment(author
=None, media_entry
=None, comment
=None):
297 author
= fixture_add_user().id
299 if media_entry
is None:
300 media_entry
= fixture_media_entry().id
304 'Auto-generated test comment by user #{0} on media #{0}'.format(
307 comment
= MediaComment(author
=author
,
308 media_entry
=media_entry
,
313 Session
.expunge(comment
)
317 def fixture_add_comment_report(comment
=None, reported_user
=None,
318 reporter
=None, created
=None, report_content
=None):
320 comment
= fixture_add_comment()
322 if reported_user
is None:
323 reported_user
= fixture_add_user()
326 reporter
= fixture_add_user()
329 created
=datetime
.now()
331 if report_content
is None:
333 'Auto-generated test report'
335 comment_report
= CommentReport(comment
=comment
,
336 reported_user
= reported_user
,
339 report_content
=report_content
)
341 comment_report
.save()
343 Session
.expunge(comment_report
)
345 return comment_report