unittest missing lxml package

.travis.yml

@@ -10,7 +10,7 @@ before_install:
 
 install:
     - python setup.py develop
-    - pip install pytest pytest-cov coveralls
+    - pip install pytest pytest-cov mock coveralls
 
 script:
     - py.test --cov=peepdf

appveyor.yml

@@ -6,7 +6,7 @@ install:
   - "set PATH=%PYTHON%;%PYTHON%/Scripts;%PATH%"
 
   - "python.exe setup.py develop"
-  - "pip.exe install pytest pytest-cov"
+  - "pip.exe install pytest pytest-cov mock"
 
 build: false
 

peepdf/main.py

@@ -71,10 +71,12 @@
 try:
     from lxml import etree
 except:
-    pass
+    etree = None
 
 
 def getPeepXML(statsDict, version, revision):
+    assert etree is not None, "lxml must be installed for --xml"
+
     root = etree.Element('peepdf_analysis', version=version + ' r' + revision, url='http://peepdf.eternal-todo.com',
                          author='Jose Miguel Esparza')
     analysisDate = etree.SubElement(root, 'date')

setup.py

@@ -11,12 +11,13 @@
         "colorama==0.3.7",
         "Pillow==3.2.0",
         "pythonaes==1.0",
-        "lxml==3.6.0",
     ],
     entry_points={
         "console_scripts": [
             "peepdf = peepdf.main:main",
         ],
     },
-    packages=["peepdf"],
+    packages=[
+        "peepdf",
+    ],
 )

tests/test_pee.py

@@ -20,7 +20,10 @@
 #        along with peepdf.    If not, see <http://www.gnu.org/licenses/>.
 #
 
+import mock
 import peepdf
+import peepdf.main
+import pytest
 
 def test_whitespace_after_opening():
     p = peepdf.PDFCore.PDFParser()
@@ -35,3 +38,9 @@ def test_whitespace_after_opening():
             assert obj.object.errors != [
                 "Decoding error: Error decompressing string"
             ]
+
+def test_lxml_missing():
+    with mock.patch.dict(peepdf.main.__dict__, {"etree": None}):
+        with pytest.raises(AssertionError) as e:
+            peepdf.main.getPeepXML(None, None, None)
+        e.match("lxml must be installed")