Example #1
0
class BetaSerieAPITestCase(TestCase):
    idShow = 2410
    # need to test on my user because there is no timeline on devUser.
    idUserTuRz4m = 15930

    def setUp(self):
        self.betaserieAPI = BetaSerieAPI("Dev047", "developer")
        self.assertEqual(self.betaserieAPI.builder.getUrl("", ""), "https://api.betaseries.com")
        self.assertIsNotNone(self.betaserieAPI.token)
        self.assertIsNotNone(self.betaserieAPI.idUser)
        self.betaserieAPI.add_show(self.idShow)

    def test_bad_login(self):
        with self.assertRaises(BadLoginException):
            BetaSerieAPI("Dev047", "TESTBADLOGIN")
        with self.assertRaises(BadLoginException):
            BetaSerieAPI("LOGIN_INEXISTANT", "TESTBADLOGIN")

    def test_shows_tvdbid(self):
        thetvdb_id = 0
        self.assertEqual(self.betaserieAPI.shows_tvdbid(self.idShow)[0], thetvdb_id)


    def test_show_list(self):
        self.assertIsNotNone(self.betaserieAPI.show_list())

    def test_timeline(self):
        self.assertIsNotNone(self.betaserieAPI.timeline(idUser=self.idUserTuRz4m))
        self.assertEqual(len(self.betaserieAPI.timeline(nb=4, idUser=self.idUserTuRz4m)), 4)

    def test_timelinesince(self):
        since = 70039042
        last_id, timeline = self.betaserieAPI.timeline_since(since=since, idUser=self.idUserTuRz4m);
        self.assertIsNotNone(timeline)
        self.assertGreater(len(timeline), 100)
        self.assertGreater(last_id, since)

    def test_timeline_updateShow_since(self):
        since = 70039042
        last_id, timeline = self.betaserieAPI.timeline_since(since=since, idUser=self.idUserTuRz4m);
        self.assertIsNotNone(timeline)
        self.assertGreater(len(timeline), 4)
        self.assertGreater(last_id, since)

    def suite(self):
        suite = TestSuite()
        suite.addTest(TestLoader().loadTestsFromTestCase(BetaSerieAPITestCase))
        return suite


    if __name__ == '__main__':
        TextTestRunner().run(suite())
Example #2
0
    # ----------- If fullUpdate, we retrieve all the current show and add them to sickbear.----------- #
    if paramDb['last_event_id'] == None:
        logger.debug("[BetaBeard] last_index_id is None")
        if param['fullUpdate'] == True:
            shows = beta.show_list();
            logger.debug("[BetaBeard] shows : %s", shows)
            logger.info("[BetaBeard] Start processing shows.")
            for show in shows:
                logger.info("[BetaBeard] Add show in SickBeard : %s (%s)", show[1], show[0])
                if (param['demoMode'] == False):
                    success,message = sickBeard.add_show(show[0], param['location'],  param['lang'],  param['flatten_folder'],  param['status'],  param['initial'],  param['archive'])
                    if (success == False):
                        logger.error("[BetaBeard] Can't add show %s (%s) to sickbeard : %s", show[1], show[0], message)

        # ----------- retrieve last  event processed in betaseries----------- #
        param['last_event_id'], emptyList = beta.timeline_since(None)
    elif param['checkTimeLine']:
        logger.info("[BetaBeard] Start processing timeline.")
        param['last_event_id'], events = beta.timeline_since(paramDb['last_event_id'])
        logger.debug("[BetaBeard] Processing timeline : %s", events)
        if (events != None):
            for event in events:
                logger.debug("[BetaBeard] Event : %s", event)

                # - ADD SERIE - #
                if (event['type'] == 'add_serie'):
                    betaid = str(event['ref_id']);
                    tvdbid, title = beta.shows_tvdbid(betaid)
                    logger.info("[BetaBeard] Add Show to sickbeard : %s (%s)", title, tvdbid)

                    if (param['demoMode'] == False):