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']