Initial support for unit tests (#201)

Samir Aguiar · Philippe Lagadec
1 parent 9f71892c
Showing 8 changed files with 32 additions and 0 deletions
.travis.yml
README.md
setup.py
tests/__init__.py
tests/rtfobj/__init__.py
tests/rtfobj/test_issue_185.py
tests/test-data/rtfobj/issue_185.rtf
tests/testdata_reader.py
+language: python
+
+python:
+  - "2.7"
+cache: pip
+script:
+  - python setup.py test
 python-oletools
 ===============
 [![PyPI](https://img.shields.io/pypi/v/oletools.svg)](https://pypi.python.org/pypi/oletools)
+![Build](https://travis-ci.org/samiraguiar/oletools.svg?branch=add-unit-tests)
  
 [oletools](http://www.decalage.info/python/oletools) is a package of python tools to analyze
 [Microsoft OLE2 files](http://en.wikipedia.org/wiki/Compound_File_Binary_Format) 
@@ -314,6 +314,7 @@ def main():
         download_url=download_url,
 #        data_files=data_files,
         entry_points=entry_points,
+        test_suite="tests",
         # scripts=scripts,
     )
  
+import unittest, sys, os
+
+from .. import testdata_reader
+from oletools import rtfobj
+
+class TestRtfObjIssue185(unittest.TestCase):
+    def test_skip_space_after_bin_control_word(self):
+        data = testdata_reader.read('rtfobj/issue_185.rtf')
+        rtfp = rtfobj.RtfObjParser(data)
+        rtfp.parse()
+        objects = rtfp.objects
+
+        self.assertTrue(len(objects) == 1)
+
+if __name__ == '__main__':
+    unittest.main()
+{\rt{\object\objautlink\objupdate\rsltpict\objw37542\objh829\objscalex59286\objscaley86308{\*\objclass \'77}{\*\objdata 32\bin6 FF}}}
 \ No newline at end of file
+import os
+
+def read(relative_path):
+    test_data = os.path.dirname(os.path.abspath(__file__)) + '/test-data/'
+    return open(test_data + relative_path, 'rb').read()
+