source: trunk/genshi/tests/builder.py

# -*- coding: utf-8 -*-
#
# Copyright (C) 2006 Edgewall Software
# All rights reserved.
#
# This software is licensed as described in the file COPYING, which
# you should have received as part of this distribution. The terms
# are also available at http://genshi.edgewall.org/wiki/License.
#
# This software consists of voluntary contributions made by many
# individuals. For the exact contribution history, see the revision
# history and logs, available at http://genshi.edgewall.org/log/.

import doctest
import unittest

from genshi.builder import Element, tag
from genshi.core import Attrs, Markup, Stream
from genshi.input import XML


class ElementFactoryTestCase(unittest.TestCase):

    def test_link(self):
        link = tag.a(href='#', accesskey=None)('Bar')
        events = list(link.generate())
        self.assertEqual((Stream.START,
                          ('a', Attrs([('href', "#")])),
                          (None, -1, -1)), events[0])
        self.assertEqual((Stream.TEXT, 'Bar', (None, -1, -1)), events[1])
        self.assertEqual((Stream.END, 'a', (None, -1, -1)), events[2])

    def test_nonstring_attributes(self):
        """
        Verify that if an attribute value is given as an int (or some other
        non-string type), it is coverted to a string when the stream is
        generated.
        """
        events = list(tag.foo(id=3))
        self.assertEqual((Stream.START, ('foo', Attrs([('id', '3')])),
                          (None, -1, -1)), events[0])

    def test_duplicate_attributes(self):
        link = tag.a(href='#1', href_='#1')('Bar')
        events = list(link.generate())
        self.assertEqual((Stream.START, ('a', Attrs([('href', "#1")])),
                         (None, -1, -1)), events[0])
        self.assertEqual((Stream.TEXT, 'Bar', (None, -1, -1)), events[1])
        self.assertEqual((Stream.END, 'a', (None, -1, -1)), events[2])

    def test_stream_as_child(self):
        events = list(tag.span(XML('<b>Foo</b>')).generate())
        self.assertEqual(5, len(events))
        self.assertEqual((Stream.START, ('span', ())), events[0][:2])
        self.assertEqual((Stream.START, ('b', ())), events[1][:2])
        self.assertEqual((Stream.TEXT, 'Foo'), events[2][:2])
        self.assertEqual((Stream.END, 'b'), events[3][:2])
        self.assertEqual((Stream.END, 'span'), events[4][:2])

    def test_markup_escape(self):
        m = Markup('See %s') % tag.a('genshi',
                                     href='http://genshi.edgwall.org')
        self.assertEqual(m, Markup('See <a href="http://genshi.edgwall.org">'
                                   'genshi</a>'))


def suite():
    suite = unittest.TestSuite()
    suite.addTest(doctest.DocTestSuite(Element.__module__))
    suite.addTest(unittest.makeSuite(ElementFactoryTestCase, 'test'))
    return suite


if __name__ == '__main__':
    unittest.main(defaultTest='suite')