示例#1
0
class GetTestCase(unittest.TestCase):

    def setUp(self):
        self.trakt = Trakt()
        self.trakt.settings["key"] = "thekey"

    def raise_(self, ex):
        raise ex

    @patch('plugins.trakt.trakt.requests')
    def test_get_valid(self, mock_request):
        response = "{\"movie_id\": 123}"
        req = StubResponse(200, response)
        mock_request.get.return_value = req
        res = self.trakt.get("http://test.now")
        self.assertEqual(res, response)

    @patch('plugins.trakt.trakt.requests')
    def test_get_error_code(self, mock_request):
        req = StubResponse(400, "")
        mock_request.get.return_value = req
        self.assertRaises(Exception, self.trakt.get, "http://test.now")

    @patch('plugins.trakt.trakt.requests')
    def test_get_error_json(self, mock_request):
        req = StubResponse(200, lambda: self.raise_(ValueError("test")))
        mock_request.get.return_value = req
        res = self.trakt.get("http://test.now")
        gc.collect()
        self.assertEquals(len(self.flushLoggedErrors(ValueError)), 1)
        self.assertEquals(res, [])
示例#2
0
    def test_single_episode(self):
        mock_get = Mock(side_effect=(lambda url: [ACTIVITY_PRESET_EPISODE_1] if "episodes" in url else []))
        self.trakt.users["adam"]["last_sync_episodes"] = Trakt.get_date(ACTIVITY_PRESET_EPISODE_1["watched_at"]) - relativedelta.relativedelta(days=1)
        mock_echo = Mock()
        self.trakt.get = mock_get
        self.trakt.echo = mock_echo

        self.trakt.update_user("adam")

        self.failUnless(mock_echo.called)
        self.assertEqual(mock_echo.call_args[0], (Trakt.format_activity(ACTIVITY_PRESET_EPISODE_1, "adam"),))
示例#3
0
class StartTestCase(unittest.TestCase):

    def setUp(self):
        self.trakt = Trakt()

    @patch('plugins.trakt.trakt.json')
    def test_user_setup(self, mock_json):
        data_users = ['adam','dave','sue','eva']
        user_json = {'users': data_users}
        mock_json.loads.return_value = user_json
        self.trakt.started("")
        self.assertEquals(self.trakt.users, dict(map(lambda u: (u, {}), data_users)))
示例#4
0
    def test_sets_last_sync_on_first_load(self):
        mock_get = Mock(side_effect=(lambda url: [ACTIVITY_PRESET_EPISODE_1] if "episodes" in url else []))
        self.trakt.get = mock_get

        self.trakt.update_user("adam")

        self.assertEquals(self.trakt.users["adam"]["last_sync_episodes"], Trakt.get_date(ACTIVITY_PRESET_EPISODE_1["watched_at"]))
        self.failIf("last_sync_movies" in self.trakt.users["adam"])
示例#5
0
    def test_no_new_episodes(self):
        mock_get = Mock(side_effect=(lambda url: [ACTIVITY_PRESET_EPISODE_1] if "episodes" in url else []))
        self.trakt.users["adam"]["last_sync_episodes"] = Trakt.get_date(ACTIVITY_PRESET_EPISODE_1["watched_at"]) + relativedelta.relativedelta(days=1)
        mock_echo = Mock()
        self.trakt.get = mock_get
        self.trakt.echo = mock_echo

        self.trakt.update_user("adam")

        self.failIf(mock_echo.called)
示例#6
0
    def test_new_activity_both_types(self):
        mock_get = Mock(side_effect=(lambda url: [ACTIVITY_PRESET_EPISODE_1] if "episodes" in url else [ACTIVITY_PRESET_MOVIE_1]))
        self.trakt.users["adam"]["last_sync_episodes"] = Trakt.get_date("2013-03-31T09:28:53.000Z")
        self.trakt.users["adam"]["last_sync_movies"] = Trakt.get_date("2013-03-31T09:28:53.000Z")
        mock_echo = Mock()
        self.trakt.get = mock_get
        self.trakt.echo = mock_echo

        self.trakt.update_user("adam")

        self.assertEqual(mock_echo.call_count, 2)
        mock_echo.assert_any_call(Trakt.format_activity(ACTIVITY_PRESET_EPISODE_1, "adam"))
        mock_echo.assert_any_call(Trakt.format_activity(ACTIVITY_PRESET_MOVIE_1, "adam"))
        self.assertEquals(self.trakt.users["adam"]["last_sync_episodes"], Trakt.get_date(ACTIVITY_PRESET_EPISODE_1["watched_at"]))
        self.assertEquals(self.trakt.users["adam"]["last_sync_movies"], Trakt.get_date(ACTIVITY_PRESET_MOVIE_1["watched_at"]))
示例#7
0
 def test_utf8(self):
     activity = json.load(open(os.path.join(self.dir, "test_format_unicode.json")))
     message = Trakt.format_activity(activity, "User")
     self.assertEqual(message, "User watched 'The Walking Dead \u263b', S05E09 'What Happened and What\u2019s Going On \u263b' http://www.trakt.tv/episodes/998958")
