def test_get_active_users_to_process(self):
db_user.create(2, 'newspotifyuser')
db_spotify.create_spotify(
user_id=2,
user_token='token',
refresh_token='refresh_token',
token_expires_ts=int(time.time()),
)
users = db_spotify.get_active_users_to_process()
self.assertEqual(len(users), 2)
self.assertEqual(users[0]['user_id'], 1)
self.assertEqual(users[1]['user_id'], 2)
# check order, the users should be sorted by latest_listened_at timestamp
db_user.create(3, 'newnewspotifyuser')
db_spotify.create_spotify(
user_id=3,
user_token='tokentoken',
refresh_token='newrefresh_token',
token_expires_ts=int(time.time()),
)
t = int(time.time())
db_spotify.update_latest_listened_at(2, t + 20)
db_spotify.update_latest_listened_at(1, t + 10)
users = db_spotify.get_active_users_to_process()
self.assertEqual(len(users), 3)
self.assertEqual(users[0]['user_id'], 2)
self.assertEqual(users[1]['user_id'], 1)
self.assertEqual(users[2]['user_id'], 3)
db_spotify.add_update_error(2, 'something broke')
db_spotify.add_update_error(3, 'oops.')
users = db_spotify.get_active_users_to_process()
self.assertEqual(len(users), 1)
self.assertEqual(users[0]['user_id'], 1)
def test_get_active_users_to_process(self):
db_user.create(2, 'newspotifyuser')
db_spotify.create_spotify(
user_id=2,
user_token='token',
refresh_token='refresh_token',
token_expires_ts=int(time.time()),
)
users = db_spotify.get_active_users_to_process()
self.assertEqual(len(users), 2)
self.assertEqual(users[0]['user_id'], 1)
self.assertEqual(users[1]['user_id'], 2)
# check order, the users should be sorted by latest_listened_at timestamp
db_user.create(3, 'newnewspotifyuser')
db_spotify.create_spotify(
user_id=3,
user_token='tokentoken',
refresh_token='newrefresh_token',
token_expires_ts=int(time.time()),
)
t = int(time.time())
db_spotify.update_latest_listened_at(2, t + 20)
db_spotify.update_latest_listened_at(1, t + 10)
users = db_spotify.get_active_users_to_process()
self.assertEqual(len(users), 3)
self.assertEqual(users[0]['user_id'], 2)
self.assertEqual(users[1]['user_id'], 1)
self.assertEqual(users[2]['user_id'], 3)
db_spotify.add_update_error(2, 'something broke')
db_spotify.add_update_error(3, 'oops.')
users = db_spotify.get_active_users_to_process()
self.assertEqual(len(users), 1)
self.assertEqual(users[0]['user_id'], 1)
def test_create_spotify(self):
db_user.create(2, 'spotify')
db_spotify.create_spotify(
user_id=2,
user_token='token',
refresh_token='refresh_token',
token_expires_ts=int(time.time()),
)
token = db_spotify.get_token_for_user(2)
self.assertEqual(token, 'token')
def setUp(self):
super(SpotifyDatabaseTestCase, self).setUp()
db_user.create(1, 'testspotifyuser')
self.user = db_user.get(1)
db_spotify.create_spotify(
user_id=self.user['id'],
user_token='token',
refresh_token='refresh_token',
token_expires_ts=int(time.time()),
)
def test_create_spotify(self):
db_user.create(2, 'spotify')
db_spotify.create_spotify(
user_id=2,
user_token='token',
refresh_token='refresh_token',
token_expires_ts=int(time.time()),
)
token = db_spotify.get_token_for_user(2)
self.assertEqual(token, 'token')
def setUp(self):
super(SpotifyDatabaseTestCase, self).setUp()
db_user.create(1, 'testspotifyuser')
self.user = db_user.get(1)
db_spotify.create_spotify(
user_id=self.user['id'],
user_token='token',
refresh_token='refresh_token',
token_expires_ts=int(time.time()),
)
def test_delete_when_spotify_import_activated(self):
user_id = db_user.create(11, 'kishore')
user = db_user.get(user_id)
self.assertIsNotNone(user)
db_spotify.create_spotify(user_id, 'user token', 'refresh token', 0)
db_user.delete(user_id)
user = db_user.get(user_id)
self.assertIsNone(user)
token = db_spotify.get_token_for_user(user_id)
self.assertIsNone(token)
def add_new_user(user_id, spot_access_token):
"""Create a spotify row for a user based on OAuth access tokens
Args:
user_id: A flask auth `current_user.id`
spot_access_token: A spotipy access token from SpotifyOAuth.get_access_token
"""
access_token = spot_access_token['access_token']
refresh_token = spot_access_token['refresh_token']
expires_at = spot_access_token['expires_at']
db_spotify.create_spotify(user_id, access_token, refresh_token, expires_at)
def add_new_user(user_id, spot_access_token):
"""Create a spotify row for a user based on OAuth access tokens
Args:
user_id: A flask auth `current_user.id`
spot_access_token: A spotipy access token from SpotifyOAuth.get_access_token
"""
access_token = spot_access_token['access_token']
refresh_token = spot_access_token['refresh_token']
expires_at = spot_access_token['expires_at']
db_spotify.create_spotify(user_id, access_token, refresh_token, expires_at)
def add_new_user(user_id, spot_access_token):
"""Create a spotify row for a user based on OAuth access tokens
Args:
user_id: A flask auth `current_user.id`
spot_access_token: A spotipy access token from SpotifyOAuth.get_access_token
"""
access_token = spot_access_token['access_token']
refresh_token = spot_access_token['refresh_token']
expires_at = int(time.time()) + spot_access_token['expires_in']
permissions = spot_access_token['scope']
active = SPOTIFY_IMPORT_PERMISSIONS[0] in permissions and SPOTIFY_IMPORT_PERMISSIONS[1] in permissions
db_spotify.create_spotify(user_id, access_token, refresh_token, expires_at, active, permissions)