import os
import sys
import unittest
from traceback import print_exc
from time import sleep
from threading import Thread, currentThread

if os.path.exists('test_TimedTaskQueue.py'):
    BASE_DIR = '..'
elif os.path.exists('tribler.py'):
    BASE_DIR = '.'
sys.path.insert(1, os.path.abspath(BASE_DIR))
    
from Tribler.Overlay.TimedTaskQueue import TimedTaskQueue

class TestTimedTaskQueue(unittest.TestCase):
    
    def setUp(self):
        pass
        
    def tearDown(self):
        pass
    
    def test_addTask(self):
        self.queue = TimedTaskQueue()
        self.queue.register()
        self.count = 0
        self.queue.add_task(self.task3a, 3)
        self.queue.add_task(self.task0, 0)
        self.queue.add_task(self.task3b, 3)
        self.queue.add_task(self.task2, 1)
        sleep(6)
        assert self.count == 11
        del self.queue
        
    def task0(self):
        self.count += 1
        assert self.count == 1
    
    def task2(self):
        self.count += 2
        assert self.count == 3
    
    def task3a(self):
        self.count += 4
        assert self.count == 7 or self.count == 11
    
    def task3b(self):
        self.count += 4
        assert self.count == 7 or self.count == 11
    
def test_suite():
    suite = unittest.TestSuite()
    suite.addTest(unittest.makeSuite(TestTimedTaskQueue))
    
    return suite
        
def main():
    unittest.main(defaultTest='test_suite')

    
if __name__ == '__main__':
    main()