class TestXsdGen(unittest.TestCase):
def setUp(self):
self.xsd_gen = XSDGenerator()
def tearDown(self):
for f in glob("*.xsd"):
os.unlink(f)
def named_element_check(self, element):
name_found = False
for el in element.iter():
if el.tag.find("element") != -1 and el.attrib["name"] == "text":
name_found = True
break
return name_found
def test_simple_xsd(self):
simple_xsd = self.xsd_gen.get_model_xsd(SimpleModel, pretty_print=False)
xsd_element = etree.XML(simple_xsd)
self.assertEquals(xsd_element.attrib["targetNamespace"], SimpleModel.get_namespace())
name_found = self.named_element_check(xsd_element)
self.assertTrue(name_found)
def test_nested_xsd(self):
nested_xsd = self.xsd_gen.get_model_xsd(NestedModel, pretty_print=False)
tree = etree.XML(nested_xsd)
self.assertEquals(tree.attrib["targetNamespace"], NestedModel.get_namespace())
def test_double_nested_xsd(self):
double_xsd = self.xsd_gen.get_model_xsd(DoubleNestedModel, pretty_print=False)
tree = etree.XML(double_xsd)
self.assertEquals(tree.attrib["targetNamespace"], DoubleNestedModel.get_namespace())
def test_xsd_file(self):
file_name = self.xsd_gen.write_model_xsd_file(SimpleModel, ".")
self.assertTrue(os.path.isfile(file_name))
def test_get_all_models(self):
ret_list = self.xsd_gen.get_all_models_xsd(DoubleNestedModel, pretty_print=True)
self.assertEquals(len(ret_list), 3)
def test_write_all_models(self):
ret_list = self.xsd_gen.write_all_models(DoubleNestedModel, ".")
self.assertEquals(len(ret_list), 3)
for file in ret_list:
self.assertTrue(os.path.isfile(file))
def setUp(self):
self.xsd_gen = XSDGenerator()
class TestXsdGen(unittest.TestCase):
def setUp(self):
self.xsd_gen = XSDGenerator()
def tearDown(self):
for f in glob('*.xsd'):
os.unlink(f)
def named_element_check(self, element):
name_found = False
for el in element.iter():
if el.tag.find('element') != -1 and el.attrib['name'] == 'text':
name_found = True
break
return name_found
def test_simple_xsd(self):
simple_xsd = self.xsd_gen.get_model_xsd(
SimpleModel,
pretty_print=False
)
xsd_element = etree.XML(simple_xsd)
self.assertEquals(xsd_element.attrib['targetNamespace'],
SimpleModel.get_namespace()
)
name_found = self.named_element_check(xsd_element)
self.assertTrue(name_found)
def test_nested_xsd(self):
nested_xsd = self.xsd_gen.get_model_xsd(
NestedModel,
pretty_print=False
)
tree = etree.XML(nested_xsd)
self.assertEquals(
tree.attrib['targetNamespace'],
NestedModel.get_namespace()
)
def test_double_nested_xsd(self):
double_xsd = self.xsd_gen.get_model_xsd(
DoubleNestedModel,
pretty_print=False
)
tree = etree.XML(double_xsd)
self.assertEquals(
tree.attrib['targetNamespace'],
DoubleNestedModel.get_namespace()
)
def test_xsd_file(self):
file_name = self.xsd_gen.write_model_xsd_file(SimpleModel, '.')
self.assertTrue(os.path.isfile(file_name))
def test_get_all_models(self):
ret_list = self.xsd_gen.get_all_models_xsd(DoubleNestedModel, pretty_print=True)
self.assertEquals(len(ret_list), 3)
def test_write_all_models(self):
ret_list = self.xsd_gen.write_all_models(DoubleNestedModel, '.')
self.assertEquals(len(ret_list), 3)
for file in ret_list:
self.assertTrue(os.path.isfile(file))
def test_customized_xms(self):
class Complex(ComplexModel):
simple = SimpleModel.customize(min_occurs=11, max_occurs="unbounded", nillable=False)
bobby = Integer(nillable=False, min_occurs=10,max_occurs=20002)
#simple = MySimple(min_occurs=1, max_occurs="unbounded", nillable=False)
xsd = self.xsd_gen.get_model_xsd(Complex, pretty_print=True)
class ExtraComplex(ComplexModel):
cp = Complex.customize(min_occurs=8, max_occurs=1010, nillable=False, foo="bar")
xsd = self.xsd_gen.get_model_xsd(ExtraComplex, pretty_print=True)
assert xsd
def setUp(self):
self.xsd_gen = XSDGenerator()
class TestXsdGen(unittest.TestCase):
def setUp(self):
self.xsd_gen = XSDGenerator()
def tearDown(self):
for f in glob('*.xsd'):
os.unlink(f)
def named_element_check(self, element):
name_found = False
for el in element.iter():
if el.tag.find('element') != -1 and el.attrib['name'] == 'text':
name_found = True
break
return name_found
def test_simple_xsd(self):
simple_xsd = self.xsd_gen.get_model_xsd(SimpleModel,
pretty_print=False)
xsd_element = etree.XML(simple_xsd)
self.assertEquals(xsd_element.attrib['targetNamespace'],
SimpleModel.get_namespace())
name_found = self.named_element_check(xsd_element)
self.assertTrue(name_found)
def test_nested_xsd(self):
nested_xsd = self.xsd_gen.get_model_xsd(NestedModel,
pretty_print=False)
tree = etree.XML(nested_xsd)
self.assertEquals(tree.attrib['targetNamespace'],
NestedModel.get_namespace())
def test_double_nested_xsd(self):
double_xsd = self.xsd_gen.get_model_xsd(DoubleNestedModel,
pretty_print=False)
tree = etree.XML(double_xsd)
self.assertEquals(tree.attrib['targetNamespace'],
DoubleNestedModel.get_namespace())
def test_xsd_file(self):
file_name = self.xsd_gen.write_model_xsd_file(SimpleModel, '.')
self.assertTrue(os.path.isfile(file_name))
def test_get_all_models(self):
ret_list = self.xsd_gen.get_all_models_xsd(DoubleNestedModel,
pretty_print=True)
self.assertEquals(len(ret_list), 3)
def test_write_all_models(self):
ret_list = self.xsd_gen.write_all_models(DoubleNestedModel, '.')
self.assertEquals(len(ret_list), 3)
for file in ret_list:
self.assertTrue(os.path.isfile(file))
