54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
#!/usr/bin/env python
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
# Allow direct execution
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
# TODO refactor test to actually test YoutubeDL archive functionality
|
|
|
|
# from youtube_dl.archive import Archive
|
|
#
|
|
#
|
|
# class TestArchive(unittest.TestCase):
|
|
# def setUp(self):
|
|
# cur_dir = os.path.dirname(os.path.abspath(__file__))
|
|
# test_archive = os.path.join(cur_dir, "test_archive.txt")
|
|
# self.archive = Archive(test_archive)
|
|
#
|
|
# def tearDown(self):
|
|
# if os.path.exists(self.archive.filepath):
|
|
# os.remove(self.archive.filepath)
|
|
#
|
|
# def test_archive_disabled(self):
|
|
# self.assertTrue(Archive(None)._disabled)
|
|
# self.assertTrue(Archive("")._disabled)
|
|
# self.assertFalse(Archive("/anything")._disabled)
|
|
#
|
|
# def test_archive_read_empty_file(self):
|
|
# self.archive._read_file()
|
|
#
|
|
# def test_archive_exists(self):
|
|
# self.archive.record_download("dl_id")
|
|
# self.assertTrue("dl_id" in self.archive._data)
|
|
# self.assertTrue("dl_id" in self.archive)
|
|
#
|
|
# def test_archive_not_exists(self):
|
|
# self.assertFalse("dl_id" in self.archive)
|
|
#
|
|
# def test_archive_multiple_writes(self):
|
|
# self.archive.record_download("dl_id 1")
|
|
# self.archive.record_download("dl_id 2")
|
|
# self.archive.record_download("dl_id 3")
|
|
# expected = "dl_id 1" + "\n" + "dl_id 2" + "\n" + "dl_id 3" + "\n"
|
|
# with open(self.archive.filepath, "r", encoding="utf-8") as f_in:
|
|
# self.assertEqual(f_in.read(), expected)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|