init

python/dn2k/test/ebookconvert/testEbookConvert.py

@@ -0,0 +1,49 @@
+__author__ = 'dev'
+
+import unittest
+import os
+import subprocess
+import ConvertFeed
+
+ABS_CUR = os.path.abspath(os.path.curdir)
+TEST_FEED_NAME = "moz"
+TEST_FEED_RECIPE_FOLDER = os.path.join(ABS_CUR, "recipe")
+TEST_FEED_RECIPE_FILE_EXTENSION = ".recipe"
+TEST_FEED_RECIPE = os.path.join(TEST_FEED_RECIPE_FOLDER, TEST_FEED_NAME + TEST_FEED_RECIPE_FILE_EXTENSION)
+
+TEST_FEED_MOBI_FOLDER = os.path.join(ABS_CUR, "mobi")
+
+TEST_FEED_MOBI_FILE_EXTENSION = ".mobi"
+TEST_FEED_MOBI =  os.path.join(TEST_FEED_MOBI_FOLDER, TEST_FEED_NAME + TEST_FEED_MOBI_FILE_EXTENSION)
+
+TEST_OUTPUT_PROFIL = "kindle"
+TEST_OUTPUT_PROFIL_OPTION = "--output-profile"
+
+TEST_ECONVERT_CMD = "ebook-convert"
+
+class TestEbookConvert(unittest.TestCase):
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def test_ebook_convert(self):
+
+        if os.path.exists(TEST_FEED_MOBI):
+            os.remove(TEST_FEED_MOBI)
+
+        ConvertFeed.convert_feed(TEST_FEED_RECIPE, TEST_FEED_MOBI)
+        self.assertTrue(os.path.exists(TEST_FEED_MOBI))
+
+    def test_ebook_convert_all_feed(self):
+
+        if os.path.exists(TEST_FEED_MOBI):
+            os.remove(TEST_FEED_MOBI)
+
+        ConvertFeed.convert_all_feed(TEST_FEED_RECIPE_FOLDER , TEST_FEED_MOBI_FOLDER)
+        self.assertTrue(os.path.exists(TEST_FEED_MOBI))
+
+if __name__ == '__main__':
+    unittest.main()