From 016fb730cbf524a0fe0d566f8f639f3f554eb188 Mon Sep 17 00:00:00 2001
From: over <overln2@gmail.com>
Date: Wed, 22 Jun 2016 17:03:55 +0600
Subject: [PATCH] implements tests for scheduler

---
 .travis.yml             |   2 +-
 tests/scheduler_test.py | 143 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 144 insertions(+), 1 deletion(-)
 create mode 100644 tests/scheduler_test.py

diff --git a/.travis.yml b/.travis.yml
index 36e6fcb89..271cdf4ee 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -16,6 +16,6 @@ before_script:
   - echo -e "[local]\nlocalhost ansible_connection=local" > ansible/localhost
 script:
   - find . -name \*.py -exec pep8 --ignore=E402,E501,E731 {} +
-  - nosetests -v -a '!fixme'
+  - nosetests -v -a '!fixme' --with-isolation
   - ansible-playbook --syntax-check -i ansible/localhost ansible/site.yml
 sudo: false
diff --git a/tests/scheduler_test.py b/tests/scheduler_test.py
new file mode 100644
index 000000000..a757b80cf
--- /dev/null
+++ b/tests/scheduler_test.py
@@ -0,0 +1,143 @@
+from datetime import datetime
+from datetime import timedelta
+import functools
+import unittest
+import viewer
+from lib import db
+from lib import assets_helper
+import settings
+import os
+
+asset_x = {
+    'mimetype': u'web',
+    'asset_id': u'4c8dbce552edb5812d3a866cfe5f159d',
+    'name': u'WireLoad',
+    'uri': u'http://www.wireload.net',
+    'start_date': datetime.now() - timedelta(days=3),
+    'end_date': datetime.now() + timedelta(days=3),
+    'duration': u'5',
+    'is_enabled': 1,
+    'nocache': 0,
+    'play_order': 1,
+}
+
+asset_x_diff = {
+    'duration': u'10'
+}
+
+asset_y = {
+    'mimetype': u'image',
+    'asset_id': u'7e978f8c1204a6f70770a1eb54a76e9b',
+    'name': u'Google',
+    'uri': u'https://www.google.com/images/srpr/logo3w.png',
+    'start_date': datetime.now() - timedelta(days=1),
+    'end_date': datetime.now() + timedelta(days=1),
+    'duration': u'6',
+    'is_enabled': 1,
+    'nocache': 0,
+    'play_order': 0,
+}
+
+asset_z = {
+    'mimetype': u'image',
+    'asset_id': u'7e978f8c1204a6f70770a1eb54a76e9c',
+    'name': u'Google',
+    'uri': u'https://www.google.com/images/srpr/logo3w.png',
+    'start_date': datetime.now() - timedelta(days=1),
+    'end_date': datetime.now() + timedelta(days=1),
+    'duration': u'6',
+    'is_enabled': 1,
+    'nocache': 0,
+    'play_order': 2,
+}
+
+FAKE_DB_PATH = '/tmp/fakedb'
+
+
+class FakeDatetime:
+    def __init__(self, need_time):
+        self.need_time = need_time
+
+    def utcnow(self):
+        return self.need_time
+
+    def now(self):
+        return self.need_time
+
+
+class SchedulerTest(unittest.TestCase):
+    def setUp(self):
+        self.old_db_path = settings.settings['database']
+        viewer.db_conn = db.conn(':memory:')
+        with db.commit(viewer.db_conn) as cursor:
+            cursor.execute(assets_helper.create_assets_table)
+
+    def tearDown(self):
+        settings.settings['database'] = self.old_db_path
+        settings.settings['shuffle_playlist'] = False
+        viewer.datetime, assets_helper.get_time = datetime, lambda: datetime.utcnow()
+        viewer.db_conn.close()
+        try:
+            os.remove(FAKE_DB_PATH)
+        except:
+            pass
+
+    def test_generate_asset_list_assets_should_be_y_and_x(self):
+        assets_helper.create_multiple(viewer.db_conn, [asset_x, asset_y])
+        assets, _ = viewer.generate_asset_list()
+        self.assertEqual(assets, [asset_y, asset_x])
+
+    def test_generate_asset_list_check_deadline(self):
+        assets_helper.create_multiple(viewer.db_conn, [asset_x, asset_y])
+        _, deadline = viewer.generate_asset_list()
+        self.assertEqual(deadline, asset_y['end_date'])
+
+    def test_get_next_asset_should_be_y_and_x(self):
+        assets_helper.create_multiple(viewer.db_conn, [asset_x, asset_y])
+        sch = viewer.Scheduler()
+
+        expect_y = sch.get_next_asset()
+        expect_x = sch.get_next_asset()
+
+        self.assertEqual([expect_y, expect_x], [asset_y, asset_x])
+
+    def test_keep_same_position_on_playlist_update(self):
+        assets_helper.create_multiple(viewer.db_conn, [asset_x, asset_y])
+        sch = viewer.Scheduler()
+
+        sch.get_next_asset()
+
+        assets_helper.create(viewer.db_conn, asset_z)
+        sch.update_playlist()
+        self.assertEqual(sch.index, 1)
+
+    def test_counter_should_increment_after_full_asset_loop(self):
+        settings.settings['shuffle_playlist'] = True
+        assets_helper.create_multiple(viewer.db_conn, [asset_x, asset_y])
+        sch = viewer.Scheduler()
+
+        self.assertEqual(sch.counter, 0)
+
+        sch.get_next_asset()
+        sch.get_next_asset()
+
+        self.assertEqual(sch.counter, 1)
+
+    def test_check_get_db_mtime(self):
+        settings.settings['database'] = FAKE_DB_PATH
+        with open(FAKE_DB_PATH, 'a'):
+            os.utime(FAKE_DB_PATH, (0, 0))
+
+        self.assertEqual(0, viewer.Scheduler().get_db_mtime())
+
+    def test_playlist_should_be_updated_after_deadline_reached(self):
+        assets_helper.create_multiple(viewer.db_conn, [asset_x, asset_y])
+        _, deadline = viewer.generate_asset_list()
+
+        fake = FakeDatetime(deadline + timedelta(seconds=1))
+        viewer.datetime, assets_helper.get_time = fake, lambda: fake.utcnow()
+
+        sch = viewer.Scheduler()
+        sch.refresh_playlist()
+
+        self.assertEqual([asset_x], sch.assets)