You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.5 KiB
59 lines
1.5 KiB
#!/usr/bin/env python |
|
# coding: utf-8 |
|
|
|
from __future__ import unicode_literals |
|
|
|
import shutil |
|
|
|
# Allow direct execution |
|
import os |
|
import sys |
|
import unittest |
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
|
|
|
|
|
from test.helper import FakeYDL |
|
from youtube_dl.cache import Cache |
|
|
|
|
|
def _is_empty(d): |
|
return not bool(os.listdir(d)) |
|
|
|
|
|
def _mkdir(d): |
|
if not os.path.exists(d): |
|
os.mkdir(d) |
|
|
|
|
|
class TestCache(unittest.TestCase): |
|
def setUp(self): |
|
TEST_DIR = os.path.dirname(os.path.abspath(__file__)) |
|
TESTDATA_DIR = os.path.join(TEST_DIR, 'testdata') |
|
_mkdir(TESTDATA_DIR) |
|
self.test_dir = os.path.join(TESTDATA_DIR, 'cache_test') |
|
self.tearDown() |
|
|
|
def tearDown(self): |
|
if os.path.exists(self.test_dir): |
|
shutil.rmtree(self.test_dir) |
|
|
|
def test_cache(self): |
|
ydl = FakeYDL({ |
|
'cachedir': self.test_dir, |
|
}) |
|
c = Cache(ydl) |
|
obj = {'x': 1, 'y': ['ä', '\\a', True]} |
|
self.assertEqual(c.load('test_cache', 'k.'), None) |
|
c.store('test_cache', 'k.', obj) |
|
self.assertEqual(c.load('test_cache', 'k2'), None) |
|
self.assertFalse(_is_empty(self.test_dir)) |
|
self.assertEqual(c.load('test_cache', 'k.'), obj) |
|
self.assertEqual(c.load('test_cache', 'y'), None) |
|
self.assertEqual(c.load('test_cache2', 'k.'), None) |
|
c.remove() |
|
self.assertFalse(os.path.exists(self.test_dir)) |
|
self.assertEqual(c.load('test_cache', 'k.'), None) |
|
|
|
|
|
if __name__ == '__main__': |
|
unittest.main()
|
|
|