Merge pull request #1599 from jackwilsdon/add-plexupdate-library-test

Add a test for PlexUpdate custom library name
This commit is contained in:
Adrian Sampson 2015-09-09 21:50:58 -07:00
commit 2f5bec2913

View file

@ -8,9 +8,12 @@ import responses
class PlexUpdateTest(unittest.TestCase, TestHelper):
def add_response_get_music_section(self):
def add_response_get_music_section(self, section_name='Music'):
"""Create response for mocking the get_music_section function.
"""
escaped_section_name = section_name.replace('"', '\\"')
body = (
'<?xml version="1.0" encoding="UTF-8"?>'
'<MediaContainer size="3" allowSync="0" '
@ -28,7 +31,7 @@ class PlexUpdateTest(unittest.TestCase, TestHelper):
'</Directory>'
'<Directory allowSync="0" art="/:/resources/artist-fanart.jpg" '
'filters="1" refreshing="0" thumb="/:/resources/artist.png" '
'key="2" type="artist" title="Music" '
'key="2" type="artist" title="' + escaped_section_name + '" '
'composite="/library/sections/2/composite/1416929243" '
'agent="com.plexapp.agents.lastfm" scanner="Plex Music Scanner" '
'language="en" uuid="90897c95-b3bd-4778-a9c8-1f43cb78f047" '
@ -91,6 +94,17 @@ class PlexUpdateTest(unittest.TestCase, TestHelper):
self.config['plex']['token'],
self.config['plex']['library_name'].get()), '2')
@responses.activate
def test_get_named_music_section(self):
# Adding response.
self.add_response_get_music_section('My Music Library')
self.assertEqual(get_music_section(
self.config['plex']['host'],
self.config['plex']['port'],
self.config['plex']['token'],
'My Music Library'), '2')
@responses.activate
def test_update_plex(self):
# Adding responses.