def setUp(self): self.bucket = KBucket(1, 20, market_id=1) self.bucket.addContact(Contact(1, 'http://foo/1')) self.bucket.addContact(Contact(2, 'http://foo/2')) self.bucket.addContact(Contact(3, 'http://foo/3')) self.bucket.addContact(Contact(4, 'http://foo/4')) self.bucket.addContact(Contact(5, 'http://foo/5')) self.bucket.addContact(Contact(6, 'http://foo/6')) self.bucket.addContact(Contact(7, 'http://foo/7')) self.bucket.addContact(Contact(8, 'http://foo/8')) self.bucket.addContact(Contact(9, 'http://foo/9')) self.bucket.addContact(Contact(10, 'http://foo/10')) self.bucket.addContact(Contact(11, 'http://foo/11')) self.bucket.addContact(Contact(12, 'http://foo/12')) self.bucket.addContact(Contact(13, 'http://foo/13')) self.bucket.addContact(Contact(14, 'http://foo/14')) self.bucket.addContact(Contact(15, 'http://foo/15')) self.bucket.addContact(Contact(16, 'http://foo/16')) self.bucket.addContact(Contact(17, 'http://foo/17')) self.bucket.addContact(Contact(18, 'http://foo/18')) self.bucket.addContact(Contact(19, 'http://foo/19')) self.bucket.addContact(Contact(20, 'http://foo/20'))
class Test(unittest.TestCase): def setUp(self): self.bucket = KBucket(1, 20, market_id=1) self.bucket.addContact(Contact(1, 'http://foo/1')) self.bucket.addContact(Contact(2, 'http://foo/2')) self.bucket.addContact(Contact(3, 'http://foo/3')) self.bucket.addContact(Contact(4, 'http://foo/4')) self.bucket.addContact(Contact(5, 'http://foo/5')) self.bucket.addContact(Contact(6, 'http://foo/6')) self.bucket.addContact(Contact(7, 'http://foo/7')) self.bucket.addContact(Contact(8, 'http://foo/8')) self.bucket.addContact(Contact(9, 'http://foo/9')) self.bucket.addContact(Contact(10, 'http://foo/10')) self.bucket.addContact(Contact(11, 'http://foo/11')) self.bucket.addContact(Contact(12, 'http://foo/12')) self.bucket.addContact(Contact(13, 'http://foo/13')) self.bucket.addContact(Contact(14, 'http://foo/14')) self.bucket.addContact(Contact(15, 'http://foo/15')) self.bucket.addContact(Contact(16, 'http://foo/16')) self.bucket.addContact(Contact(17, 'http://foo/17')) self.bucket.addContact(Contact(18, 'http://foo/18')) self.bucket.addContact(Contact(19, 'http://foo/19')) self.bucket.addContact(Contact(20, 'http://foo/20')) def tearDown(self): pass def testCantAddContacts(self): self.assertEqual(len(self.bucket.getContacts()), 20, "Unexpected contact count") def testGetContacts(self): allContacts = self.bucket.getContacts() self.assertEqual(len(allContacts), 20, "Unexpected contact count") # test it will exclude contact c18 = Contact(18, 'http://foo/18') excluding18 = self.bucket.getContacts(excludeContact=c18) # check size of contact list self.assertEqual(19, len(excluding18), "list size is not 19, should be 20") # check it did exclude the contact we asked self.assertEqual(True, c18 not in excluding18, "getContact() did not exclude the contact we asked for") # test it won't choke if I tell it to exclude a contact that is not there yet self.bucket.getContacts(excludeContact=Contact(21, 'http://foo/21')) def testGetContact(self): c14 = Contact(14, 'http://foo/14') self.assertEqual(self.bucket.getContact(14), c14, "did not find requested contact.")