def setUp(self): self.krace = Contact(first_name="Kracekumar", last_name="", website="http://kracekumar.com", notes="Co-host of bangpypers", full_address="bellandhur gate, bangalore", phone_number="8553029521", email="*****@*****.**") self.siva = Contact(first_name="Siva", last_name="", website="http://sivaa.in", notes="Host of Bangalore Django User Group", full_address="bellandhur gate, bangalore", phone_number="786569", email="*****@*****.**") ContactBook.add(self.krace) ContactBook.add(self.siva)
def test_index(self): res = ContactBook.index() self.assertTrue(len(res['k']), 1) self.assertTrue(len(res['s']), 1)
def test_contact_book_add(self): self.assertTrue(len(ContactBook.all()), 2)