add test cases

Author: BruceZhang1993

.travis.yml

@@ -11,6 +11,7 @@ install:
   - pip install -r requirements.txt
 script:
   - python setup.py build
+  - pytest -v
 deploy:
   provider: pypi
   user: $PYPI_USER

setup.cfg

@@ -0,0 +1,2 @@
+[aliases]
+test=pytest

setup.py

@@ -27,6 +27,7 @@
     long_description_content_type="text/markdown",
     keywords=['index', 'listing', 'directory'],
     install_requires=requirements,
+    tests_require=['pytest'],
     entry_points={
         'console_scripts': [
             'index-generator=index_generator.__main__:main'

tests/test_entries.py

@@ -0,0 +1,18 @@
+from index_generator.models.entries import sizeof_fmt, Entry
+
+
+def test_sizeof_fmt():
+    assert sizeof_fmt(100) == '100 B'
+    assert sizeof_fmt(1099) == '1.1 KiB'
+    assert sizeof_fmt(1499099) == '1.4 MiB'
+    assert sizeof_fmt(2899099001) == '2.7 GiB'
+
+
+def test_entry():
+    entry = Entry('CODE_OF_CONDUCT.md', '.', base='/', human=False)
+    assert entry.path == '/CODE_OF_CONDUCT.md'
+    assert entry.mime == 'text/markdown'
+    assert type(entry.modified) is float
+    assert type(entry.size) is int
+    assert entry.isDir is False
+    assert entry.name == 'CODE_OF_CONDUCT.md'