def test_find_with_pagination(self): for i in range(0, 20): BlockedContact('%d' % i, '').save() records = BlockedContact.find_with_pagination(page_request={'page_no': 2, 'size': 10}) print 'items', records self.assertEqual(10, len(records))
def test_find_all(self): self.blocked_contact.save() another_blocked_contact = BlockedContact('99999999', u'中介') another_blocked_contact.save() records = BlockedContact.findall() print 'BlockedContacts', records self.assertEqual(2, len(records))
def test_remove_blocked_records(self): for i in range(0, 20): job_item = JobItem() job_item.job_title=u'人员_%d' % i job_item.contact = str(random.randint(90000000, 99999999)) job_item.save() # mark the contact as blocked BlockedContact(job_item.contact, u'人员').save() # run the remove action JobItem.remove_blocked_records() conn = self.connect_db() try: c = conn.cursor() c.execute('SELECT COUNT(*) FROM ' + JobItem.table_name) self.assertEqual(c.fetchone()[0], 0, 'Count of job items should be 0') except: pass finally: conn.close()
def setUp(self): self.clean_db() self.blocked_contact = BlockedContact('8888888', u'中介')