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)