use both pid and cluster_id

This commit is contained in:
Jason McVetta
2020-02-12 18:23:47 +07:00
parent d22c916bfc
commit 14a602ee8b
4 changed files with 13 additions and 10 deletions
+1 -1
View File
@@ -345,7 +345,7 @@ def test_bad_secret(broker, monkeypatch):
monkeypatch.setattr(Conf, "SECRET_KEY", "OOPS")
stat = Stat.get_all()
assert len(stat) == 0
assert Stat.get(s.parent_pid) is None
assert Stat.get(pid=s.parent_pid, cluster_id=cluster_id) is None
task_queue = Queue()
pusher(task_queue, stop_event, broker=broker)
result_queue = Queue()
+4 -2
View File
@@ -1,4 +1,5 @@
import pytest
import uuid
from django_q.tasks import async_task
from django_q.brokers import get_broker
@@ -10,7 +11,8 @@ from django_q.conf import Conf
@pytest.mark.django_db
def test_monitor(monkeypatch):
assert Stat.get(0).sentinel == 0
cluster_id = uuid.uuid4()
assert Stat.get(pid=0, cluster_id=4).sentinel == 0
c = Cluster()
c.start()
stats = monitor(run_once=True)
@@ -18,7 +20,7 @@ def test_monitor(monkeypatch):
assert len(stats) > 0
found_c = False
for stat in stats:
if stat.cluster_id == c.pid:
if stat.cluster_id == c.cluster_id:
found_c = True
assert stat.uptime() > 0
assert stat.empty_queues() is True