__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()