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()