def test_add_activity(self):
user1 = backend.add_user('username1','photo_url','weibo_id1')
user2 = backend.add_user('username2','photo_url','weibo_id2')
post1 = backend.add_post('title1',user1['id'],'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',user1['id'],'video_url',
pic_small='pic_small')
ac1 = {
'post_id':post1['id'],
'from_id':user1['id'],
'to_id':user2['id'],
'atype':'like'
}
ac2 = {
'post_id':post1['id'],
'from_id':user1['id'],
'to_id':user2['id'],
'atype':'comment'
}
ret = backend.add_activity(ac1)
assert ret['to_id'] == user2['id']
ret = backend.add_activity(ac2)
rets = backend.get_activity_by_user(user2['id'])
assert len(rets) == 2
def test_get_latest_feed(self):
user1 = backend.add_user('username01', 'photo_url01', 'weibo_id01')
user2 = backend.add_user('username02', 'photo_url02', 'weibo_id02')
user3 = backend.add_user('username03', 'photo_url03', 'weibo_id03')
user4 = backend.add_user('username04', 'photo_url04', 'weibo_id04')
post1 = backend.add_post('title01',
user1['id'],
'video_url01',
pic_small='pic_small01')
post2 = backend.add_post('title02',
user2['id'],
'video_url02',
pic_small='pic_small')
post3 = backend.add_post('title03',
user3['id'],
'video_url03',
pic_small='pic_small03')
post4 = backend.add_post('title04',
user4['id'],
'video_url04',
pic_small='pic_small04')
backend.follow_user(user4['id'], user1['id'])
backend.follow_user(user4['id'], user2['id'])
ret = backend.get_latest_feed(user4['id'], limit=10, offset=0)
assert len(ret) == 3
backend.set_post(post3['id'], {'recommended': True})
ret = backend.get_latest_feed(user4['id'], limit=10, offset=0)
assert len(ret) == 4
def test_feeds_view(self):
user1 = backend.add_user('username01','photo_url01','weibo_id01')
user2 = backend.add_user('username02','photo_url02','weibo_id02')
user3 = backend.add_user('username03','photo_url03','weibo_id03')
user4 = backend.add_user('username04','photo_url04','weibo_id04')
post1 = backend.add_post('title01',user1['id'],'video_url01',
pic_small='pic_small01')
post2 = backend.add_post('title02',user2['id'],'video_url02',
pic_small='pic_small')
post3 = backend.add_post('title03',user3['id'],'video_url03',
pic_small='pic_small03')
post4 = backend.add_post('title04',user4['id'],'video_url04',
pic_small='pic_small04')
backend.follow_user(user4['id'],user1['id'])
backend.follow_user(user4['id'],user2['id'])
headers = self.generate_header('weibo_id04')
resp = self.client.get('/feeds/%d' % user4['id'],headers=headers)
ret = json.loads(resp.data)
assert len(ret['results']) == 3
backend.set_post(post3['id'],{'recommended':True})
resp = self.client.get('/feeds/%d'% user4['id'],headers=headers)
ret = json.loads(resp.data)
assert len(ret['results']) == 4
def test_is_following_user(self):
user1 = backend.add_user('username01', 'photo_url01', 'weibo_id01')
user2 = backend.add_user('username02', 'photo_url02', 'weibo_id02')
backend.follow_user(user1['id'], user2['id'])
ret = backend.is_following_user(user1['id'], user2['id'])
assert ret == True
def test_is_following_user(self):
user1 = backend.add_user('username01','photo_url01','weibo_id01')
user2 = backend.add_user('username02','photo_url02','weibo_id02')
backend.follow_user(user1['id'],user2['id'])
ret = backend.is_following_user(user1['id'],user2['id'])
assert ret == True
def test_get_latest_feed(self):
user1 = backend.add_user('username01','photo_url01','weibo_id01')
user2 = backend.add_user('username02','photo_url02','weibo_id02')
user3 = backend.add_user('username03','photo_url03','weibo_id03')
user4 = backend.add_user('username04','photo_url04','weibo_id04')
post1 = backend.add_post('title01',user1['id'],'video_url01',
pic_small='pic_small01')
post2 = backend.add_post('title02',user2['id'],'video_url02',
pic_small='pic_small')
post3 = backend.add_post('title03',user3['id'],'video_url03',
pic_small='pic_small03')
post4 = backend.add_post('title04',user4['id'],'video_url04',
pic_small='pic_small04')
backend.follow_user(user4['id'],user1['id'])
backend.follow_user(user4['id'],user2['id'])
ret = backend.get_latest_feed(user4['id'],limit=10,offset=0)
assert len(ret) == 3
backend.set_post(post3['id'],{'recommended':True})
ret = backend.get_latest_feed(user4['id'],limit=10,offset=0)
assert len(ret) == 4
def test_add_activity(self):
user1 = backend.add_user('username1', 'photo_url', 'weibo_id1')
user2 = backend.add_user('username2', 'photo_url', 'weibo_id2')
post1 = backend.add_post('title1',
user1['id'],
'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',
user1['id'],
'video_url',
pic_small='pic_small')
ac1 = {
'post_id': post1['id'],
'from_id': user1['id'],
'to_id': user2['id'],
'atype': 'like'
}
ac2 = {
'post_id': post1['id'],
'from_id': user1['id'],
'to_id': user2['id'],
'atype': 'comment'
}
ret = backend.add_activity(ac1)
assert ret['to_id'] == user2['id']
ret = backend.add_activity(ac2)
rets = backend.get_activity_by_user(user2['id'])
assert len(rets) == 2
def test_get_user_by_uid(self):
user1 = backend.add_user('username01', 'photo_url01', 'weibo_id01')
user2 = backend.add_user('username02', 'photo_url02', 'weibo_id02')
user = backend.get_user_by_uid(user1['uid'])
assert user['username'] == user1['username']
users = backend.get_user_by_uid([user1['uid'], user2['uid']])
assert len(users) == 2
def test_get_user_by_uid(self):
user1 = backend.add_user('username01','photo_url01','weibo_id01')
user2 = backend.add_user('username02','photo_url02','weibo_id02')
user = backend.get_user_by_uid(user1['uid'])
assert user['username'] == user1['username']
users = backend.get_user_by_uid([user1['uid'],user2['uid']])
assert len(users) == 2
def test_user_activity_view(self):
# get
user1 = backend.add_user('username1','photo_url','weibo_id1')
user2 = backend.add_user('username2','photo_url','weibo_id2')
post1 = backend.add_post('title1',user1['id'],'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',user1['id'],'video_url',
pic_small='pic_small')
comment1 = backend.add_comment(post1['id'],'comment1',user1['id'])
ac1 = {
'post_id':post1['id'],
'from_id':user1['id'],
'to_id':user2['id'],
'atype':'like'
}
ac2 = {
'post_id':post1['id'],
'from_id':user1['id'],
'comment_id':comment1['id'],
'to_id':user2['id'],
'atype':'comment'
}
ret = backend.add_activity(ac1)
ret = backend.add_activity(ac2)
backend.new_install(user2['id'],'device_token')
headers = self.generate_header('weibo_id2')
resp = self.client.get('/user/%d/activity' % user2['id'],headers=headers)
_data = json.loads(resp.data)
assert resp.status_code == 200
assert len(_data['results']) == 2
redis.flushall()
redis.hset('ACTIVITY::UPDATETIME::%(user_id)s' % {'user_id':user2['id']},
'last_update_time',int(time.time() - 3600 * 6))
resp = self.client.get('/user/%d/activity/count' % user2['id'],
headers=headers)
_data = json.loads(resp.data)
assert resp.status_code == 200
assert _data['count'] == 2
redis.hset('ACTIVITY::UPDATETIME::%(user_id)s' % {'user_id':user2['id']},
'last_update_time',int(time.time()) + 2)
resp = self.client.get('/user/%d/activity/count' % user2['id'],
headers=headers)
_data = json.loads(resp.data)
assert resp.status_code == 200
assert _data['count'] == 0
def test_get_user_follower(self):
user1 = backend.add_user('username01', 'photo_url01', 'weibo_id01')
user2 = backend.add_user('username02', 'photo_url02', 'weibo_id02')
user3 = backend.add_user('username03', 'photo_url03', 'weibo_id03')
backend.follow_user(user2['id'], user1['id'])
backend.follow_user(user3['id'], user1['id'])
users = backend.get_user_follower(user1['id'])
assert len(users) == 2
count = backend.get_user_follower_count(user1['id'])
assert count == 2
def test_get_user_follower(self):
user1 = backend.add_user('username01','photo_url01','weibo_id01')
user2 = backend.add_user('username02','photo_url02','weibo_id02')
user3 = backend.add_user('username03','photo_url03','weibo_id03')
backend.follow_user(user2['id'],user1['id'])
backend.follow_user(user3['id'],user1['id'])
users = backend.get_user_follower(user1['id'])
assert len(users) == 2
count = backend.get_user_follower_count(user1['id'])
assert count == 2
def test_post_view(self):
# post
user1 = backend.add_user('username01','photo_url01','weibo_id01')
headers = self.generate_header('weibo_id01')
data = {
'title':'title01',
'author_id':user1['id'],
'video_file':(StringIO('AAAA' * 10000),'hello.mp4'),
'pic_file':(StringIO('AAAA' * 1000),'hello.png')
}
resp = self.client.post('/post',data=data,headers=headers)
_data = json.loads(resp.data)
assert resp.status_code == 200
assert _data['title'] == 'title01'
# get
resp = self.client.get('/post/%d' % _data['id'],headers=headers)
data_get = json.loads(resp.data)
print resp.data
assert data_get['title'] == 'title01'
# put
put_in = {'pic_small':'pic_small'}
resp = self.client.put('/post/%d'%_data['id'],data=json.dumps(put_in),
headers=headers,content_type='application/json')
print resp.data
assert resp.status_code == 204
def test_post_view(self):
# post
user1 = backend.add_user('username01', 'photo_url01', 'weibo_id01')
headers = self.generate_header('weibo_id01')
data = {
'title': 'title01',
'author_id': user1['id'],
'video_file': (StringIO('AAAA' * 10000), 'hello.mp4'),
'pic_file': (StringIO('AAAA' * 1000), 'hello.png')
}
resp = self.client.post('/post', data=data, headers=headers)
_data = json.loads(resp.data)
assert resp.status_code == 200
assert _data['title'] == 'title01'
# get
resp = self.client.get('/post/%d' % _data['id'], headers=headers)
data_get = json.loads(resp.data)
print resp.data
assert data_get['title'] == 'title01'
# put
put_in = {'pic_small': 'pic_small'}
resp = self.client.put('/post/%d' % _data['id'],
data=json.dumps(put_in),
headers=headers,
content_type='application/json')
print resp.data
assert resp.status_code == 204
def test_set_user(self):
user = backend.add_user('username', 'photo_url', 'weibo_id')
_user = backend.set_user(user['id'], {
'username': '******',
'photo_url': 'photo_url2'
})
assert _user['username'] == 'username2'
def test_tag_view(self):
tag1 = Tag(name='tag1',show=True,pic_url='pic_url',
recommended=True)
tag2 = Tag(name='tag2',show=True,pic_url='pic_url',
recommended=True)
db.session.add(tag1)
db.session.add(tag2)
db.session.commit()
user1 = backend.add_user('username01','photo_url01','weibo_id01')
post1 = backend.add_post('post01',user1['id'],
'video_url','pic_small1')
post2 = backend.add_post('post02',user1['id'],
'video_url','pic_small2')
tagging1 = Tagging(taggable_type='post',taggable_id=post1['id'],tag_id=tag1.id)
tagging2 = Tagging(taggable_type='post',taggable_id=post2['id'],tag_id=tag1.id)
headers = self.generate_header('weibo_id01')
# get
resp = self.client.get('/tag/%d'% tag1.id,headers=headers)
data_get = json.loads(resp.data)
assert data_get['tag']['name'] == 'tag1'
# get tags
resp = self.client.get('/tags',headers=headers)
data_get = json.loads(resp.data)
assert len(data_get['results']) == 2
def test_comment_view(self):
# post
user1 = backend.add_user('username01','photo_url01','weibo_id01')
headers = self.generate_header('weibo_id01')
post1 = backend.add_post('post01',user1['id'],
'video_url','pic_small1')
data = {
'author_id':user1['id'],
'content':'comment01',
'post_id':post1['id']
}
resp = self.client.post('/comment',data=json.dumps(data),headers=headers,
content_type='application/json')
_data = json.loads(resp.data)
assert resp.status_code == 200
assert _data['content'] == 'comment01'
# delete
resp = self.client.delete('/comment/%d'%_data['id'],
headers=headers,content_type='application/json')
print resp.data
assert resp.status_code == 204
def test_get_post(self):
user = backend.add_user('username', 'photo_url', 'weibo_id')
post = Post(title='title', author_id=user['id'], pic_small='pic_small')
db.session.add(post)
db.session.commit()
_post = backend.get_post(post.id)
assert _post['title'] == 'title'
def test_get_post(self):
user = backend.add_user('username','photo_url','weibo_id')
post = Post(title='title',author_id=user['id'],pic_small='pic_small')
db.session.add(post)
db.session.commit()
_post = backend.get_post(post.id)
assert _post['title'] == 'title'
def test_add_post(self):
user = backend.add_user('username','photo_url','weibo_id')
post = backend.add_post('title',user['id'],'video_url',
pic_small='pic_small')
assert post['title'] == 'title'
post = backend.set_post(post['id'],{'title':'title2'})
assert post['title'] == 'title2'
def test_add_like(self):
user1 = backend.add_user('username1','photo_url','weibo_id1')
post1 = backend.add_post('title1',user1['id'],'video_url',
pic_small='pic_small')
ret = backend.add_like(user1['id'],post1['id'])
assert ret == 1
ret = backend.del_like(user1['id'],post1['id'])
assert ret == True
def test_get_user_liked_post(self):
user1 = backend.add_user('username1','photo_url','weibo_id1')
user2 = backend.add_user('username2','photo_url','weibo_id2')
post1 = backend.add_post('title1',user1['id'],'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',user1['id'],'video_url',
pic_small='pic_small')
ula1 = UserLikeAsso(user_id=user2['id'],post_id=post1['id'])
ula2 = UserLikeAsso(user_id=user2['id'],post_id=post2['id'])
db.session.add(ula1)
db.session.add(ula2)
db.session.commit()
posts = backend.get_user_liked_post(user2['id'])
assert len(posts) == 2
count = backend.get_user_liked_post_count(user2['id'])
assert count == 2
def test_add_post(self):
user = backend.add_user('username', 'photo_url', 'weibo_id')
post = backend.add_post('title',
user['id'],
'video_url',
pic_small='pic_small')
assert post['title'] == 'title'
post = backend.set_post(post['id'], {'title': 'title2'})
assert post['title'] == 'title2'
def test_comment(self):
user1 = backend.add_user('username1','photo_url','weibo_id1')
user2 = backend.add_user('username2','photo_url','weibo_id2')
post1 = backend.add_post('title1',user1['id'],'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',user1['id'],'video_url',
pic_small='pic_small')
comment1 = backend.add_comment(post1['id'],'comment1',user2['id'])
assert comment1['post_id'] == post1['id']
comment2 = backend.add_comment(post1['id'],'comment2',user2['id'])
comments = backend.get_post_comment(post1['id'])
assert len(comments) == 2
ret = backend.get_post_comment_count(post1['id'])
assert ret == 2
def test_get_user_post(self):
user = backend.add_user('username','photo_url','weibo_id')
post1 = backend.add_post('title1',user['id'],'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',user['id'],'video_url',
pic_small='pic_small')
posts = backend.get_user_post(user['id'])
assert len(posts) == 2
count = backend.get_user_post_count(user['id'])
assert count == 2
def test_add_like(self):
user1 = backend.add_user('username1', 'photo_url', 'weibo_id1')
post1 = backend.add_post('title1',
user1['id'],
'video_url',
pic_small='pic_small')
ret = backend.add_like(user1['id'], post1['id'])
assert ret == 1
ret = backend.del_like(user1['id'], post1['id'])
assert ret == True
def test_is_like_post(self):
user1 = backend.add_user('username1','photo_url','weibo_id1')
post1 = backend.add_post('title1',user1['id'],'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',user1['id'],'video_url',
pic_small='pic_small')
backend.add_like(user1['id'],post1['id'])
backend.add_like(user1['id'],post2['id'])
ret = backend.is_like_post(user1['id'],[post1['id'],post2['id']])
assert ret[post1['id']] == True
def test_new_install(self):
user1 = backend.add_user('username1','photo_url','weibo_id1')
device_token = '1234567890'
install = backend.new_install(user1['id'],device_token)
assert install['device_token'] == device_token
install = backend.get_install_by_user(user1['id'])
assert install['device_token'] == device_token
install = backend.set_install(user1['id'],{'badge':20})
print install
assert install['badge'] == 20
def test_new_install(self):
user1 = backend.add_user('username1', 'photo_url', 'weibo_id1')
device_token = '1234567890'
install = backend.new_install(user1['id'], device_token)
assert install['device_token'] == device_token
install = backend.get_install_by_user(user1['id'])
assert install['device_token'] == device_token
install = backend.set_install(user1['id'], {'badge': 20})
print install
assert install['badge'] == 20
def test_feeds_view(self):
user1 = backend.add_user('username01', 'photo_url01', 'weibo_id01')
user2 = backend.add_user('username02', 'photo_url02', 'weibo_id02')
user3 = backend.add_user('username03', 'photo_url03', 'weibo_id03')
user4 = backend.add_user('username04', 'photo_url04', 'weibo_id04')
post1 = backend.add_post('title01',
user1['id'],
'video_url01',
pic_small='pic_small01')
post2 = backend.add_post('title02',
user2['id'],
'video_url02',
pic_small='pic_small')
post3 = backend.add_post('title03',
user3['id'],
'video_url03',
pic_small='pic_small03')
post4 = backend.add_post('title04',
user4['id'],
'video_url04',
pic_small='pic_small04')
backend.follow_user(user4['id'], user1['id'])
backend.follow_user(user4['id'], user2['id'])
headers = self.generate_header('weibo_id04')
resp = self.client.get('/feeds/%d' % user4['id'], headers=headers)
ret = json.loads(resp.data)
assert len(ret['results']) == 3
backend.set_post(post3['id'], {'recommended': True})
resp = self.client.get('/feeds/%d' % user4['id'], headers=headers)
ret = json.loads(resp.data)
assert len(ret['results']) == 4
def test_user_posts(self):
user1 = backend.add_user('username01','photo_url01','weibo_id01')
post1 = backend.add_post('post01',user1['id'],
'video_url','pic_small1')
post2 = backend.add_post('post02',user1['id'],
'video_url','pic_small2')
post3 = backend.add_post('post03',user1['id'],
'video_url','pic_small3')
headers = self.generate_header('weibo_id01')
resp = self.client.get('/posts/user/%d'%user1['id'],headers=headers)
data_get = json.loads(resp.data)
assert len(data_get['posts']) == 3
def test_post_comment_view(self):
user1 = backend.add_user('username01','photo_url01','weibo_id01')
post1 = backend.add_post('post01',user1['id'],
'video_url','pic_small1')
comment1 = backend.add_comment(post1['id'],'comment1',user1['id'])
comment2 = backend.add_comment(post1['id'],'comment2',user1['id'])
comment3 = backend.add_comment(post1['id'],'comment3',user1['id'])
headers = self.generate_header('weibo_id01')
resp = self.client.get('/post/%d/comment' % post1['id'],headers=headers)
data_get = json.loads(resp.data)
assert len(data_get['comments']) == 3
def test_get_user_liked_post(self):
user1 = backend.add_user('username1', 'photo_url', 'weibo_id1')
user2 = backend.add_user('username2', 'photo_url', 'weibo_id2')
post1 = backend.add_post('title1',
user1['id'],
'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',
user1['id'],
'video_url',
pic_small='pic_small')
ula1 = UserLikeAsso(user_id=user2['id'], post_id=post1['id'])
ula2 = UserLikeAsso(user_id=user2['id'], post_id=post2['id'])
db.session.add(ula1)
db.session.add(ula2)
db.session.commit()
posts = backend.get_user_liked_post(user2['id'])
assert len(posts) == 2
count = backend.get_user_liked_post_count(user2['id'])
assert count == 2
def test_user_posts(self):
user1 = backend.add_user('username01', 'photo_url01', 'weibo_id01')
post1 = backend.add_post('post01', user1['id'], 'video_url',
'pic_small1')
post2 = backend.add_post('post02', user1['id'], 'video_url',
'pic_small2')
post3 = backend.add_post('post03', user1['id'], 'video_url',
'pic_small3')
headers = self.generate_header('weibo_id01')
resp = self.client.get('/posts/user/%d' % user1['id'], headers=headers)
data_get = json.loads(resp.data)
assert len(data_get['posts']) == 3
def test_comment(self):
user1 = backend.add_user('username1', 'photo_url', 'weibo_id1')
user2 = backend.add_user('username2', 'photo_url', 'weibo_id2')
post1 = backend.add_post('title1',
user1['id'],
'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',
user1['id'],
'video_url',
pic_small='pic_small')
comment1 = backend.add_comment(post1['id'], 'comment1', user2['id'])
assert comment1['post_id'] == post1['id']
comment2 = backend.add_comment(post1['id'], 'comment2', user2['id'])
comments = backend.get_post_comment(post1['id'])
assert len(comments) == 2
ret = backend.get_post_comment_count(post1['id'])
assert ret == 2
def test_is_like_post(self):
user1 = backend.add_user('username1', 'photo_url', 'weibo_id1')
post1 = backend.add_post('title1',
user1['id'],
'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',
user1['id'],
'video_url',
pic_small='pic_small')
backend.add_like(user1['id'], post1['id'])
backend.add_like(user1['id'], post2['id'])
ret = backend.is_like_post(user1['id'], [post1['id'], post2['id']])
assert ret[post1['id']] == True
def test_get_user_post(self):
user = backend.add_user('username', 'photo_url', 'weibo_id')
post1 = backend.add_post('title1',
user['id'],
'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',
user['id'],
'video_url',
pic_small='pic_small')
posts = backend.get_user_post(user['id'])
assert len(posts) == 2
count = backend.get_user_post_count(user['id'])
assert count == 2
def test_tag(self):
tag1 = Tag(name='tag1', show=True, pic_url='pic_url', recommended=True)
tag2 = Tag(name='tag2', show=True, pic_url='pic_url', recommended=True)
db.session.add(tag1)
db.session.add(tag2)
db.session.commit()
tags = backend.get_all_tags()
assert len(tags) == 2
tags = backend.get_recommend_tags()
assert len(tags) == 2
tag = backend.get_tag(tag1.id)
assert tag['name'] == 'tag1'
user = backend.add_user('username', 'photo_url', 'weibo_id')
post1 = backend.add_post('title1',
user['id'],
'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',
user['id'],
'video_url',
pic_small='pic_small')
tagging1 = Tagging(taggable_type='post',
taggable_id=post1['id'],
tag_id=tag1.id)
tagging2 = Tagging(taggable_type='post',
taggable_id=post2['id'],
tag_id=tag1.id)
db.session.add(tagging1)
db.session.add(tagging2)
db.session.commit()
posts = backend.get_tag_post(tag1.id)
assert len(posts) == 2
post_count = backend.get_tag_post_count(tag1.id)
assert post_count == 2
def test_tag(self):
tag1 = Tag(name='tag1',show=True,pic_url='pic_url',
recommended=True)
tag2 = Tag(name='tag2',show=True,pic_url='pic_url',
recommended=True)
db.session.add(tag1)
db.session.add(tag2)
db.session.commit()
tags = backend.get_all_tags()
assert len(tags) == 2
tags = backend.get_recommend_tags()
assert len(tags) == 2
tag = backend.get_tag(tag1.id)
assert tag['name'] == 'tag1'
user = backend.add_user('username','photo_url','weibo_id')
post1 = backend.add_post('title1',user['id'],'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',user['id'],'video_url',
pic_small='pic_small')
tagging1 = Tagging(taggable_type='post',taggable_id=post1['id'],
tag_id=tag1.id)
tagging2 = Tagging(taggable_type='post',taggable_id=post2['id'],
tag_id=tag1.id)
db.session.add(tagging1)
db.session.add(tagging2)
db.session.commit()
posts = backend.get_tag_post(tag1.id)
assert len(posts) == 2
post_count = backend.get_tag_post_count(tag1.id)
assert post_count == 2
def test_install(self):
user1 = backend.add_user('username01','photo_url01','weibo_id01')
device_token = 'device_token'
headers = self.generate_header('weibo_id01')
data = {
'device_type':'ios',
'device_token':device_token,
'version':'1.1.10',
'user_id':user1['id']
}
resp = self.client.post('/install',data=json.dumps(data),
headers=headers,content_type='application/json')
_data = json.loads(resp.data)
install_id = _data['install_id']
install = Install.query.filter(Install.user_id == user1['id']).first()
assert install is not None
assert install.device_token.encode('utf-8') == device_token
def test_post_like(self):
user1 = backend.add_user('username01', 'photo_url01', 'weibo_id01')
post1 = backend.add_post('post01', user1['id'], 'video_url',
'pic_small1')
headers = self.generate_header('weibo_id01')
_data = {'user_id': user1['id'], 'post_id': post1['id']}
resp = self.client.post('/post/like',
data=json.dumps(_data),
headers=headers,
content_type='application/json')
data_get = json.loads(resp.data)
assert data_get['like_count'] == 1
resp = self.client.post('/post/unlike',
data=json.dumps(_data),
headers=headers,
content_type='application/json')
data_get = json.loads(resp.data)
assert data_get['like_count'] == 0
def test_post_like(self):
user1 = backend.add_user('username01','photo_url01','weibo_id01')
post1 = backend.add_post('post01',user1['id'],
'video_url','pic_small1')
headers = self.generate_header('weibo_id01')
_data = {
'user_id':user1['id'],
'post_id':post1['id']
}
resp = self.client.post('/post/like',data=json.dumps(_data),
headers=headers,content_type='application/json')
data_get = json.loads(resp.data)
assert data_get['like_count'] == 1
resp = self.client.post('/post/unlike',data=json.dumps(_data),
headers=headers,content_type='application/json')
data_get = json.loads(resp.data)
assert data_get['like_count'] == 0
def test_user_liked_posts(self):
user1 = backend.add_user('username01','photo_url01','weibo_id01')
post1 = backend.add_post('post01',user1['id'],
'video_url','pic_small1')
post2 = backend.add_post('post02',user1['id'],
'video_url','pic_small2')
post3 = backend.add_post('post03',user1['id'],
'video_url','pic_small3')
headers = self.generate_header('weibo_id01')
ula1 = UserLikeAsso(user_id=user1['id'],post_id=post1['id'])
ula2 = UserLikeAsso(user_id=user1['id'],post_id=post2['id'])
ula3 = UserLikeAsso(user_id=user1['id'],post_id=post3['id'])
db.session.add(ula1)
db.session.add(ula2)
db.session.add(ula3)
db.session.commit()
resp = self.client.get('/posts/user/%d/liked'%user1['id'],headers=headers)
data_get = json.loads(resp.data)
assert len(data_get['posts']) == 3
def test_user_liked_posts(self):
user1 = backend.add_user('username01', 'photo_url01', 'weibo_id01')
post1 = backend.add_post('post01', user1['id'], 'video_url',
'pic_small1')
post2 = backend.add_post('post02', user1['id'], 'video_url',
'pic_small2')
post3 = backend.add_post('post03', user1['id'], 'video_url',
'pic_small3')
headers = self.generate_header('weibo_id01')
ula1 = UserLikeAsso(user_id=user1['id'], post_id=post1['id'])
ula2 = UserLikeAsso(user_id=user1['id'], post_id=post2['id'])
ula3 = UserLikeAsso(user_id=user1['id'], post_id=post3['id'])
db.session.add(ula1)
db.session.add(ula2)
db.session.add(ula3)
db.session.commit()
resp = self.client.get('/posts/user/%d/liked' % user1['id'],
headers=headers)
data_get = json.loads(resp.data)
assert len(data_get['posts']) == 3
def test_set_user(self):
user = backend.add_user('username','photo_url','weibo_id')
_user = backend.set_user(user['id'],{'username':'******',
'photo_url':'photo_url2'})
assert _user['username'] == 'username2'
def test_add_user(self):
user = backend.add_user('username','photo_url','weibo_id')
assert user['username'] == 'username'
})
except BackendError,ex:
if ex.message == 'EmptyError':
user = {}
else:
return jsonify(error='服务器开小差了')
else:
return jsonify(**user)
_data = self.get_weibo_info(data['uid'],data['access_token'])
data.update(_data)
try:
user = backend.add_user(
data['username'].encode('utf-8'),
data['photo_url'].encode('utf-8'),
data['uid'].encode('utf-8'),
data['signature'].encode('utf-8'),
data['access_token'].encode('utf-8')
)
except BackendError,ex:
return jsonify(error='add new user error')
print 'add user',user
user.update({'new':True})
return jsonify(**user)
def put(self,user_id):
data = UpdateUserSchema().deserialize(request.json)
try:
user = backend.set_user(user_id,data)
except BackendError,ex:
def test_add_user(self):
user = backend.add_user('username', 'photo_url', 'weibo_id')
assert user['username'] == 'username'
def test_user_activity_view(self):
# get
user1 = backend.add_user('username1', 'photo_url', 'weibo_id1')
user2 = backend.add_user('username2', 'photo_url', 'weibo_id2')
post1 = backend.add_post('title1',
user1['id'],
'video_url',
pic_small='pic_small')
post2 = backend.add_post('title2',
user1['id'],
'video_url',
pic_small='pic_small')
comment1 = backend.add_comment(post1['id'], 'comment1', user1['id'])
ac1 = {
'post_id': post1['id'],
'from_id': user1['id'],
'to_id': user2['id'],
'atype': 'like'
}
ac2 = {
'post_id': post1['id'],
'from_id': user1['id'],
'comment_id': comment1['id'],
'to_id': user2['id'],
'atype': 'comment'
}
ret = backend.add_activity(ac1)
ret = backend.add_activity(ac2)
backend.new_install(user2['id'], 'device_token')
headers = self.generate_header('weibo_id2')
resp = self.client.get('/user/%d/activity' % user2['id'],
headers=headers)
_data = json.loads(resp.data)
assert resp.status_code == 200
assert len(_data['results']) == 2
redis.flushall()
redis.hset(
'ACTIVITY::UPDATETIME::%(user_id)s' % {'user_id': user2['id']},
'last_update_time', int(time.time() - 3600 * 6))
resp = self.client.get('/user/%d/activity/count' % user2['id'],
headers=headers)
_data = json.loads(resp.data)
assert resp.status_code == 200
assert _data['count'] == 2
redis.hset(
'ACTIVITY::UPDATETIME::%(user_id)s' % {'user_id': user2['id']},
'last_update_time',
int(time.time()) + 2)
resp = self.client.get('/user/%d/activity/count' % user2['id'],
headers=headers)
_data = json.loads(resp.data)
assert resp.status_code == 200
assert _data['count'] == 0
'access_token': data['access_token'].encode('utf-8')
})
except BackendError, ex:
if ex.message == 'EmptyError':
user = {}
else:
return jsonify(error='服务器开小差了')
else:
return jsonify(**user)
_data = self.get_weibo_info(data['uid'], data['access_token'])
data.update(_data)
try:
user = backend.add_user(data['username'].encode('utf-8'),
data['photo_url'].encode('utf-8'),
data['uid'].encode('utf-8'),
data['signature'].encode('utf-8'),
data['access_token'].encode('utf-8'))
except BackendError, ex:
return jsonify(error='add new user error')
print 'add user', user
user.update({'new': True})
return jsonify(**user)
def put(self, user_id):
data = UpdateUserSchema().deserialize(request.json)
try:
user = backend.set_user(user_id, data)
except BackendError, ex:
raise ex
