def test_to_dict_with_subtitle(self):
e = TemplateElement(self.mock_title, subtitle=self.mock_subtitle)
e_dict = e.to_dict()
assert self.mock_title == e_dict['title']
assert self.mock_subtitle == e_dict['subtitle']
def test_to_dict_with_image_url(self):
e = TemplateElement(self.mock_title, image_url=self.mock_image_url)
e_dict = e.to_dict()
assert self.mock_title == e_dict['title']
assert self.mock_image_url == e_dict['image_url']
def test_to_dict_with_button(self):
mock_button_title, mock_button_url = 'mock-button-title', 'test://mock-button-url/'
e = TemplateElement(
self.mock_title,
buttons=[UrlButton(mock_button_title, mock_button_url)])
e_dict = e.to_dict()
assert self.mock_title == e_dict['title']
assert mock_button_title == e_dict['buttons'][0]['title']
assert mock_button_url == e_dict['buttons'][0]['url']
def test_to_dict_minimal(self):
e = TemplateElement(self.mock_title)
e_dict = e.to_dict()
assert self.mock_title == e_dict['title']
