def test_message_expires(self):
new = Message('abc')
new.save('123')
cache.clear()
new = Message('abc')
eq_(new.get(), None)
def test_message_expires(self):
new = Message("abc")
new.save("123")
cache.clear()
new = Message("abc")
eq_(new.get(), None)
def test_poll_failed(self):
msg = Message('file-viewer:%s' % self.file_viewer)
msg.save('I like cheese.')
res = self.client.get(self.poll_url())
eq_(res.status_code, 200)
data = json.loads(res.content)
eq_(data['status'], False)
eq_(data['msg'], ['I like cheese.'])
def test_poll_failed(self):
msg = Message('file-viewer:%s' % self.file_viewer)
msg.save('I like cheese.')
res = self.client.get(self.poll_url())
eq_(res.status_code, 200)
data = json.loads(res.content)
eq_(data['status'], False)
eq_(data['msg'], ['I like cheese.'])
def test_message_get_delete(self):
new = Message('abc')
new.save('123')
new = Message('abc')
eq_(new.get(delete=False), '123')
eq_(new.get(delete=True), '123')
eq_(new.get(), None)
def compare_poll(request, diff):
msgs = []
for f in (diff.left, diff.right):
m = Message('file-viewer:%s' % f).get(delete=True)
if m:
msgs.append(m)
return {'status': diff.is_extracted(), 'msg': msgs}
def test_guard_copes(self):
try:
with guard('abc'):
1 / 0
except ZeroDivisionError:
pass
eq_(Message('abc').get(), None)
def test_message_expires(self):
new = Message('abc')
new.save('123')
cache.clear()
new = Message('abc')
eq_(new.get(), None)
def test_message_get_delete(self):
new = Message("abc")
new.save("123")
new = Message("abc")
eq_(new.get(delete=False), "123")
eq_(new.get(delete=True), "123")
eq_(new.get(), None)
def test_message_get_delete(self):
new = Message('abc')
new.save('123')
new = Message('abc')
eq_(new.get(delete=False), '123')
eq_(new.get(delete=True), '123')
eq_(new.get(), None)
def extract_file(viewer, **kw):
# This message is for end users so they'll see a nice error.
msg = Message('file-viewer:%s' % viewer)
msg.delete()
# This flag is so that we can signal when the extraction is completed.
flag = Message(viewer._extraction_cache_key())
task_log.debug('[1@%s] Unzipping %s for file viewer.' % (
extract_file.rate_limit, viewer))
try:
flag.save('extracting') # Set the flag to a truthy value.
viewer.extract()
except Exception, err:
if settings.DEBUG:
msg.save(_('There was an error accessing file %s. %s.') %
(viewer, err))
else:
msg.save(_('There was an error accessing file %s.') % viewer)
task_log.error('[1@%s] Error unzipping: %s' %
(extract_file.rate_limit, err))
def extract_file(viewer, **kw):
# This message is for end users so they'll see a nice error.
msg = Message('file-viewer:%s' % viewer)
msg.delete()
# This flag is so that we can signal when the extraction is completed.
flag = Message(viewer._extraction_cache_key())
task_log.debug('[1@%s] Unzipping %s for file viewer.' %
(extract_file.rate_limit, viewer))
try:
viewer.extract()
except Exception, err:
if settings.DEBUG:
msg.save(
_('There was an error accessing file %s. %s.') % (viewer, err))
else:
msg.save(_('There was an error accessing file %s.') % viewer)
task_log.error('[1@%s] Error unzipping: %s' %
(extract_file.rate_limit, err))
def extract_file(viewer, **kw):
msg = Message('file-viewer:%s' % viewer)
msg.delete()
task_log.info('[1@%s] Unzipping %s for file viewer.' %
(extract_file.rate_limit, viewer))
try:
viewer.extract()
except Exception, err:
if settings.DEBUG:
msg.save(
_('There was an error accessing file %s. %s.') % (viewer, err))
else:
msg.save(_('There was an error accessing file %s.') % viewer)
task_log.error('[1@%s] Error unzipping: %s' %
(extract_file.rate_limit, err))
def extract_file(viewer, **kw):
msg = Message('file-viewer:%s' % viewer)
msg.delete()
task_log.info('[1@%s] Unzipping %s for file viewer.' % (
extract_file.rate_limit, viewer))
try:
viewer.extract()
except Exception, err:
if settings.DEBUG:
msg.save(_('There was an error accessing file %s. %s.') %
(viewer, err))
else:
msg.save(_('There was an error accessing file %s.') % viewer)
task_log.error('[1@%s] Error unzipping: %s' %
(extract_file.rate_limit, err))
def test_guard_blocks(self):
Message('abc').save(True)
with guard('abc') as locked:
eq_(locked, True)
def test_guard_deletes(self):
with guard('abc'):
pass
eq_(Message('abc').get(), None)
def test_watermark_locked(self):
dest = self.file.watermarked_file_path(self.user.pk)
msg = Message('marketplace.watermark.%s' % dest)
msg.save(True)
res = self.client.get(self.url)
eq_(res.status_code, 404)
def test_guard(self):
with guard('abc') as locked:
eq_(locked, False)
eq_(Message('abc').get(), True)
def poll(request, viewer):
return {
'status': viewer.is_extracted(),
'msg': [Message('file-viewer:%s' % viewer).get(delete=True)]
}
def test_message_save(self):
new = Message('abc')
new.save('123')
new = Message('abc')
eq_(new.get(), '123')
def test_message_save(self):
new = Message('abc')
new.save('123')
new = Message('abc')
eq_(new.get(), '123')
def is_extracted(self):
"""If the file has been extracted or not."""
return (os.path.exists(self.dest) and not
Message(self._extraction_cache_key()).get())
def test_watermark_locked(self):
dest = self.file.watermarked_file_path(self.user.pk)
msg = Message('marketplace.watermark.%s' % dest)
msg.save(True)
res = self.client.get(self.url)
eq_(res.status_code, 404)
def test_message_save(self):
new = Message("abc")
new.save("123")
new = Message("abc")
eq_(new.get(), "123")
