forked from metabrainz/listenbrainz-server
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_user.py
91 lines (73 loc) · 3.38 KB
/
test_user.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
# -*- coding: utf-8 -*-
import time
import sqlalchemy
from listenbrainz import db
from listenbrainz.db.testing import DatabaseTestCase
import listenbrainz.db.user as db_user
class UserTestCase(DatabaseTestCase):
def test_create(self):
user_id = db_user.create("izzy_cheezy")
self.assertIsNotNone(db_user.get(user_id))
def test_update_token(self):
user = db_user.get_or_create('testuserplsignore')
old_token = user['auth_token']
db_user.update_token(user['id'])
user = db_user.get_by_mb_id('testuserplsignore')
self.assertNotEqual(old_token, user['auth_token'])
def test_update_last_login(self):
""" Tests db.user.update_last_login """
user = db_user.get_or_create('testlastloginuser')
# set the last login value of the user to 0
with db.engine.connect() as connection:
connection.execute(sqlalchemy.text("""
UPDATE "user"
SET last_login = to_timestamp(0)
WHERE id = :id
"""), {
'id': user['id']
})
user = db_user.get(user['id'])
self.assertEquals(int(user['last_login'].strftime('%s')), 0)
db_user.update_last_login(user['musicbrainz_id'])
user = db_user.get_by_mb_id(user['musicbrainz_id'])
# after update_last_login, the val should be greater than the old value i.e 0
self.assertGreater(int(user['last_login'].strftime('%s')), 0)
def test_update_latest_import(self):
user = db_user.get_or_create('testlatestimportuser')
val = int(time.time())
db_user.update_latest_import(user['musicbrainz_id'], val)
user = db_user.get_by_mb_id(user['musicbrainz_id'])
self.assertEqual(val, int(user['latest_import'].strftime('%s')))
db_user.update_latest_import(user['musicbrainz_id'], val - 10)
user = db_user.get_by_mb_id(user['musicbrainz_id'])
self.assertEqual(val, int(user['latest_import'].strftime('%s')))
val += 10
db_user.update_latest_import(user['musicbrainz_id'], val)
user = db_user.get_by_mb_id(user['musicbrainz_id'])
self.assertEqual(val, int(user['latest_import'].strftime('%s')))
def test_get_recently_logged_in_users(self):
"""Tests getting recently logged in users"""
# create two users, set one's last_login
# to a very old value and one's last_login
# to now and then call get_recently_logged_in_users
user1 = db_user.get_or_create('recentuser1')
with db.engine.connect() as connection:
connection.execute(sqlalchemy.text("""
UPDATE "user"
SET last_login = to_timestamp(0)
WHERE musicbrainz_id = :musicbrainz_id
"""), {
'musicbrainz_id': 'recentuser1'
})
user2 = db_user.get_or_create('recentuser2')
with db.engine.connect() as connection:
connection.execute(sqlalchemy.text("""
UPDATE "user"
SET last_login = NOW()
WHERE musicbrainz_id = :musicbrainz_id
"""), {
'musicbrainz_id': 'recentuser2'
})
recent_users = db_user.get_recently_logged_in_users()
self.assertEqual(len(recent_users), 1)
self.assertEqual(recent_users[0]['musicbrainz_id'], 'recentuser2')