44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
# Copyright 2024 (c) Anna Schumaker.
|
|
"""Tests our ListenBrainz priority queue."""
|
|
import emmental.listenbrainz.task
|
|
import unittest
|
|
|
|
|
|
class TestTaskQueue(unittest.TestCase):
|
|
"""Test the ListenBrainz queue."""
|
|
|
|
def setUp(self):
|
|
"""Set up common variables."""
|
|
self.queue = emmental.listenbrainz.task.Queue()
|
|
|
|
def test_init(self):
|
|
"""Test that the queue was set up properly."""
|
|
self.assertIsNotNone(self.queue)
|
|
|
|
def test_push_set_token(self):
|
|
"""Test calling push() with the 'set-token' operation."""
|
|
self.assertIsNone(self.queue._set_token)
|
|
|
|
self.queue.push("set-token", "abcde")
|
|
self.assertTupleEqual(self.queue._set_token, ("set-token", "abcde"))
|
|
self.queue.push("set-token", "fghij")
|
|
self.assertTupleEqual(self.queue._set_token, ("set-token", "fghij"))
|
|
|
|
self.assertTupleEqual(self.queue.pop(), ("set-token", "fghij"))
|
|
self.assertIsNone(self.queue._set_token)
|
|
|
|
self.queue.push("set-token", "abcde")
|
|
self.queue.clear("set-token")
|
|
self.assertIsNone(self.queue._set_token)
|
|
|
|
def test_push_clear_token(self):
|
|
"""Test calling push() with the 'clear-token' operation."""
|
|
self.queue.push("clear-token")
|
|
self.assertTupleEqual(self.queue._set_token, ("clear-token",))
|
|
self.assertTupleEqual(self.queue.pop(), ("clear-token",))
|
|
self.assertIsNone(self.queue._set_token)
|
|
|
|
self.queue.push("clear-token")
|
|
self.queue.clear("clear-token")
|
|
self.assertIsNone(self.queue._set_token)
|