castero/tests/test_queue.py

import os
from unittest import mock


from castero.config import Config
from castero.feed import Feed
from castero.queue import Queue
from castero.player import Player

my_dir = os.path.dirname(os.path.realpath(__file__))

feed = Feed(file=my_dir + "/feeds/valid_basic.xml")


def test_queue_init(display):
    myqueue = Queue(display)
    assert isinstance(myqueue, Queue)


def test_queue_first(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    assert myqueue.first == player1


def test_queue_get(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)
    player2 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.add(player2)
    retrieved_player1 = myqueue[0]
    assert retrieved_player1 == player1
    retrieved_player2 = myqueue[1]
    assert retrieved_player2 == player2


def test_queue_add(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    assert myqueue.length == 1


def test_queue_length(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)
    player2 = mock.MagicMock(spec=Player)
    player3 = mock.MagicMock(spec=Player)

    assert myqueue.length == 0
    myqueue.add(player1)
    assert myqueue.length == 1
    myqueue.add(player2)
    assert myqueue.length == 2
    myqueue.add(player3)
    assert myqueue.length == 3


def test_queue_clear(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)
    player2 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.add(player2)
    assert myqueue.length == 2
    myqueue.clear()
    assert myqueue.length == 0


def test_queue_remove(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)
    player2 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.add(player2)
    assert myqueue.length == 2
    removed_index = myqueue.remove(player1)
    assert removed_index == 0
    assert myqueue.length == 1
    assert myqueue.first == player2


def test_queue_next(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)
    player2 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.add(player2)
    assert myqueue.length == 2
    myqueue.next()
    assert myqueue.length == 1


def test_queue_jump(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)
    player2 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.add(player2)
    assert myqueue.length == 2
    myqueue.jump(player2)
    assert myqueue.length == 1


def test_queue_play(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)
    player1.episode.progress = None
    player1.episode.time = 100

    myqueue.add(player1)
    myqueue.play()
    assert player1.play.call_count == 1


def test_queue_play_progress(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    player1.time = 0
    player1.episode.progress = 1000
    myqueue.play()
    assert player1.play_from.call_count == 1


def test_queue_play_from_progress(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)
    player1.episode.progress = 1000

    myqueue.add(player1)
    myqueue._play_from_progress()

    assert player1.play_from.call_count == 1
    player1.play_from.assert_called_with(1)


def test_queue_play_from_progress_with_rewind(display):
    myqueue = Queue(display)
    myqueue._resume_rewind = 2
    player1 = mock.MagicMock(spec=Player)
    player1.episode.progress = 5000

    myqueue.add(player1)
    player1.state = 0  # set state to stopped
    myqueue.play()
    assert player1.play_from.call_count == 1
    player1.play_from.assert_called_with(3)


def test_queue_pause(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.pause()
    assert player1.pause.call_count == 1


def test_queue_stop(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.stop()
    assert player1.stop.call_count == 1


def test_queue_toggle(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)
    player1.episode.progress = None

    myqueue.add(player1)
    myqueue.toggle()
    assert player1.play.call_count == 1
    player1.state = 1
    myqueue.toggle()
    assert player1.pause.call_count == 1


def test_queue_seek_forward(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.seek(1)
    player1.seek.assert_called_with(1, int(Config["seek_distance_forward"]))


def test_queue_seek_backward(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.seek(-1)
    player1.seek.assert_called_with(-1, int(Config["seek_distance_backward"]))


def test_queue_change_rate(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.change_rate(-1, display)
    assert player1.set_rate.call_count == 1


def test_queue_change_volume(display):
    myqueue = Queue(display)
    player1 = mock.MagicMock(spec=Player)

    myqueue.add(player1)
    myqueue.change_volume(1)
    assert player1.set_volume.call_count == 1