def login_as_admin(self):
add_user('admin', 'admin', 'admin')
self.app.post('/api/user/current/',
data={
'name': 'admin',
'password': '******'
})
def test_authenticated(self):
add_user('name1', 'pw1', 'normal')
add_user('name2', 'pw2', 'disable')
import fm
self.api = Api(fm.app)
self.api.add_resource(AuthResource, '/test/api/auth/')
rv = self.app.get('/test/api/auth/')
rv = json.loads(rv.data)
assert rv['status'] == 403
rv = self.app.post('/api/user/current/',
data={
'name': 'name1',
'password': '******'
})
rv = self.app.get('/test/api/auth/')
rv = json.loads(rv.data)
assert rv is True
rv = self.app.post('/test/api/auth/')
rv = json.loads(rv.data)
assert rv['status'] == 403
rv = self.app.delete('/api/user/current/')
rv = self.app.post('/api/user/current/',
data={
'name': 'name2',
'password': '******'
})
rv = self.app.get('/test/api/auth/')
rv = json.loads(rv.data)
assert rv['status'] == 403
def test_authenticated(self):
add_user('name1', 'pw1', 'normal')
add_user('name2', 'pw2', 'disable')
import fm
self.api = Api(fm.app)
self.api.add_resource(AuthResource, '/test/api/auth/')
rv = self.app.get('/test/api/auth/')
rv = json.loads(rv.data)
assert rv['status'] == 403
rv = self.app.post('/api/user/current/',
data={'name': 'name1',
'password': '******'})
rv = self.app.get('/test/api/auth/')
rv = json.loads(rv.data)
assert rv is True
rv = self.app.post('/test/api/auth/')
rv = json.loads(rv.data)
assert rv['status'] == 403
rv = self.app.delete('/api/user/current/')
rv = self.app.post('/api/user/current/',
data={'name': 'name2',
'password': '******'})
rv = self.app.get('/test/api/auth/')
rv = json.loads(rv.data)
assert rv['status'] == 403
def test_delete(self):
add_user('name1', 'pw1', 'normal')
rv = self.app.post('/api/user/current/',
data={'name': 'name1',
'password': '******'})
rv = self.app.delete('/api/user/current/')
rv = self.app.get('/api/user/current/')
rv = json.loads(rv.data)
assert rv is None
def test_add_user(self):
name = 'user_name'
password = '******'
level = 'normal'
user = add_user(name, password, level)
assert add_user(name, password, level) is None
assert len(get_user(key=user.key)) == 1
user = get_user()[0]
assert user.name == 'user_name'
assert user.password == _encrypt('user_name', 'user_password')
assert check_user_password(user, 'user_password')
assert user.level == 'normal'
def test_get(self):
rv = self.app.get('/api/user/current/')
rv = json.loads(rv.data)
assert rv is None
add_user('name1', 'pw1', 'normal')
rv = self.app.post('/api/user/current/',
data={'name': 'name1',
'password': '******'})
rv = self.app.get('/api/user/current/')
rv = json.loads(rv.data)
assert rv['name'] == 'name1'
assert rv['level'] == 'normal'
def test_add_user_history(self):
user = add_user('name1', 'pw1', 'normal')
add_user_history(user, 'favor', 'f' * 24)
user = get_user(key=user.key)[0]
assert len(user.history) == 1
assert len(user.favor) == 1
assert len(user.dislike) == 0
add_user_history(user, 'dislike', 'f' * 24)
user = get_user(key=user.key)[0]
assert len(user.history) == 2
assert len(user.favor) == 0
assert len(user.dislike) == 1
add_user_history(user, 'favor', 'f' * 24)
user = get_user(key=user.key)[0]
assert len(user.history) == 3
assert len(user.favor) == 1
assert len(user.dislike) == 0
add_user_history(user, 'listened', 'f' * 24)
user = get_user(key=user.key)[0]
assert len(user.history) == 4
assert user.listened == 1
add_user_history(user, 'shared', 'f' * 24)
user = get_user(key=user.key)[0]
assert len(user.history) == 5
assert len(user.shared) == 1
def test_delete(self):
self.login_as_admin() # this will add a user name admin!
user = add_user('name1', 'pw1', 'normal')
user_key = user.key
assert len(get_user(key=user_key)) == 1
self.app.delete('/api/user/%s/' % (user_key))
assert len(get_user(key=user_key)) == 0
def test_delete_user(self):
name = 'user_name'
password = '******'
level = 'normal'
user = add_user(name, password, level)
assert len(get_user()) == 1
user = get_user()[0]
delete_user(user)
assert len(get_user()) == 0
def add_rekening():
os.system('cls')
users = load_users()
print(f"{'Add rekening baru':^50}")
print("=" * 50)
user_name = input_string("Nama lengkap: ")
user_age = input_int("Umur: ")
while True:
user_id = input_id("Id: ")
if user_id not in [user.id for user in users]:
break
else:
print("Id telah digunakan")
user_pin = input_pin("Pin: ")
user_email = input_email("Email: ")
user = User(user_id, user_name, user_pin, user_age, 0, "active",
user_email)
add_user(user)
print('Rekening berhasil ditambahkan')
os.system('pause')
def test_patch(self):
self.login_as_admin() # this will add a user name admin!
user = add_user('name1', 'pw1', 'normal')
user_key = user.key
rv = self.app.patch('/api/user/%s/' % (user_key),
data={'password': '******',
'level': 'admin'})
rv = json.loads(rv.data)
assert rv['level'] == 'admin'
user = get_user(key=user_key)[0]
assert check_user_password(user, 'new_password')
def test_update_user(self):
name = 'user_name'
password = '******'
level = 'normal'
add_user(name, password, level)
user = get_user()[0]
update_user(user, password='******', level='disable')
user = get_user()[0]
assert user.name == 'user_name'
assert user.password == _encrypt(user.name, 'new_password')
assert check_user_password(user, 'new_password')
assert user.level == 'disable'
try:
update_user(user, name='some_name')
except AttributeError:
pass
try:
update_user(user, level='some_level')
except AssertionError:
pass
def test_get_user_history(self):
# add user
user = add_user('name1', 'pw1', 'normal')
# add music
music1 = add_music('title', 'artist', 'album', 'company', '2013', '64',
self.cover, self.audio, 'uuid1')
self.cover.seek(0)
self.audio.seek(0)
music2 = add_music('title', 'artist', 'album', 'company', '2013', '64',
self.cover, self.audio, 'uuid2')
add_user_history(user, 'favor', music1.key)
add_user_history(user, 'dislike', music2.key)
assert len(get_user_history(user, 0, 10)) == 2
def test_post(self):
add_user('name1', 'pw1', 'normal')
add_user('name2', 'pw2', 'disable')
rv = self.app.post('/api/user/current/',
data={'name': 'name1',
'password': '******'})
rv = json.loads(rv.data)
assert rv['name'] == 'name1'
assert rv['level'] == 'normal'
rv = self.app.post('/api/user/current/',
data={'name': 'name1',
'password': '******'})
rv = json.loads(rv.data)
assert rv is None
rv = self.app.post('/api/user/current/',
data={'name': 'name2',
'password': '******'})
rv = json.loads(rv.data)
assert rv is None
def setup():
'''setup db & update channel & get demo music'''
print 'setuping...'
print 'setting admin'
try:
assert add_user(ADMIN_NAME, ADMIN_PASSWORD, 'admin'), 'admin already exist!!!this will pass'
except:
pass
print 'login douban...'
assert login(), 'check network or the DOUBAN_USER_NAME, DOUBAN_USER_PASSWORD in config.py'
print 'update channel list'
update_channel_list()
print 'update demo music'
channel = get_channel()[0]
music_list = update_music_by_channel(channel, 1)
assert len(music_list) == 1
print 'add demo channel to playlist'
update_channel(channel, playable=True)
def test_get(self):
self.login_as_admin() # this will add a user name admin!
user = add_user('name1', 'pw1', 'normal')
add_user('name2', 'pw2', 'normal')
add_user('name3', 'pw3', 'normal')
add_user('name4', 'pw4', 'normal')
rv = self.app.get('/api/user/?key=' + user.key)
# print json.loads(rv.data)
assert len(json.loads(rv.data)) == 1
rv = self.app.get('/api/user/')
assert json.loads(rv.data)['count'] == 4 + 1
rv = self.app.get('/api/user/?name=name1')
assert len(json.loads(rv.data)) == 1
rv = self.app.get('/api/user/?start=0&end=2')
assert len(json.loads(rv.data)) == 2
rv = self.app.get('/api/user/?start=0')
assert len(json.loads(rv.data)) == 4 + 1
def setup():
'''setup db & update channel & get demo music'''
print 'setuping...'
print 'setting admin'
try:
assert add_user(ADMIN_NAME, ADMIN_PASSWORD,
'admin'), 'admin already exist!!!this will pass'
except:
pass
print 'login douban...'
assert login(
), 'check network or the DOUBAN_USER_NAME, DOUBAN_USER_PASSWORD in config.py'
print 'update channel list'
update_channel_list()
print 'update demo music'
channel = get_channel()[0]
music_list = update_music_by_channel(channel, 1)
assert len(music_list) == 1
print 'add demo channel to playlist'
update_channel(channel, playable=True)
def post(self):
args = UserRegArgs().args
user = add_user(args['name'], args['password'], 'normal')
if user:
return marshal(user, user_fields)
return None
def process_new_user(data):
val = user.add_user(data.get('email'), data.get('grad_year'),
data.get('username'))
if val == False:
raise NameError('Cannot Add User')
def post(self):
args = UserRegArgs().args
user = add_user(args['name'], args['password'], 'normal')
if user:
return marshal(user, user_fields)
return None
def process_new_user(data):
val = user.add_user(data.get('email'), data.get('grad_year'), data.get('username'))
if val == False:
raise NameError('Cannot Add User')
def login_as_admin(self):
add_user('admin', 'admin', 'admin')
self.app.post('/api/user/current/',
data={'name': 'admin',
'password': '******'})
