def test_playlist():
#添加测试频道
channel_name = u"test_channel_name"
channel_uuid = u"mk_test_douban-cid"
channel = channel_model.add_channel(channel_name, channel_uuid)
assert len(playlist.get_music_by_channel(channel, 20)) == 0
#添加测试音乐
music_information = get_test_music()
new_music_list = []
for i in range(20):
music_information[u"cover"].seek(0)
music_information[u"audio"].seek(0)
music_information[u"uuid"] += unicode(i)
music = music_model.add_music(music_information[u"title"], music_information[u"artist"], music_information[u"album"]
, music_information[u"company"], music_information[u"public_time"], music_information[u"kbps"], music_information[u"cover"], music_information[u"audio"], music_information[u"uuid"])
new_music_list.append(music.key)
#往测试频道中添加测试音乐信息
channel_model.update_channel(channel, music_list=new_music_list)
channel = channel_model.get_channel(key=channel.key)[0]
assert len(playlist.get_music_by_channel(channel, 30)) == 20
assert len(playlist.get_music_by_channel(channel, 20)) == 20
assert len(playlist.get_music_by_channel(channel, 10)) == 10
#删除
channel_model.delete_channel(channel)
music_list = music_model.get_music(title=music_information[u"title"])
for music in music_list:
music_model.delete_music(music)
def update_channel_list():
r = requests.get("http://www.douban.com/j/app/radio/channels")
r = json.loads(r.text)
channel_list = []
assert 'channels' in r
for channel in r['channels']:
cid = int(channel['channel_id'])
uuid = DOUBAN_CHANNEL_UUID_FORMAT % cid
if cid != 0 and not get_channel(uuid=uuid):
# not private list and not in db
new_channel = add_channel(channel['name'], uuid)
channel_list.append(new_channel.name)
return channel_list
def test_channel():
channel_name = u"测试频道"
channel_uuid = u"test_channel"
# 测试添加
channel = channel_model.add_channel(channel_name, channel_uuid)
assert channel.name == channel_name and channel_uuid == channel.uuid
# 测试获取
channel = channel_model.get_channel(name=channel_name)[0]
assert channel.name == channel_name and channel.uuid == channel.uuid
# 测试更新
update_channel_name = u"更新测试频道"
channel_model.update_channel(channel, name=update_channel_name)
channel = channel_model.get_channel(uuid=channel_uuid)[0]
assert channel.name == update_channel_name
# 测试删除
channel_model.delete_channel(channel)
channel = channel_model.get_channel(uuid=channel_uuid)
assert not channel
