acpt: add scenarios for Section.orientation getter

Steve Canny · Steve Canny · commit 212a65575c8a · 2014-06-21T00:03:22.000-07:00
Also, add WD_ORIENTATION enumeration.
diff --git a/docx/enum/section.py b/docx/enum/section.py
@@ -9,6 +9,26 @@
 from .base import alias, XmlEnumeration, XmlMappedEnumMember
 
 
+@alias('WD_ORIENT')
+class WD_ORIENTATION(XmlEnumeration):
+    """
+    Specifies the page layout orientation.
+    """
+
+    __ms_name__ = 'WdOrientation'
+
+    __url__ = 'http://msdn.microsoft.com/en-us/library/office/ff837902.aspx'
+
+    __members__ = (
+        XmlMappedEnumMember(
+            'PORTRAIT', 0, 'portrait', 'Portrait orientation.'
+        ),
+        XmlMappedEnumMember(
+            'LANDSCAPE', 1, 'landscape', 'Landscape orientation.'
+        ),
+    )
+
+
 @alias('WD_SECTION')
 class WD_SECTION_START(XmlEnumeration):
     """
diff --git a/features/sct-section-props.feature b/features/sct-section-props.feature
@@ -41,3 +41,14 @@ Feature: Access and change section properties
       And I set the section page height to 8.5 inches
      Then the reported page width is 11 inches
       And the reported page height is 8.5 inches
+
+
+  @wip
+  Scenario Outline: Get section orientation
+    Given a section known to have <orientation> orientation
+     Then the reported page orientation is <reported-orientation>
+
+    Examples: Section page orientations
+      | orientation | reported-orientation |
+      | landscape   | WD_ORIENT.LANDSCAPE  |
+      | portrait    | WD_ORIENT.PORTRAIT   |
diff --git a/features/steps/section.py b/features/steps/section.py
@@ -9,7 +9,7 @@
 from behave import given, then, when
 
 from docx import Document
-from docx.enum.section import WD_SECTION
+from docx.enum.section import WD_ORIENT, WD_SECTION
 from docx.shared import Inches
 
 from helpers import test_docx
@@ -36,6 +36,16 @@ def given_a_section_having_start_type(context, start_type):
     context.section = document.sections[section_idx]
 
 
+@given('a section known to have {orientation} orientation')
+def given_a_section_having_known_orientation(context, orientation):
+    section_idx = {
+        'landscape': 0,
+        'portrait':  1
+    }[orientation]
+    document = Document(test_docx('sct-section-props'))
+    context.section = document.sections[section_idx]
+
+
 # when =====================================================
 
 @when('I set the section page height to {y} inches')
@@ -63,6 +73,15 @@ def when_I_set_the_section_start_type_to_start_type(context, start_type):
 
 # then =====================================================
 
+@then('the reported page orientation is {orientation}')
+def then_the_reported_page_orientation_is_orientation(context, orientation):
+    expected_value = {
+        'WD_ORIENT.LANDSCAPE': WD_ORIENT.LANDSCAPE,
+        'WD_ORIENT.PORTRAIT':  WD_ORIENT.PORTRAIT,
+    }[orientation]
+    assert context.section.orientation == expected_value
+
+
 @then('the reported page width is {x} inches')
 def then_the_reported_page_width_is_width(context, x):
     assert context.section.page_width == Inches(float(x))
diff --git a/features/steps/test_files/sct-section-props.docx b/features/steps/test_files/sct-section-props.docx
