def test_should_insert_repeatly(self):
self.assertEqual(
'<p>Foo Bar</p><p>FooBar</p>!<p>Foo</p>!<p>Foo</p>!<p>Foo</p>!',
embed_html(
'<p>Foo Bar</p><p>FooBar</p><p>Foo</p><p>Foo</p><p>Foo</p>',
'!',
initial_words=3,
subsequent_words=1))
def test_should_not_insert_if_inital_budget_has_not_been_reached(self):
self.assertEqual(
'<p>Foo</p><p>Bar</p><p>FooBar</p>!<p>Hello World</p>',
embed_html('<p>Foo</p><p>Bar</p><p>FooBar</p><p>Hello World</p>',
'!',
initial_words=3))
def test_should_embed_into_text_without_paragraphs(self):
self.assertEqual('Hello World!', embed_html('Hello World', '!'))
def test_should_embed_into_empty_html(self):
self.assertEqual('!', embed_html('', '!'))
def test_should_embed_into_none(self):
self.assertEqual('!', embed_html(None, '!'))
def test_should_insert_after_inital_budget(self):
self.assertEqual(
'<p>Foo Bar FooBar</p>!<p>Hello World</p>',
embed_html('<p>Foo Bar FooBar</p><p>Hello World</p>',
'!',
initial_words=3))