125 lines
4.2 KiB
Python
125 lines
4.2 KiB
Python
import mock
|
|
from builtins import bytes
|
|
from lxml import etree
|
|
|
|
from .test_helper import *
|
|
|
|
from kiwi.exceptions import (
|
|
KiwiSchemaImportError,
|
|
KiwiValidationError,
|
|
KiwiDescriptionInvalid,
|
|
KiwiDataStructureError,
|
|
KiwiDescriptionConflict
|
|
)
|
|
from kiwi.xml_description import XMLDescription
|
|
|
|
|
|
class TestSchema(object):
|
|
def setup(self):
|
|
test_xml = bytes(
|
|
b"""<?xml version="1.0" encoding="utf-8"?>
|
|
<image schemaversion="1.4" name="bob">
|
|
<description type="system">
|
|
<author>John Doe</author>
|
|
<contact>john@example.com</contact>
|
|
<specification>
|
|
say hello
|
|
</specification>
|
|
</description>
|
|
<preferences>
|
|
<packagemanager>zypper</packagemanager>
|
|
<version>1.1.1</version>
|
|
<type image="ext3"/>
|
|
</preferences>
|
|
<repository type="rpm-md">
|
|
<source path="repo"/>
|
|
</repository>
|
|
</image>"""
|
|
)
|
|
self.description_from_file = XMLDescription(description='description')
|
|
self.description_from_data = XMLDescription(xml_content=test_xml)
|
|
|
|
@raises(KiwiDescriptionConflict)
|
|
def test_constructor_conflict(self):
|
|
XMLDescription(description='description', xml_content='content')
|
|
|
|
def test_load_schema_from_xml_content(self):
|
|
schema = etree.parse('../../kiwi/schema/kiwi.rng')
|
|
lookup = '{http://relaxng.org/ns/structure/1.0}attribute'
|
|
for attribute in schema.iter(lookup):
|
|
if attribute.get('name') == 'schemaversion':
|
|
schemaversion = attribute.find(
|
|
'{http://relaxng.org/ns/structure/1.0}value'
|
|
).text
|
|
parsed = self.description_from_data.load()
|
|
assert parsed.get_schemaversion() == schemaversion
|
|
|
|
@raises(KiwiSchemaImportError)
|
|
@patch('lxml.etree.RelaxNG')
|
|
@patch.object(XMLDescription, '_xsltproc')
|
|
def test_load_schema_import_error(self, mock_xslt, mock_relax):
|
|
mock_relax.side_effect = KiwiSchemaImportError(
|
|
'ImportError'
|
|
)
|
|
self.description_from_file.load()
|
|
|
|
@raises(KiwiValidationError)
|
|
@patch('lxml.etree.RelaxNG')
|
|
@patch('lxml.etree.parse')
|
|
@patch.object(XMLDescription, '_xsltproc')
|
|
def test_load_schema_validation_error_from_file(
|
|
self, mock_xslt, mock_parse, mock_relax
|
|
):
|
|
mock_validate = mock.Mock()
|
|
mock_validate.validate.side_effect = KiwiValidationError(
|
|
'ValidationError'
|
|
)
|
|
mock_relax.return_value = mock_validate
|
|
self.description_from_file.load()
|
|
|
|
@raises(KiwiDescriptionInvalid)
|
|
@patch('lxml.etree.RelaxNG')
|
|
@patch('lxml.etree.parse')
|
|
@patch.object(XMLDescription, '_xsltproc')
|
|
def test_load_schema_description_from_file_invalid(
|
|
self, mock_xslt, mock_parse, mock_relax
|
|
):
|
|
mock_validate = mock.Mock()
|
|
mock_validate.validate = mock.Mock(
|
|
return_value=False
|
|
)
|
|
mock_relax.return_value = mock_validate
|
|
self.description_from_file.load()
|
|
|
|
@raises(KiwiDescriptionInvalid)
|
|
@patch('lxml.etree.RelaxNG')
|
|
@patch('lxml.etree.parse')
|
|
@patch.object(XMLDescription, '_xsltproc')
|
|
def test_load_schema_description_from_data_invalid(
|
|
self, mock_xslt, mock_parse, mock_relax
|
|
):
|
|
mock_validate = mock.Mock()
|
|
mock_validate.validate = mock.Mock(
|
|
return_value=False
|
|
)
|
|
mock_relax.return_value = mock_validate
|
|
self.description_from_data.load()
|
|
|
|
@raises(KiwiDataStructureError)
|
|
@patch('lxml.etree.RelaxNG')
|
|
@patch('lxml.etree.parse')
|
|
@patch('kiwi.xml_parse.parse')
|
|
@patch.object(XMLDescription, '_xsltproc')
|
|
def test_load_data_structure_error(
|
|
self, mock_xsltproc, mock_xml_parse, mock_etree_parse, mock_relax
|
|
):
|
|
mock_validate = mock.Mock()
|
|
mock_validate.validate = mock.Mock(
|
|
return_value=True
|
|
)
|
|
mock_relax.return_value = mock_validate
|
|
mock_xml_parse.side_effect = KiwiDataStructureError(
|
|
'DataStructureError'
|
|
)
|
|
self.description_from_file.load()
|