#!/usr/bin/python import unittest import FragmentFilter import copy import xml.sax.saxutils #from Ft.Lib.pDomlette import PyExpatReader from xml.sax import make_parser, parseString, parse from Ft.Xml import InputSource from Ft.Xml.Domlette import NonvalidatingReader from StringIO import StringIO """ The contents of this file are subject to the Mozilla Public License Version 1.1 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.mozilla.org/MPL/ Software distributed under the License is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License for the specific language governing rights and limitations under the License. The Original Code is available at http://downloads.xmlschemata.org/python/xvif/ The Initial Developer of the Original Code is Eric van der Vlist. Portions created by Eric van der Vlist are Copyright (C) 2002. All Rights Reserved. The name, specification and XML syntax of Regular Fragmentation is from Simon St.Laurent and his original Java implementation available at http://simonstl.com/projects/fragment/original . Contributor(s): """ class TestRegFrag(unittest.TestCase): def rulesParser(self, rulesString, valid=1, display=0): parser = make_parser() rules = FragmentFilter.RulesLoader() parser.setContentHandler(rules) parser.setFeature("http://xml.org/sax/features/namespaces", 1) factory = InputSource.DefaultFactory isrc=factory.fromString(rulesString, "dummy") exception = 0 display=1 if display: print "-------------------------" print rulesString msg="" try: parser.parse(isrc.stream) if display: print rules except FragmentFilter.InvalidRuleException, msg: exception=1 if valid: self.failUnless(exception==0, 'the rules %s should be valid (%s)' % (rulesString, msg)) else: self.failUnless(exception==1, 'the rules %s should be invalid' % rulesString) return rules def fragmentDocument(self, rules, instanceString, valid=1, expected=""): parser = make_parser() fragFilter = FragmentFilter.FragmentFilter(rules) parser.setContentHandler(fragFilter) parser.setFeature("http://xml.org/sax/features/namespaces", 1) out = StringIO() fragFilter.setContentHandler(xml.sax.saxutils.XMLGenerator(out)) factory = InputSource.DefaultFactory isrc=factory.fromString(instanceString, "dummy") msg = "" try: parser.parse(isrc.stream) exception = 0 except FragmentFilter.InvalidFragmentException, msg: exception = 1 if valid: self.failUnless(exception == 0, 'Unexpected exception :'+ msg) self.failUnless(expected.strip() == out.getvalue().strip(), 'Unexpected result:\n'+ out.getvalue()) else: self.failUnless(exception == 1, "Shouldn't have been valid:\n" + instanceString) def test1element(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, valid=0) def test2(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, valid=0) def test2bis(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 02 """, expected=""" Hello! This document contains a gYearMonth. 197011 02 """) def test3(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 1911 1911 """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, valid=0) def test4(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) def test5(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 19year: 70!11 19year: 70!11 """) def test6(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. """) def test7(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) def test8(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 197011 197011 """) def test9(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a gYearMonth. 1970-11 1970-11 """, expected=""" Hello! This document contains a gYearMonth. 19 19 """) def test10(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a splittable piece. 2100-b4a50-999F-2345-5432-1234 """, expected=""" Hello! This document contains a splittable piece. 2100b4a50999F """) self.fragmentDocument(rules, """ Hello! This document contains a splittable piece. 2100-b4a50-999F """, expected=""" Hello! This document contains a splittable piece. 2100b4a50999F """) def test11(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a splittable piece. 2100-b4a50-999F-2345-5432-1234 """, expected=""" Hello! This document contains a splittable piece. 2100b4a50999F234554321234 """) self.fragmentDocument(rules, """ Hello! This document contains a splittable piece. 2100-b4a50-999F """, expected=""" Hello! This document contains a splittable piece. 2100b4a50999F """) def test12(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ Hello! This document contains a splittable piece. 2100-b4a50-999F-2345-5432-1234 """, expected=""" Hello! This document contains a splittable piece. b4a50 """) self.fragmentDocument(rules, """ Hello! This document contains a splittable piece. 2100-b4a50-999F """, expected=""" Hello! This document contains a splittable piece. b4a50 """) self.fragmentDocument(rules, """ Hello! This document contains a splittable piece. 2100-b4a50-999F """, valid=0) def test13(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ 12345 1 """, expected=""" 12345 1 """) def test14(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ azerty 1 12 123 1234 a123 """, expected=""" 1 12 123 """) def test15(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ azerty 1 12 123 1234 a123 """, expected=""" 1 12 12 """) def test15(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ azerty 1 12 123 1234 a123 """, expected=""" """) def test16(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ azerty 1 12 123 1234 a123 """, expected=""" 1 1, 2 1, 2 """) def test17(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ azerty 1 12 123 1234 a123 """, expected=""" 1 1 2 1 2 3 1 2 3 """) def test18(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ azerty 1 12 123 1234 a123 """, expected=""" 2 2 3 2 3 """) def test19(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ azerty 1 12 """, expected=""" """) self.fragmentDocument(rules, """ 12 """, valid=0) self.fragmentDocument(rules, """ 123 """, valid=0) def test20element(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ """, expected=""" 197011 197011 197011200206 """) def test21element(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ """, expected=""" 19 70 11 19 70 11 19 70 11 20 02 06 """) def test22element(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ """, valid=0) self.fragmentDocument(rules, """ """, expected=""" """) def test23element(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ """, valid=0) self.fragmentDocument(rules, """ """, expected=""" 1 9 1 9 """) self.fragmentDocument(rules, """ """, valid=0) def test24element(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ """, valid=0) self.fragmentDocument(rules, """ """, expected=""" """) self.fragmentDocument(rules, """ """, valid=0) def test25(self): rules = self.rulesParser(""" """) self.fragmentDocument(rules, """ """, valid=0) self.fragmentDocument(rules, """ """, expected=""" """) self.fragmentDocument(rules, """ """, expected=""" """) if __name__ == "__main__": unittest.main()