示例#8
0
 def test_watch_movie(self):
     activity = json.load(open(os.path.join(self.dir, "test_format_watch_movie.json")))
     message = Trakt.format_activity(activity, "User")
     self.assertEqual(message, "User watched 'Soul Kitchen' (2009) http://www.trakt.tv/movies/19911")
示例#9
0
 def test_scrobble_episode(self):
     activity = json.load(open(os.path.join(self.dir, "test_format_scrobble_episode.json")))
     message = Trakt.format_activity(activity, "User")
     self.assertEqual(message, "User scrobbled 'The Simpsons', S26E10 'The Man Who Came to Be Dinner' http://www.trakt.tv/episodes/1390653")
示例#10
0
 def test_watch_episode(self):
     activity = json.load(open(os.path.join(self.dir, "test_format_watch_episode.json")))
     message = Trakt.format_activity(activity, "User")
     self.assertEqual(message, "User watched 'Marvel's Agents of S.H.I.E.L.D.', S01E11 'The Magical Place' http://www.trakt.tv/episodes/74015")
示例#11
0
 def setUp(self):
     self.trakt = Trakt()
     self.trakt.users = {"adam": {}}
示例#12
0
class UpdateTestCase(unittest.TestCase):

    def setUp(self):
        self.trakt = Trakt()
        self.trakt.users = {"adam": {}}

    def test_no_entries(self):
        mock_get = Mock(return_value=[])
        self.trakt.get = mock_get

        self.trakt.update_user("adam")

        mock_get.assert_any_call(API_ACTIVITY.format("adam", "episodes"))
        mock_get.assert_any_call(API_ACTIVITY.format("adam", "movies"))
        self.failIf("last_sync_episodes" in self.trakt.users["adam"])
        self.failIf("last_sync_movies" in self.trakt.users["adam"])

    def test_sets_last_sync_on_first_load(self):
        mock_get = Mock(side_effect=(lambda url: [ACTIVITY_PRESET_EPISODE_1] if "episodes" in url else []))
        self.trakt.get = mock_get

        self.trakt.update_user("adam")

        self.assertEquals(self.trakt.users["adam"]["last_sync_episodes"], Trakt.get_date(ACTIVITY_PRESET_EPISODE_1["watched_at"]))
        self.failIf("last_sync_movies" in self.trakt.users["adam"])

    def test_single_episode(self):
        mock_get = Mock(side_effect=(lambda url: [ACTIVITY_PRESET_EPISODE_1] if "episodes" in url else []))
        self.trakt.users["adam"]["last_sync_episodes"] = Trakt.get_date(ACTIVITY_PRESET_EPISODE_1["watched_at"]) - relativedelta.relativedelta(days=1)
        mock_echo = Mock()
        self.trakt.get = mock_get
        self.trakt.echo = mock_echo

        self.trakt.update_user("adam")

        self.failUnless(mock_echo.called)
        self.assertEqual(mock_echo.call_args[0], (Trakt.format_activity(ACTIVITY_PRESET_EPISODE_1, "adam"),))

    def test_no_new_episodes(self):
        mock_get = Mock(side_effect=(lambda url: [ACTIVITY_PRESET_EPISODE_1] if "episodes" in url else []))
        self.trakt.users["adam"]["last_sync_episodes"] = Trakt.get_date(ACTIVITY_PRESET_EPISODE_1["watched_at"]) + relativedelta.relativedelta(days=1)
        mock_echo = Mock()
        self.trakt.get = mock_get
        self.trakt.echo = mock_echo

        self.trakt.update_user("adam")

        self.failIf(mock_echo.called)

    def test_new_activity_both_types(self):
        mock_get = Mock(side_effect=(lambda url: [ACTIVITY_PRESET_EPISODE_1] if "episodes" in url else [ACTIVITY_PRESET_MOVIE_1]))
        self.trakt.users["adam"]["last_sync_episodes"] = Trakt.get_date("2013-03-31T09:28:53.000Z")
        self.trakt.users["adam"]["last_sync_movies"] = Trakt.get_date("2013-03-31T09:28:53.000Z")
        mock_echo = Mock()
        self.trakt.get = mock_get
        self.trakt.echo = mock_echo

        self.trakt.update_user("adam")

        self.assertEqual(mock_echo.call_count, 2)
        mock_echo.assert_any_call(Trakt.format_activity(ACTIVITY_PRESET_EPISODE_1, "adam"))
        mock_echo.assert_any_call(Trakt.format_activity(ACTIVITY_PRESET_MOVIE_1, "adam"))
        self.assertEquals(self.trakt.users["adam"]["last_sync_episodes"], Trakt.get_date(ACTIVITY_PRESET_EPISODE_1["watched_at"]))
        self.assertEquals(self.trakt.users["adam"]["last_sync_movies"], Trakt.get_date(ACTIVITY_PRESET_MOVIE_1["watched_at"]))
示例#13
0
 def setUp(self):
     self.trakt = Trakt()
示例#14
0
 def setUp(self):
     self.trakt = Trakt()
     self.trakt.settings["key"] = "thekey"