def test_parse_invalidxml():
with assert_raises(TVDBException) as cm:
Episode.from_xml('<dfdf></erer>')
assert 'Error parsing Episode XML' in str(cm.exception)
with assert_raises(TVDBException) as cm:
Episode.from_xml('<dfdf></dfdf>')
assert 'Root node is not an Episode' in str(cm.exception)
def test_episode():
import os
xml = open(os.path.join(os.getcwd(),os.sep.join(['tests','inputs','episode.xml'])),'r').read()
image = open(os.path.join(os.getcwd(),os.sep.join(['tests','inputs','297789.jpg'])),'rb').read()
e = Episode.from_xml(xml)
print e
assert e.episode_number ==2
assert len(e.guest_stars) == 8
assert e.filename == 'episodes/78859/297789.jpg'
e.enrich_art()
assert e.filename == image
def test_parse_wrongtype():
with assert_raises(TVDBException) as cm:
Episode.from_xml(1)
assert 'Unrecognised type for xml input <type \'int\'> must be one of str, lxml.etree' == str(cm.exception)
