Python PageGenerator示例

示例#1

文件： test_generator.py 项目： ASUSE/sana.protocol_builder

    def setUp(self):
        self.page = factories.PageFactory(display_index=0)

        factories.ElementFactory(page=self.page)

        self.generator = PageGenerator(self.page)
        self.page_etree_element = self.generator.generate(ElementTree.Element("test"))

示例#2

文件： test_generator.py 项目： ASUSE/sana.protocol_builder

class PageGeneratorTest(TestCase):
    def setUp(self):
        self.page = factories.PageFactory(display_index=0)

        factories.ElementFactory(page=self.page)

        self.generator = PageGenerator(self.page)
        self.page_etree_element = self.generator.generate(ElementTree.Element("test"))

    @raises(ValueError)
    def test_error_if_no_elements(self):
        self.page.elements.all().delete()
        PageGenerator(self.page).generate(ElementTree.Element("test"))

    def test_element_has_correct_name(self):
        assert_equals(self.page_etree_element.tag, self.generator.name)

    def test_element_has_no_display_index(self):
        assert_equals(len(self.page_etree_element.attrib), 0)