acpt: add par-insert-paragraph.feature

Steve Canny · Steve Canny · commit cbf3d5cc4898 · 2014-06-24T21:04:45.000-07:00
Acceptance test for Paragraph.insert_paragraph_before() API method.
diff --git a/features/par-insert-paragraph.feature b/features/par-insert-paragraph.feature
@@ -0,0 +1,13 @@
+Feature: Insert a paragraph before or after a paragraph
+  In order to add new content in the middle of an existing document
+  As a developer using python-docx
+  I need a way to insert a paragraph relative to another paragraph
+
+
+  @wip
+  Scenario: Add a new paragraph above an existing paragraph
+    Given a document containing three paragraphs
+     When I insert a paragraph above the second paragraph
+     Then the document contains four paragraphs
+      And the text of the second paragraph matches the text I set
+      And the style of the second paragraph matches the style I set
diff --git a/features/steps/paragraph.py b/features/steps/paragraph.py
@@ -4,7 +4,7 @@
 Step implementations for paragraph-related features
 """
 
-from behave import then, when
+from behave import given, then, when
 
 from docx import Document
 
@@ -14,6 +14,17 @@
 TEST_STYLE = 'Heading1'
 
 
+# given ===================================================
+
+@given('a document containing three paragraphs')
+def given_a_document_containing_three_paragraphs(context):
+    document = Document()
+    document.add_paragraph('foo')
+    document.add_paragraph('bar')
+    document.add_paragraph('baz')
+    context.document = document
+
+
 # when ====================================================
 
 @when('I add a run to the paragraph')
@@ -26,6 +37,12 @@ def when_add_new_text_to_run(context):
     context.r.add_text(test_text)
 
 
+@when('I insert a paragraph above the second paragraph')
+def when_I_insert_a_paragraph_above_the_second_paragraph(context):
+    paragraph = context.document.paragraphs[1]
+    paragraph.insert_paragraph_before('foobar', 'Heading1')
+
+
 @when('I set the paragraph style')
 def when_I_set_the_paragraph_style(context):
     context.paragraph.add_run().add_text(test_text)
@@ -34,6 +51,11 @@ def when_I_set_the_paragraph_style(context):
 
 # then =====================================================
 
+@then('the document contains four paragraphs')
+def then_the_document_contains_four_paragraphs(context):
+    assert len(context.document.paragraphs) == 4
+
+
 @then('the document contains the text I added')
 def then_document_contains_text_I_added(context):
     document = Document(saved_docx_path)
@@ -47,3 +69,15 @@ def then_document_contains_text_I_added(context):
 def then_the_paragraph_has_the_style_I_set(context):
     paragraph = Document(saved_docx_path).paragraphs[-1]
     assert paragraph.style == TEST_STYLE
+
+
+@then('the style of the second paragraph matches the style I set')
+def then_the_style_of_the_second_paragraph_matches_the_style_I_set(context):
+    second_paragraph = context.document.paragraphs[1]
+    assert second_paragraph.style == 'Heading1'
+
+
+@then('the text of the second paragraph matches the text I set')
+def then_the_text_of_the_second_paragraph_matches_the_text_I_set(context):
+    second_paragraph = context.document.paragraphs[1]
+    assert second_paragraph.text == 'foobar'
