add py.test testcases

Author: rambo

project/ndaparser/parser.py

@@ -128,7 +128,7 @@ def ascii2scandic(string):
 
 if __name__ == "__main__":
     transactions = []
-    with open("./testdata.nda") as f:
+    with open("./tests/testdata.nda") as f:
         for line in f:
             transaction = parseLine(line)
             if transaction is not None:

project/ndaparser/tests.py

@@ -0,0 +1 @@
+# -*- coding: utf-8 -*-

project/ndaparser/tests/__init__.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+import os
+
+import pytest
+from ndaparser.parser import parseLine
+
+
+def test_parser():
+    transactions = []
+    with open(os.path.join(os.path.dirname(__file__), "testdata.nda")) as f:
+        for line in f:
+            transaction = parseLine(line)
+            if transaction is not None:
+                transactions.append(transaction)
+    assert len(transactions) == 12
+    assert transactions[2].amount == 40

project/ndaparser/tests/test_parser.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+import datetime
+import os
+
+import pytest
+from django.core.files.uploadedfile import SimpleUploadedFile
+from ndaparser.models import UploadedTransaction
+
+
+@pytest.mark.django_db
+def test_model_can_be_saved(admin_user):
+    with open(os.path.join(os.path.dirname(__file__), "testdata.nda"), 'rb') as f:
+        ut = UploadedTransaction(
+            last_transaction=datetime.datetime.now().date(),
+            user=admin_user,
+            file=SimpleUploadedFile('pytest.nda', f.read())
+        )
+    ut.save()
+    assert ut.pk > 0