def add_message(self, **kwargs):
uid = self.id
post_data = {
'name': kwargs.get('name', 'asdf'),
'link': kwargs.get('link', 'sadf'),
'order': kwargs.get('order', '1'),
'logo': kwargs.get('logo', 'asf'),
}
MLink.create_link(uid, post_data)
def p_user_add_link(self):
'''
user add link.
'''
if self.check_post_role()['ADD']:
pass
else:
return False
post_data = self.get_post_data()
post_data['user_name'] = self.get_current_user()
cur_uid = tools.get_uudd(2)
while MLink.get_by_uid(cur_uid):
cur_uid = tools.get_uudd(2)
if MLink.create_link(cur_uid, post_data):
output = {
'addinfo ': 1,
}
else:
output = {
'addinfo ': 0,
}
return json.dump(output, self)
def user_add_link(self):
if self.check_post_role()['ADD']:
pass
else:
return False
post_data = self.get_post_data()
post_data['user_name'] = self.get_current_user()
cur_uid = tools.get_uudd(2)
while MLink.get_by_uid(cur_uid):
cur_uid = tools.get_uudd(2)
MLink.create_link(cur_uid, post_data)
self.redirect('/link/list')
def test_create_link(self):
uid = self.id
post_data = {
'name': 'asdf',
'link': 'sadf',
'order': '1',
'logo': 'asf',
}
MLink.create_link(uid, post_data)
new_count = MLink.get_counts()
tt = MLink.get_by_uid(uid)
assert tt.name == post_data['name']
assert tt.link == post_data['link']
assert tt.order == int(post_data['order'])
assert tt.logo == post_data['logo']
assert self.raw_count + 1 == new_count
def test_upate(self):
uid = self.id
post_data = {
'name': 'asdf',
'link': 'sadf',
'order': '1',
'logo': 'asf',
}
MLink.create_link(uid, post_data)
new_count = MLink.get_counts()
assert self.raw_count + 1 == new_count
post_data2 = {
'name': 'asdlkjf',
'link': 'sakljdf',
'order': '13',
'logo': 'asfa',
}
MLink.update(uid, post_data2)
new_count = MLink.get_counts()
assert self.raw_count + 1 == new_count
tt = MLink.get_by_uid(uid)
assert tt.name != post_data['name']
assert tt.link != post_data['link']
assert tt.order != int(post_data['order'])
assert tt.logo != post_data['logo']
assert tt.name == post_data2['name']
assert tt.link == post_data2['link']
assert tt.order == int(post_data2['order'])
assert tt.logo == post_data2['logo']
