#

# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

# All rights reserved.

# This component and the accompanying materials are made available

# under the terms of "Eclipse Public License v1.0"

# which accompanies this distribution, and is available

# at the URL "http://www.eclipse.org/legal/epl-v10.html".

#

# Initial Contributors:

# Nokia Corporation - initial contribution.

#

# Contributors:

#

# Description: 

#

import unittest

import os

import StringIO

from testautomation.base_testcase import BaseTestCase

from cone.public import api, plugin, utils, exceptions

from cone.validation import schemavalidation 

ROOT_PATH = os.path.dirname(os.path.abspath(__file__))

CONFML1_NAMESPACE = 'http://www.s60.com/xml/confml/1'

CONFML2_NAMESPACE = 'http://www.s60.com/xml/confml/2'

class DummyResource(object):

    def __init__(self, data):

        self.data = data

    def read(self):

        return self.data

    def close(self):

        pass

class DummyConfiguration(object):

    def __init__(self, resources):

        self.resources = resources

    def get_resource(self, ref):

        if ref in self.resources:

            return DummyResource(self.resources[ref])

        else:

            raise exceptions.NotFound("No such resource '%s'!" % ref)

class TestConfmlSchemaValidation(BaseTestCase, schemavalidation.SchemaValidationTestMixin):

    def test_valid_confml2_files(self):

        self.assert_schemavalidation_succeeds(

            type = 'confml',

            dir = os.path.join(ROOT_PATH, 'testdata/schema/confml2/valid'),

            namespace = CONFML2_NAMESPACE)

    def test_invalid_confml2_files(self):

        self.assert_schemavalidation_fails(

            type = 'confml',

            dir = os.path.join(ROOT_PATH, 'testdata/schema/confml2/invalid'),

            namespace = CONFML2_NAMESPACE)

    def test_valid_confml1_files(self):

        self.assert_schemavalidation_succeeds(

            type = 'confml',

            dir = os.path.join(ROOT_PATH, 'testdata/schema/confml1/valid'),

            namespace = CONFML1_NAMESPACE)

    def test_invalid_confml1_files(self):

        self.assert_schemavalidation_fails(

            type = 'confml',

            dir = os.path.join(ROOT_PATH, 'testdata/schema/confml1/invalid'),

            namespace = CONFML1_NAMESPACE)

    def test_validate_confml_invalid_xml_data(self):

        config = DummyConfiguration({'foo.confml': 'foo'})

        problems = schemavalidation.validate_confml_file(config, 'foo.confml')

        self.assertEquals(len(problems), 1)

        prob = problems[0]

        #self.assertEquals(prob.type, api.Problem.TYPE_XML_PROBLEM)

        self.assertEquals(prob.severity, api.Problem.SEVERITY_ERROR)

        self.assertEquals(prob.line, 1)

    def test_validate_confml_invalid_xml_data_but_valid_root(self):

        REF = 'test.confml'

        DATA = """<?xml version="1.0" encoding="UTF-8"?>

            <configuration xmlns="http://www.s60.com/xml/confml/2">

            <test someattr/>

            </configuration>""".encode('utf-8')

        config = DummyConfiguration({REF: DATA})

        problems = schemavalidation.validate_confml_file(config, REF)

        self.assertEquals(len(problems), 1)

        prob = problems[0]

        #self.assertEquals(prob.type, api.Problem.TYPE_XML_PROBLEM)

        self.assertEquals(prob.severity, api.Problem.SEVERITY_ERROR)

        self.assertEquals(prob.line, 3)

    def test_validate_confml_unsupported_namespace(self):

        DATA = """<?xml version="1.0" encoding="UTF-8"?>

            <unsupported xmlns="http://www.test.com/xml/unsupported">

                <test someattr="yay"/>

            </unsupported>""".encode('utf-8')

        self.assertRaises(exceptions.ConfmlParseError, schemavalidation.validate_confml_data, DATA)

