1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
|
import unittest
import os
import sys
# Insert .. at the beginning of path so we use this version instead
# of something that's already been installed
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
import musicbrainzngs
from musicbrainzngs import mbxml
from test import _common
DATA_DIR = os.path.join(os.path.dirname(__file__), "data")
class UrlTest(unittest.TestCase):
""" Test that the correct URL is generated when a search query is made """
def setUp(self):
self.opener = _common.FakeOpener("<response/>")
musicbrainzngs.compat.build_opener = lambda *args: self.opener
musicbrainzngs.set_useragent("a", "1")
musicbrainzngs.set_rate_limit(False)
def testSearchArtist(self):
musicbrainzngs.search_artists("Dynamo Go")
self.assertEqual("http://musicbrainz.org/ws/2/artist/?query=Dynamo+Go", self.opener.get_url())
def testSearchEvent(self):
musicbrainzngs.search_events("woodstock")
self.assertEqual("http://musicbrainz.org/ws/2/event/?query=woodstock", self.opener.get_url())
def testSearchLabel(self):
musicbrainzngs.search_labels("Waysafe")
self.assertEqual("http://musicbrainz.org/ws/2/label/?query=Waysafe", self.opener.get_url())
def testSearchPlace(self):
musicbrainzngs.search_places("Fillmore")
self.assertEqual("http://musicbrainz.org/ws/2/place/?query=Fillmore", self.opener.get_url())
def testSearchRelease(self):
musicbrainzngs.search_releases("Affordable Pop Music")
self.assertEqual("http://musicbrainz.org/ws/2/release/?query=Affordable+Pop+Music", self.opener.get_url())
def testSearchReleaseGroup(self):
musicbrainzngs.search_release_groups("Affordable Pop Music")
self.assertEqual("http://musicbrainz.org/ws/2/release-group/?query=Affordable+Pop+Music", self.opener.get_url())
def testSearchRecording(self):
musicbrainzngs.search_recordings("Thief of Hearts")
self.assertEqual("http://musicbrainz.org/ws/2/recording/?query=Thief+of+Hearts", self.opener.get_url())
def testSearchWork(self):
musicbrainzngs.search_works("Fountain City")
self.assertEqual("http://musicbrainz.org/ws/2/work/?query=Fountain+City", self.opener.get_url())
class SearchArtistTest(unittest.TestCase):
def testFields(self):
fn = os.path.join(DATA_DIR, "search-artist.xml")
with open(fn, 'rb') as msg:
res = mbxml.parse_message(msg)
self.assertEqual(25, len(res["artist-list"]))
self.assertEqual(349, res["artist-count"])
one = res["artist-list"][0]
self.assertEqual(9, len(one.keys()))
# Score is a key that is only in search results -
# so check for it here
self.assertEqual("100", one["ext:score"])
class SearchReleaseTest(unittest.TestCase):
def testFields(self):
fn = os.path.join(DATA_DIR, "search-release.xml")
with open(fn, 'rb') as msg:
res = mbxml.parse_message(msg)
self.assertEqual(25, len(res["release-list"]))
self.assertEqual(16739, res["release-count"])
one = res["release-list"][0]
self.assertEqual("100", one["ext:score"])
# search results have a medium-list/track-count element
self.assertEqual(4, one["medium-track-count"])
self.assertEqual(1, one["medium-count"])
self.assertEqual("CD", one["medium-list"][0]["format"])
class SearchReleaseGroupTest(unittest.TestCase):
def testFields(self):
fn = os.path.join(DATA_DIR, "search-release-group.xml")
with open(fn, 'rb') as msg:
res = mbxml.parse_message(msg)
self.assertEqual(25, len(res["release-group-list"]))
self.assertEqual(14641, res["release-group-count"])
one = res["release-group-list"][0]
self.assertEqual("100", one["ext:score"])
class SearchWorkTest(unittest.TestCase):
def testFields(self):
fn = os.path.join(DATA_DIR, "search-work.xml")
with open(fn, 'rb') as msg:
res = mbxml.parse_message(msg)
self.assertEqual(25, len(res["work-list"]))
self.assertEqual(174, res["work-count"])
one = res["work-list"][0]
self.assertEqual("100", one["ext:score"])
class SearchLabelTest(unittest.TestCase):
def testFields(self):
fn = os.path.join(DATA_DIR, "search-label.xml")
with open(fn, 'rb') as msg:
res = mbxml.parse_message(msg)
self.assertEqual(1, len(res["label-list"]))
self.assertEqual(1, res["label-count"])
one = res["label-list"][0]
self.assertEqual("100", one["ext:score"])
class SearchRecordingTest(unittest.TestCase):
def testFields(self):
fn = os.path.join(DATA_DIR, "search-recording.xml")
with open(fn, 'rb') as msg:
res = mbxml.parse_message(msg)
self.assertEqual(25, len(res["recording-list"]))
self.assertEqual(1258, res["recording-count"])
one = res["recording-list"][0]
self.assertEqual("100", one["ext:score"])
class SearchInstrumentTest(unittest.TestCase):
def testFields(self):
fn = os.path.join(DATA_DIR, "search-instrument.xml")
with open(fn, 'rb') as msg:
res = mbxml.parse_message(msg)
self.assertEqual(23, len(res["instrument-list"]))
self.assertEqual(23, res["instrument-count"])
one = res["instrument-list"][0]
self.assertEqual("100", one["ext:score"])
end = res["instrument-list"][-1]
self.assertEqual("29", end["ext:score"])
class SearchPlaceTest(unittest.TestCase):
def testFields(self):
fn = os.path.join(DATA_DIR, "search-place.xml")
with open(fn, 'rb') as msg:
res = mbxml.parse_message(msg)
self.assertEqual(14, res["place-count"])
self.assertEqual(14, len(res["place-list"]))
one = res["place-list"][0]
self.assertEqual("100", one["ext:score"])
two = res["place-list"][1]
self.assertEqual("63", two["ext:score"])
self.assertEqual("Southampton", two["disambiguation"])
class SearchEventTest(unittest.TestCase):
def testFields(self):
fn = os.path.join(DATA_DIR, "search-event.xml")
with open(fn, 'rb') as msg:
res = mbxml.parse_message(msg)
self.assertEqual(3, res["event-count"])
self.assertEqual(3, len(res["event-list"]))
one = res["event-list"][0]
self.assertEqual("100", one["ext:score"])
two = res["event-list"][1]
self.assertEqual(1, len(two["place-relation-list"]))
|