mirror of
https://github.com/Screenly/Anthias.git
synced 2026-01-24 14:08:28 -05:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
from __future__ import unicode_literals
|
|
from os import getenv, path, listdir, system
|
|
import unittest
|
|
|
|
from celery_tasks import celery as celeryapp
|
|
from celery_tasks import cleanup
|
|
|
|
|
|
class CeleryTasksTestCase(unittest.TestCase):
|
|
REPO_URL = 'https://github.com/Screenly/screenly-ose'
|
|
|
|
def setUp(self):
|
|
self.image_url = f'{self.REPO_URL}/raw/master/static/img/standby.png'
|
|
celeryapp.conf.update(
|
|
CELERY_ALWAYS_EAGER=True,
|
|
CELERY_RESULT_BACKEND='',
|
|
CELERY_BROKER_URL='')
|
|
|
|
def download_image(self, image_url, image_path):
|
|
system('curl {} > {}'.format(image_url, image_path))
|
|
|
|
|
|
class TestCleanup(CeleryTasksTestCase):
|
|
def setUp(self):
|
|
super(TestCleanup, self).setUp()
|
|
self.assets_path = path.join(getenv('HOME'), 'screenly_assets')
|
|
self.image_path = path.join(self.assets_path, 'image.tmp')
|
|
|
|
def test_cleanup(self):
|
|
cleanup.apply()
|
|
tmp_files = [
|
|
x for x in listdir(self.assets_path) if x.endswith('.tmp')]
|
|
self.assertEqual(len(tmp_files), 0)
|
|
|
|
def tearDown(self):
|
|
self.download_image(self.image_url, self.image_path)
|