class TestSource(TestCase):
def setUp(self):
self.source = XmlSource()
def check(self, xml):
root = xml.getroot()
self.assertEqual(root.tag, "entry")
self.assertEqual(root.text, "Hello")
def test_load_file_without_locator(self):
with make_tempfile() as (fd, name):
os.write(fd, xml_content)
self.check(self.source.load(resource=name))
def test_load_file_with_locator(self):
loc = FileResourceLocator("/tmp")
with make_tempfile() as (fd, name):
os.write(fd, xml_content)
self.check(self.source.load(resource=os.path.basename(name), resource_locator=loc))
def setUp(self):
self.source = XmlSource()
