#!/usr/bin/env python

from __future__ import unicode_literals

# Allow direct execution
import os
import sys
import unittest
from time import sleep

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

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_last_read_on_write(self):
        t1 = self.archive._last_read
        self.archive.record_download("dl_id")
        t2 = self.archive._last_read
        self.assertNotEqual(t1, t2)

    def test_archive_last_read_on_read(self):
        t1 = self.archive._last_read
        self.archive.record_download("dl_id 1")
        t2 = self.archive._last_read
        self.assertNotEqual(t1, t2)

    def test_archive_file_not_changed(self):
        self.archive.record_download("dl_id")
        self.assertFalse(self.archive._file_changed())

    def test_archive_file_changed(self):
        self.archive.record_download("dl_id 1")
        with open(self.archive.filepath, "a", encoding="utf-8") as f_out:
            sleep(0.01)
            f_out.write("dl_id 2\n")
        self.assertTrue(self.archive._file_changed())

    def test_archive_file_changed_exists(self):
        self.archive.record_download("dl_id 1")
        with open(self.archive.filepath, "a", encoding="utf-8") as f_out:
            sleep(0.01)
            f_out.write("dl_id 2\n")
        self.assertTrue(self.archive._file_changed())
        self.assertFalse("dl_id 2" in self.archive._data)
        self.assertTrue("dl_id 2" 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()