class TestImplmlSchemaValidation(BaseTestCase, schemavalidation.SchemaValidationTestMixin):

    DUMMY1_XSD_DATA = """<?xml version="1.0" encoding="UTF-8"?>

    <xs:schema xmlns:xs="http://www.w3.org/2001/XMLSchema"

               targetNamespace="http://www.dummy.com/dummy1"

               elementFormDefault="qualified">

        <xs:element name="dummy1">

            <xs:complexType>

                <xs:choice minOccurs="0" maxOccurs="unbounded">

                    <xs:element name="myElem" type="xs:string" minOccurs="0" maxOccurs="unbounded"/>

                </xs:choice>

            </xs:complexType>

        </xs:element>

    </xs:schema>"""

    DUMMY2_XSD_DATA = """<?xml version="1.0" encoding="UTF-8"?>

    <xs:schema xmlns:xs="http://www.w3.org/2001/XMLSchema"

               targetNamespace="http://www.dummy.com/dummy2"

               elementFormDefault="qualified">

        <xs:element name="dummy2">

            <xs:complexType>

                <xs:choice minOccurs="0" maxOccurs="unbounded">

                    <xs:element name="someElem" type="xs:string" minOccurs="0" maxOccurs="unbounded"/>

                </xs:choice>

            </xs:complexType>

        </xs:element>

    </xs:schema>"""

    def setUp(self):

        class MockReader1(plugin.ReaderBase):

            NAMESPACE = "http://www.dummy.com/dummy1"

            NAMESPACE_ID = "dummy1ml"

            FILE_EXTENSIONS = ['dummy1ml']

            ROOT_ELEMENT_NAME = 'dummy1'

            @classmethod

            def get_schema_data(cls):

                return self.DUMMY1_XSD_DATA

        class MockReader2(plugin.ReaderBase):

            NAMESPACE = "http://www.dummy.com/dummy2"

            NAMESPACE_ID = "dummy2ml"

            FILE_EXTENSIONS = ['dummy2ml']

            ROOT_ELEMENT_NAME = 'dummy2'

            @classmethod

            def get_schema_data(cls):

                return self.DUMMY2_XSD_DATA

        class MockReader3(plugin.ReaderBase):

            NAMESPACE = "http://www.dummy.com/dummy3"

            NAMESPACE_ID = "dummy3ml"

            FILE_EXTENSIONS = ['dummy3ml']

            ROOT_ELEMENT_NAME = 'dummy3'

        plugin.ImplFactory.set_reader_classes_override([MockReader1, MockReader2, MockReader3])

    def tearDown(self):

        plugin.ImplFactory.set_reader_classes_override(None)

    def test_valid_implml_files(self):

        self.assert_schemavalidation_succeeds(

            type = 'implml',

            dir = os.path.join(ROOT_PATH, 'testdata/schema/implml/valid'))

    def test_invalid_implml_files(self):

        self.assert_schemavalidation_fails(

            type = 'implml',

            dir = os.path.join(ROOT_PATH, 'testdata/schema/implml/invalid'))

    def test_validate_implml_invalid_xml_data(self):

        self.assertRaises(exceptions.XmlParseError, schemavalidation.validate_implml_data, "foo")

    def test_validate_implml_invalid_xml_data_but_valid_root(self):

        DATA = """<?xml version="1.0" encoding="UTF-8"?>

            <implml>

            <test someattr/>

            </implml>""".encode('utf-8')

        self.assertRaises(exceptions.ImplmlParseError, schemavalidation.validate_implml_data, DATA)

    def test_validate_implml_unsupported_namespace(self):

        DATA = """<?xml version="1.0" encoding="UTF-8"?>

            <unsupported xmlns="http://www.test.com/xml/unsupported">

                <test someattr="yay"/>

            </unsupported>""".encode('utf-8')

        self.assertRaises(exceptions.ImplmlParseError, schemavalidation.validate_implml_data, DATA)

if __name__ == '__main__':

    unittest.main()