fix broken test case where session.close() is needed

Author: chfw

pyexcel_webio/__init__.py

@@ -107,8 +107,12 @@ def get_records(self, sheet_name=None, name_columns_by_row=0, **keywords):
         else:
             return None
 
-    def save_to_database(self, session=None, table=None, sheet_name=None, name_columns_by_row=0, **keywords):
-        sheet = self.load_single_sheet(sheet_name=sheet_name, name_columns_by_row=name_columns_by_row, **keywords)
+    def save_to_database(self, session=None, table=None,
+                         sheet_name=None, name_columns_by_row=0, name_rows_by_column=-1, **keywords):
+        sheet = self.load_single_sheet(sheet_name=sheet_name,
+                                       name_columns_by_row=name_columns_by_row,
+                                       name_rows_by_column=name_rows_by_column,
+                                       **keywords)
         if sheet:
             sheet.save_to_database(session, table)
 

tests/test_webio.py

@@ -129,6 +129,7 @@ def test_save_to_database(self):
         myinput.save_to_database(filename=self.testfile, session=self.session, table=Signature,)
         array = pe.get_array(session=self.session, table=Signature)
         assert array == self.data
+        self.session.close()
 
     def tearDown(self):
         os.unlink(self.testfile)
@@ -167,6 +168,7 @@ def test_save_to_database(self):
         assert array == self.data
         array = pe.get_array(session=self.session, table=Signature2)
         assert array == self.data1
+        self.session.close()
 
     def tearDown(self):
         os.unlink(self.testfile)
@@ -212,12 +214,13 @@ def test_make_response_from_table(self):
         Base.metadata.create_all(engine)
         row1 = Signature(X=1,Y=2, Z=3)
         row2 = Signature(X=4, Y=5, Z=6)
-        session =Session()
+        session = Session()
         session.add(row1)
         session.add(row2)
         session.commit()
         webio.make_response_from_a_table(session, Signature, "xls")
         self.verify()
+        session.close()
 
     def verify(self):
         sheet2 = pe.load(OUTPUT)