Add tests for importing INPX files

[m_librarian.git] / tests / test_inp.py

diff --git a/tests/test_inp.py b/tests/test_inp.py
new file mode 100755 (executable)
index 0000000..ff3cfa4
--- /dev/null
+++ b/tests/test_inp.py
@@ -0,0 +1,23 @@
+#! /usr/bin/env python
+
+
+import os
+from tests import TestCase, main
+from m_librarian.db import Author, Book
+from m_librarian.inp import import_inpx
+
+
+class TestInp(TestCase):
+    def test_import_bad_inpx(self):
+        self.assertRaises(
+            ValueError, import_inpx,
+            os.path.join(os.path.dirname(__file__), 'bad.inpx'))
+
+    def test_import_inpx(self):
+        import_inpx(os.path.join(os.path.dirname(__file__), 'test.inpx'))
+        self.assertEqual(Author.select().count(), 4)
+        self.assertEqual(Book.select().count(), 4)
+
+
+if __name__ == "__main__":
+    main()