diff --git a/tests/pp_tests.py b/tests/pp_tests.py index 78df8e613472867d4bc4556b942cbe0e901344a0..2360ea9de8c864178b3f00079259d5b1fc616d35 100644 --- a/tests/pp_tests.py +++ b/tests/pp_tests.py @@ -80,13 +80,20 @@ class PPPrivateTests(test.SickbeardTestDBCase): class PPBasicTests(test.SickbeardTestDBCase): - def setUp(self): - super(PPBasicTests, self).setUp() - self.pp = PostProcessor(test.FILEPATH) - @unittest.skip("this test is not fully configured / implmented") def test_process(self): - self.assertTrue(self.pp.process()) + show = TVShow(3) + show.name = test.SHOWNAME + show.location = test.SHOWDIR + show.saveToDB() + + sickbeard.showList = [show] + ep = TVEpisode(show, test.SEASON, test.EPISODE) + ep.name = "some ep name" + ep.saveToDB() + + pp = PostProcessor(test.FILEPATH) + self.assertTrue(pp.process()) if __name__ == '__main__': diff --git a/tests/snatch_tests.py b/tests/snatch_tests.py index 6ca7df9280728eccfb911c89c3fb06f0d373bc35..5ae0ea3297c3914b8c42f02d4d80e354b3772b9f 100644 --- a/tests/snatch_tests.py +++ b/tests/snatch_tests.py @@ -44,12 +44,6 @@ def _create_fake_xml(items): return xml -# the real one tries to contact tvdb just stop it from getting more info on the ep -def _fake_specifyEP(self, season, episode): - pass - -TVEpisode.specifyEpisode = _fake_specifyEP - searchItems = [] diff --git a/tests/test_lib.py b/tests/test_lib.py index 3bc81216f2f0f68d8b70cfc21306740f570a9f29..b004146ead92921b6bd5d0fdc73537a7bec7652d 100644 --- a/tests/test_lib.py +++ b/tests/test_lib.py @@ -47,6 +47,8 @@ FILENAME = u"show name - s0" + str(SEASON) + "e0" + str(EPISODE) + ".mkv" FILEDIR = os.path.join(TESTDIR, SHOWNAME) FILEPATH = os.path.join(FILEDIR, FILENAME) +SHOWDIR = os.path.join(TESTDIR, SHOWNAME+" final") + #sickbeard.logger.sb_log_instance = sickbeard.logger.SBRotatingLogHandler(os.path.join(TESTDIR, 'sickbeard.log'), sickbeard.logger.NUM_LOGS, sickbeard.logger.LOG_SIZE) sickbeard.logger.SBRotatingLogHandler.log_file = os.path.join(os.path.join(TESTDIR, 'Logs'), 'test_sickbeard.log') @@ -96,6 +98,12 @@ def _dummy_saveConfig(): # this might be considered a hack mainDB.sickbeard.save_config = _dummy_saveConfig +# the real one tries to contact tvdb just stop it from getting more info on the ep +def _fake_specifyEP(self, season, episode): + pass + +sickbeard.tv.TVEpisode.specifyEpisode = _fake_specifyEP + #================= # test classes @@ -105,11 +113,13 @@ class SickbeardTestDBCase(unittest.TestCase): sickbeard.showList = [] setUp_test_db() setUp_test_episode_file() + setUp_test_show_dir() def tearDown(self): sickbeard.showList = [] tearDown_test_db() tearDown_test_episode_file() + tearDown_test_show_dir() class TestDBConnection(db.DBConnection, object): @@ -185,6 +195,15 @@ def setUp_test_episode_file(): def tearDown_test_episode_file(): shutil.rmtree(FILEDIR) + +def setUp_test_show_dir(): + if not os.path.exists(SHOWDIR): + os.makedirs(SHOWDIR) + + +def tearDown_test_show_dir(): + shutil.rmtree(SHOWDIR) + tearDown_test_db() if __name__ == '__main__':