beets/test/test_play.py
2016-01-03 16:50:17 -08:00

66 lines
2.2 KiB
Python

# -*- coding: utf-8 -*-
"""Tests for the play plugin"""
from __future__ import (division, absolute_import, print_function,
unicode_literals)
from mock import patch, ANY
from test._common import unittest
from test.helper import TestHelper
class PlayPluginTest(unittest.TestCase, TestHelper):
def setUp(self):
self.setup_beets()
self.load_plugins('play')
self.item = self.add_item(title='aNiceTitle')
def tearDown(self):
self.teardown_beets()
self.unload_plugins()
@patch('beetsplug.play.util.interactive_open')
def test_basic(self, open_mock):
self.run_command('play', 'title:aNiceTitle')
open_mock.assert_called_once_with(ANY, None)
self.assertPlaylistCorrect(open_mock)
@patch('beetsplug.play.util.interactive_open')
def test_args_option(self, open_mock):
self.config['play']['command'] = 'echo'
self.run_command('play', '-A', 'foo', 'title:aNiceTitle')
open_mock.assert_called_once_with(ANY, 'echo foo')
self.assertPlaylistCorrect(open_mock)
@patch('beetsplug.play.util.interactive_open')
def test_args_option_in_middle(self, open_mock):
self.config['play']['command'] = 'echo $args other'
self.run_command('play', '-A', 'foo', 'title:aNiceTitle')
open_mock.assert_called_once_with(ANY, 'echo foo other')
self.assertPlaylistCorrect(open_mock)
@patch('beetsplug.play.util.interactive_open')
def test_relative_to(self, open_mock):
self.config['play']['command'] = 'echo'
self.config['play']['relative_to'] = '/something'
self.run_command('play', 'title:aNiceTitle')
open_mock.assert_called_once_with(ANY, 'echo')
self.assertPlaylistCorrect(open_mock, '..{}\n')
def assertPlaylistCorrect(self, open_mock, expected='{}\n'):
playlist = open(open_mock.call_args[0][0][0], 'r')
self.assertEqual(expected.format(self.item.path.decode('utf-8')),
playlist.read().decode('utf-8'))
def suite():
return unittest.TestLoader().loadTestsFromName(__name__)
if __name__ == b'__main__':
unittest.main(defaultTest='suite')