50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
__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()
|