acpt: add shp-add-inline-picture.feature

Steve Canny · Steve Canny · commit 4bccc6fa375d · 2013-12-26T02:51:46.000-08:00
diff --git a/features/shp-add-inline-picture.feature b/features/shp-add-inline-picture.feature
@@ -0,0 +1,11 @@
+Feature: Add inline picture
+  In order to generate a document containing an image
+  As a python-docx developer
+  I need the ability to add an inline picture to a document
+
+  @wip
+  Scenario: Add inline picture to document
+    Given a document
+     When I add an inline picture to the document
+     Then its inline shape type is WD_INLINE_SHAPE.PICTURE
+      And the document contains the inline picture
diff --git a/features/steps/helpers.py b/features/steps/helpers.py
@@ -24,3 +24,10 @@ def test_docx(name):
     Return the absolute path to test .docx file with root name *name*.
     """
     return absjoin(thisdir, 'test_files', '%s.docx' % name)
+
+
+def test_file_path(name):
+    """
+    Return the absolute path to file with *name* in test_files directory
+    """
+    return absjoin(thisdir, 'test_files', '%s' % name)
diff --git a/features/steps/shape.py b/features/steps/shape.py
@@ -6,13 +6,15 @@
 
 from __future__ import absolute_import, print_function, unicode_literals
 
-from behave import given, then
+import hashlib
+
+from behave import given, then, when
 
 from docx import Document
 from docx.enum.shape import WD_INLINE_SHAPE
 from docx.parts import InlineShape, InlineShapes
 
-from .helpers import test_docx
+from .helpers import test_docx, test_file_path
 
 
 # given ===================================================
@@ -44,6 +46,16 @@ def given_inline_shape_known_to_be_shape_of_type(context, shp_of_type):
     context.inline_shape = document.inline_shapes[inline_shape_idx]
 
 
+# when =====================================================
+
+@when('I add an inline picture to the document')
+def when_add_inline_picture_to_document(context):
+    document = context.document
+    context.inline_shape = (
+        document.add_picture(test_file_path('monty-truth.png'))
+    )
+
+
 # then =====================================================
 
 @then('I can access each inline shape by index')
@@ -86,6 +98,21 @@ def then_inline_shape_type_is_shape_type(context, shape_type):
     assert inline_shape.type == expected_value
 
 
+@then('the document contains the inline picture')
+def then_the_document_contains_the_inline_picture(context):
+    document = context.document
+    picture_shape = document.inline_shapes[0]
+    blip = picture_shape._inline.graphic.graphicData.pic.blipFill.blip
+    rId = blip.embed
+    image_part = document.related_parts[rId]
+    image_sha1 = hashlib.sha1(image_part.blob).hexdigest()
+    expected_sha1 = '92abcF0eab86674as0e029342309023423'
+    assert image_sha1 == expected_sha1, (
+        "image SHA1 doesn't match, expected %s, got %s" %
+        (expected_sha1, image_sha1)
+    )
+
+
 @then('the length of the inline shape collection is 5')
 def then_len_of_inline_shape_collection_is_5(context):
     inline_shapes = context.document.inline_shapes
