fix KeyError at source, not in test
parent
b063fe6c3f
commit
f2f02237a8
|
@ -697,14 +697,9 @@ class DownloadPageTest(TestCase):
|
|||
eb2.save()
|
||||
|
||||
anon_client = Client()
|
||||
try:
|
||||
response = anon_client.get("/work/%s/download/" % w.id, follow=True)
|
||||
except KeyError:
|
||||
# user agent testing in views.download fails with anon client,
|
||||
# but isn't relevant to this bit of functionality
|
||||
pass
|
||||
self.assertContains(response, "/download_ebook/%s/"% eb1.id, count=3) # no extra now, readmill not for google books?
|
||||
self.assertContains(response, "/download_ebook/%s/"% eb2.id, count=3)
|
||||
response = anon_client.get("/work/%s/download/" % w.id, follow=True)
|
||||
self.assertContains(response, "/download_ebook/%s/"% eb1.id, count=9)
|
||||
self.assertContains(response, "/download_ebook/%s/"% eb2.id, count=4)
|
||||
|
||||
|
||||
class LocaldatetimeTest(TestCase):
|
||||
|
|
|
@ -2428,7 +2428,7 @@ def download(request, work_id):
|
|||
readmill_epub_url = readmill_epub_ebook.url
|
||||
except:
|
||||
readmill_epub_url = None
|
||||
agent = request.META['HTTP_USER_AGENT']
|
||||
agent = request.META.get('HTTP_USER_AGENT','')
|
||||
iOS = 'iPad' in agent or 'iPhone' in agent or 'iPod' in agent
|
||||
iOS_app = iOS and not 'Safari' in agent
|
||||
android = 'Android' in agent
|
||||
|
|
Loading…
Reference in New Issue