OpcPackage round-trips an xlsx file

Steve Canny · Steve Canny · commit 6636d016648b · 2013-08-06T01:35:04.000-07:00
diff --git a/features/save-package.feature b/features/save-package.feature
@@ -14,3 +14,9 @@ Feature: Save an OPC package
       When I open a PowerPoint file
        And I save the presentation package
       Then I see the pptx file in the working directory
+
+  Scenario: Round-trip an .xlsx file
+     Given a clean working directory
+      When I open an Excel file
+       And I save the spreadsheet package
+      Then I see the xlsx file in the working directory
diff --git a/features/steps/opc_steps.py b/features/steps/opc_steps.py
@@ -26,15 +26,17 @@ def absjoin(*paths):
 test_file_dir = absjoin(thisdir, '../../tests/test_files')
 basic_docx_path = absjoin(test_file_dir, 'test.docx')
 basic_pptx_path = absjoin(test_file_dir, 'test.pptx')
+basic_xlsx_path = absjoin(test_file_dir, 'test.xlsx')
 saved_docx_path = absjoin(scratch_dir, 'test_out.docx')
 saved_pptx_path = absjoin(scratch_dir, 'test_out.pptx')
+saved_xlsx_path = absjoin(scratch_dir, 'test_out.xlsx')
 
 
 # given ====================================================
 
 @given('a clean working directory')
 def step_given_clean_working_dir(context):
-    files_to_clean_out = (saved_docx_path, saved_pptx_path)
+    files_to_clean_out = (saved_docx_path, saved_pptx_path, saved_xlsx_path)
     for path in files_to_clean_out:
         if os.path.isfile(path):
             os.remove(path)
@@ -47,6 +49,11 @@ def step_given_python_opc_working_environment(context):
 
 # when =====================================================
 
+@when('I open an Excel file')
+def step_when_open_basic_xlsx(context):
+    context.pkg = OpcPackage.open(basic_xlsx_path)
+
+
 @when('I open a PowerPoint file')
 def step_when_open_basic_pptx(context):
     context.pkg = OpcPackage.open(basic_pptx_path)
@@ -71,6 +78,13 @@ def step_when_save_presentation_package(context):
     context.pkg.save(saved_pptx_path)
 
 
+@when('I save the spreadsheet package')
+def step_when_save_spreadsheet_package(context):
+    if os.path.isfile(saved_xlsx_path):
+        os.remove(saved_xlsx_path)
+    context.pkg.save(saved_xlsx_path)
+
+
 # then =====================================================
 
 @then('the expected package rels are loaded')
@@ -224,3 +238,12 @@ def step_then_see_pptx_file_in_working_dir(context):
     minimum = 20000
     filesize = os.path.getsize(saved_pptx_path)
     assert filesize > minimum
+
+
+@then('I see the xlsx file in the working directory')
+def step_then_see_xlsx_file_in_working_dir(context):
+    reason = "file '%s' not found" % saved_xlsx_path
+    assert os.path.isfile(saved_xlsx_path), reason
+    minimum = 30000
+    filesize = os.path.getsize(saved_xlsx_path)
+    assert filesize > minimum
diff --git a/tests/test_files/test.xlsx b/tests/test_files/test.xlsx
