acpt: add run-add-picture.feature

Steve Canny · Steve Canny · commit 6ac7cedb5f55 · 2014-07-11T20:48:19.000-07:00
diff --git a/features/run-add-picture.feature b/features/run-add-picture.feature
@@ -0,0 +1,25 @@
+Feature: Add picture to a run
+  In order to place an inline picture at an arbitrary place in a document
+  As a developer using python-docx
+  I need a way to add a picture to a run
+
+  @wip
+  Scenario: Add a picture to a body paragraph run
+    Given a run
+     When I add a picture to the run
+     Then the picture appears at the end of the run
+      And the document contains the inline picture
+
+
+  @wip
+  Scenario Outline: Add a picture to a run in a table cell
+    Given a run inside a table cell retrieved from <cell-source>
+     When I add a picture to the run
+     Then the picture appears at the end of the run
+      And the document contains the inline picture
+
+    Examples: Table cell sources
+      | cell-source        |
+      | Table.cell         |
+      | Table.row.cells    |
+      | Table.column.cells |
diff --git a/features/steps/text.py b/features/steps/text.py
@@ -6,6 +6,8 @@
 
 from __future__ import absolute_import, print_function, unicode_literals
 
+import hashlib
+
 from behave import given, then, when
 
 from docx import Document
@@ -14,7 +16,7 @@
 from docx.oxml.ns import nsdecls, qn
 from docx.text import Run
 
-from helpers import test_docx, test_text
+from helpers import test_docx, test_file, test_text
 
 
 # given ===================================================
@@ -23,6 +25,7 @@
 def given_a_run(context):
     document = Document()
     run = document.add_paragraph().add_run()
+    context.document = document
     context.run = run
 
 
@@ -79,6 +82,21 @@ def given_a_run_having_style_char_style(context, char_style):
     context.run = document.paragraphs[0].runs[run_idx]
 
 
+@given('a run inside a table cell retrieved from {cell_source}')
+def given_a_run_inside_a_table_cell_from_source(context, cell_source):
+    document = Document()
+    table = document.add_table(rows=2, cols=2)
+    if cell_source == 'Table.cell':
+        cell = table.cell(0, 0)
+    elif cell_source == 'Table.row.cells':
+        cell = table.rows[0].cells[1]
+    elif cell_source == 'Table.column.cells':
+        cell = table.columns[1].cells[0]
+    run = cell.paragraphs[0].add_run()
+    context.document = document
+    context.run = run
+
+
 # when ====================================================
 
 @when('I add a column break')
@@ -99,6 +117,12 @@ def when_add_page_break(context):
     run.add_break(WD_BREAK.PAGE)
 
 
+@when('I add a picture to the run')
+def when_I_add_a_picture_to_the_run(context):
+    run = context.run
+    run.add_picture(test_file('monty-truth.png'))
+
+
 @when('I add a run specifying its text')
 def when_I_add_a_run_specifying_its_text(context):
     context.run = context.paragraph.add_run(test_text)
@@ -184,6 +208,23 @@ def then_last_item_in_run_is_a_break(context):
     assert context.last_child.tag == expected_tag
 
 
+@then('the picture appears at the end of the run')
+def then_the_picture_appears_at_the_end_of_the_run(context):
+    run = context.run
+    r = run._r
+    blip_rId = r.xpath(
+        './w:drawing/wp:inline/a:graphic/a:graphicData/pic:pic/pic:blipFill/'
+        'a:blip/@r:embed'
+    )[0]
+    image_part = run.part.related_parts[blip_rId]
+    image_sha1 = hashlib.sha1(image_part.blob).hexdigest()
+    expected_sha1 = '79769f1e202add2e963158b532e36c2c0f76a70c'
+    assert image_sha1 == expected_sha1, (
+        "image SHA1 doesn't match, expected %s, got %s" %
+        (expected_sha1, image_sha1)
+    )
+
+
 @then('the run appears in {boolean_prop_name} unconditionally')
 def then_run_appears_in_boolean_prop_name(context, boolean_prop_name):
     run = context.run
