acpt: migrate doc-add-heading.feature

Steve Canny · Steve Canny · commit f615c58a9ac6 · 2015-02-14T17:30:47.000-08:00
diff --git a/features/api-add-heading.feature b/features/api-add-heading.feature
diff --git a/features/doc-add-heading.feature b/features/doc-add-heading.feature
@@ -0,0 +1,25 @@
+Feature: Add a heading paragraph
+  In order add a heading to a document
+  As a developer using python-docx
+  I need a way to add a heading with its text and level in a single step
+
+
+  Scenario: Add a heading specifying only its text
+    Given a blank document
+     When I add a heading specifying only its text
+     Then the style of the last paragraph is 'Heading 1'
+      And the last paragraph contains the heading text
+
+
+  Scenario Outline: Add a heading specifying level
+    Given a blank document
+     When I add a heading specifying level=<level>
+     Then the style of the last paragraph is '<style>'
+
+   Examples: Heading level styles
+     | level | style     |
+     |   0   | Title     |
+     |   1   | Heading 1 |
+     |   2   | Heading 2 |
+     |   5   | Heading 5 |
+     |   9   | Heading 9 |
diff --git a/features/steps/api.py b/features/steps/api.py
@@ -36,20 +36,6 @@ def when_add_2x2_table_specifying_style_name(context, style_name):
     document.add_table(rows=2, cols=2, style=style_name)
 
 
-@when('I add a heading specifying level={level_str}')
-def when_add_heading_specifying_level(context, level_str):
-    level = int(level_str)
-    document = context.document
-    document.add_heading(level=level)
-
-
-@when('I add a heading specifying only its text')
-def when_add_heading_specifying_only_its_text(context):
-    document = context.document
-    context.heading_text = 'Spam vs. Eggs'
-    document.add_heading(context.heading_text)
-
-
 @when('I add a page break to the document')
 def when_add_page_break_to_document(context):
     document = context.document
@@ -124,14 +110,6 @@ def then_last_paragraph_contains_only_a_page_break(context):
     assert p.runs[0]._r[0].type == 'page'
 
 
-@then('the last paragraph contains the heading text')
-def then_last_p_contains_heading_text(context):
-    document = context.document
-    text = context.heading_text
-    p = document.paragraphs[-1]
-    assert p.text == text
-
-
 @then('the last paragraph contains the text I specified')
 def then_last_p_contains_specified_text(context):
     document = context.document
@@ -152,12 +130,3 @@ def then_last_p_is_empty_paragraph_added(context):
     document = context.document
     p = document.paragraphs[-1]
     assert p.text == ''
-
-
-@then('the style of the last paragraph is \'{style_name}\'')
-def then_the_style_of_the_last_paragraph_is_style(context, style_name):
-    document = context.document
-    paragraph = document.paragraphs[-1]
-    assert paragraph.style.name == style_name, (
-        'got %s' % paragraph.style.name
-    )
diff --git a/features/steps/document.py b/features/steps/document.py
@@ -43,6 +43,18 @@ def given_a_single_section_document_having_portrait_layout(context):
 
 # when ====================================================
 
+@when('I add a heading specifying level={level}')
+def when_add_heading_specifying_level(context, level):
+    context.document.add_heading(level=int(level))
+
+
+@when('I add a heading specifying only its text')
+def when_add_heading_specifying_only_its_text(context):
+    document = context.document
+    context.heading_text = text = 'Spam vs. Eggs'
+    document.add_heading(text)
+
+
 @when('I add a paragraph specifying its style as a {kind}')
 def when_I_add_a_paragraph_specifying_its_style_as_a(context, kind):
     document = context.document
@@ -122,6 +134,14 @@ def then_the_first_section_is_portrait(context):
     assert first_section.page_height == expected_height
 
 
+@then('the last paragraph contains the heading text')
+def then_last_p_contains_heading_text(context):
+    document = context.document
+    text = context.heading_text
+    paragraph = document.paragraphs[-1]
+    assert paragraph.text == text
+
+
 @then('the length of the section collection is 3')
 def then_the_length_of_the_section_collection_is_3(context):
     sections = context.document.sections
@@ -137,3 +157,12 @@ def then_the_second_section_is_landscape(context):
     assert new_section.orientation == WD_ORIENT.LANDSCAPE
     assert new_section.page_width == expected_width
     assert new_section.page_height == expected_height
+
+
+@then('the style of the last paragraph is \'{style_name}\'')
+def then_the_style_of_the_last_paragraph_is_style(context, style_name):
+    document = context.document
+    paragraph = document.paragraphs[-1]
+    assert paragraph.style.name == style_name, (
+        'got %s' % paragraph.style.name
+    )
