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, [])
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"),))
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)))
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_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"]))
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")
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")
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")
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")
def setUp(self):
self.trakt = Trakt()
self.trakt.users = {"adam": {}}
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"]))
def setUp(self):
self.trakt = Trakt()
def setUp(self):
self.trakt = Trakt()
self.trakt.settings["key"] = "thekey"
