class MemberlistManagerTest(unittest.TestCase):
def setUp(self):
self.manager = MemberlistManager()
def testAddMember(self):
self._add_member('localhost', 1, 2)
memberlist = self.manager.get_copy()
self.assertEquals(len(memberlist), 1)
def testUpdateMember(self):
self._add_member('localhost', 1, 2)
self._add_member('localhost2', 1, 2)
self._add_member('localhost2', 2, 3)
memberlist = self.manager.get_copy()
self.assertEquals(len(memberlist), 2)
self.assertEquals(memberlist.get_by_address('localhost2').seq_num, 2)
def testDeleteMember(self):
self._add_member('localhost1', 1, 2)
self._add_member('localhost2', 2, 3)
self._delete_member('localhost2')
memberlist = self.manager.get_copy()
self.assertEquals(len(memberlist), 1)
self.assertEquals(memberlist.get_by_address('localhost1').seq_num, 1)
def _add_member(self, address, seq_num=1, timestamp=1, state=MemberState.Undefined):
def update_func(updater):
updater.add_or_update(Member(address, seq_num, timestamp, state))
self.manager.update(update_func)
def _delete_member(self, address):
def update_func(updater):
updater.delete(address)
self.manager.update(update_func)
def setUp(self):
self.manager = MemberlistManager()
