Refactoring to make testing more stable

django_q/tests/__init__.py

@@ -1 +0,0 @@
-__author__ = 'ilan'

django_q/tests/settings.py

@@ -103,3 +103,5 @@ USE_TZ = True
 
 STATIC_URL = '/static/'
 
+# Django Q specific
+#Q_PREFIX = 'test_django_q'

django_q/tests/test_q.py

@@ -0,0 +1,130 @@
+import sys
+import os
+from time import sleep
+from multiprocessing import Queue, Event
+
+import pytest
+
+myPath = os.path.dirname(os.path.abspath(__file__))
+sys.path.insert(0, myPath + '/../')
+
+from django_q.core import Cluster, r, async, Q_LIST, pusher, worker, monitor
+from django_q.humanhash import DEFAULT_WORDLIST
+from django_q import result
+
+
+class WordClass(object):
+    def __init__(self):
+        self.word_list = DEFAULT_WORDLIST
+
+    def get_words(self):
+        return self.word_list
+
+
+def test_admin_view(admin_client):
+    response = admin_client.get('/admin/django_q/')
+    assert response.status_code == 200
+    response = admin_client.get('/admin/django_q/failure/')
+    assert response.status_code == 200
+    response = admin_client.get('/admin/django_q/success/')
+    assert response.status_code == 200
+
+
+def test_redis_connection():
+    assert r.ping() is True
+
+
+def test_cluster_initial():
+    c = Cluster()
+    assert c.sentinel is None
+    assert c.is_idle
+    c.start()
+    while c.is_starting:
+        sleep(0.2)
+    assert c.sentinel.is_alive() is True
+    assert c.is_running
+    c.stop()
+    while c.is_stopping:
+        sleep(0.2)
+    assert c.sentinel.is_alive() is False
+    assert c.has_stopped
+
+
+@pytest.mark.django_db
+def test_cluster():
+    task = async('django_q.tests.tasks.count_letters', DEFAULT_WORDLIST)
+    task_count = r.llen(Q_LIST)
+    assert task_count >= 1
+    task_queue = Queue()
+    assert task_queue.qsize() == 0
+    result_queue = Queue()
+    assert result_queue.qsize() == 0
+    event = Event()
+    event.set()
+    # Test push
+    pusher(task_queue, event)
+    assert task_queue.qsize() == 1
+    assert r.llen(Q_LIST) == task_count - 1
+    # Test work
+    task_queue.put('STOP')
+    worker(task_queue, result_queue)
+    assert task_queue.qsize() == 0
+    assert result_queue.qsize() == 1
+    # Test monitor
+    result_queue.put('STOP')
+    monitor(result_queue)
+    assert result_queue.qsize() == 0
+
+    # check result
+    assert result(task) == 1506
+
+
+@pytest.mark.django_db
+def test_async():
+    a = async('django_q.tests.tasks.count_letters', DEFAULT_WORDLIST, hook='django_q.tests.test_q.assert_result')
+    b = async('django_q.tests.tasks.count_letters2', WordClass(), hook='django_q.tests.test_q.assert_result')
+    c = Cluster()
+    assert c.start() > 0
+    while not c.is_running:
+        sleep(0.5)
+    assert isinstance(a, str)
+    assert isinstance(b, str)
+    while c.stat.task_q_size > 0 and c.stat.done_q_size > 0:
+        sleep(0.5)
+    assert c.stop() is True
+    result_a = result(a)
+    assert result_a is not None
+    assert result_a.success is True
+    assert result_a.result == 1506
+    result_b = result(b)
+    assert result_b is not None
+    assert result_b.success is True
+    assert result_b.result == 1506
+
+
+
+@pytest.mark.django_db
+def assert_result(task):
+    assert task is not None
+    assert task.success is True
+    assert task.result == 1506
+
+
+@pytest.mark.django_db
+def broken_package():
+    a = async('django_q.tests.tasks.count_letters', DEFAULT_WORDLIST, 'oneargumentoomany',
+              hook='django_q.tests.test_q.assert_bad_result')
+    b = async('django_q.tests.tasks.does_not_exist', WordClass(), hook='django_q.tests.test_q.assert_bad_result')
+    assert isinstance(a, str)
+    assert isinstance(b, str)
+    sleep(5)
+    result_a = result(a)
+    assert result_a.success is False
+    result_b = result(b)
+    assert result_b.success is False
+
+
+@pytest.mark.django_db
+def assert_bad_result(task):
+    assert task is not None
+    assert task.success is False