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'中介')
