class ButtonSheet_with_a_template_and_an_element_factory_Test: def setUp(self): self.sheet = ButtonSheet() self.template = Mock() self.sentinel_paragraph = Mock() self.sentinel_separator = Mock() self.element_factory = Mock() self.text1 = "Peter Graf" @istest def printsOneButton(self): self.sheet.add_button(self.text1) when(self.element_factory).create_paragraph(self.text1).thenReturn(self.sentinel_paragraph) self.sheet.build(self.template, self.element_factory) story = [self.sentinel_paragraph] verify(self.template).build(story) @istest def separates_two_buttons_with_an_indicator(self): self.sheet.add_button(self.text1) when(self.element_factory).create_paragraph(self.text1).thenReturn(self.sentinel_paragraph) when(self.element_factory).create_separator().thenReturn(self.sentinel_separator) self.sheet.build(self.template, self.element_factory) story = [self.sentinel_paragraph, self.sentinel_separator, self.sentinel_paragraph] verify(self.template).build(story)
def setUp(self): self.sheet = ButtonSheet() self.template = Mock() self.sentinel_paragraph = Mock() self.sentinel_separator = Mock() self.element_factory = Mock() self.text1 = "Peter Graf"