def setUp(self):
        self.element = factories.ElementFactory(display_index=0,
                                                element_type='ENTRY',
                                                question='Which valve',
                                                answer='[]')

        self.generator = generators.ElementGenerator(self.element)
        self.element_etree_element = self.generator.generate(
            ElementTree.Element('test'))
        self.attribs = self.element_etree_element.attrib
    def test_error_if_no_action(self):
        element = factories.PluginElementFactory(action=None)

        generators.ElementGenerator(element).generate(
            ElementTree.Element('test'))
    def test_error_if_no_mime_type(self):
        element = factories.PluginElementFactory(mime_type=None)

        generators.ElementGenerator(element).generate(
            ElementTree.Element('test'))
    def test_error_if_no_choices(self):
        element = factories.ChoiceElementFactory(choices=[])

        generators.ElementGenerator(element).generate(
            ElementTree.Element('test'))
    def test_error_if_no_answer(self):
        element = factories.ElementFactory(answer=None)

        generators.ElementGenerator(element).generate(
            ElementTree.Element('test'))
    def test_error_if_blank_question(self):
        element = factories.ElementFactory(question='')

        generators.ElementGenerator(element).generate(
            ElementTree.Element('test'))
    def test_concept_optional(self):
        element = factories.ElementFactory(concept=None)

        element_etree = generators.ElementGenerator(element).generate(
            ElementTree.Element('test'))
        assert_false('concept' in element_etree.attrib)
    def test_error_if_no_concept(self):
        element = factories.ElementFactory(concept=None)

        generators.ElementGenerator(element).generate(
            ElementTree.Element('test'))