class TestTagIpsetActor(BaseTestCase): def setUp(self): super(TestTagIpsetActor, self).setUp() self.m_ipset = Mock(spec=Ipset) self.m_ipset.max_elem = 1234 self.m_ipset.set_name = "felix-a_set_name" self.m_ipset.temp_set_name = "felix-a_set_name-tmp" self.tag_ipset = TagIpset("tag-123", "IPv4", max_elem=1024) self.tag_ipset._ipset = self.m_ipset self.m_mgr = Mock() self.tag_ipset._manager = self.m_mgr self.tag_ipset._id = "tag-123" def test_lifecycle(self): self.tag_ipset.replace_members(set(["1.2.3.4"]), async=True) self.step_actor(self.tag_ipset) self.assertEqual( self.m_mgr.on_object_startup_complete.mock_calls, [call("tag-123", self.tag_ipset, async=True)] ) self.tag_ipset.on_unreferenced(async=True) self.step_actor(self.tag_ipset) self.assertEqual( self.m_mgr.on_object_cleanup_complete.mock_calls, [call("tag-123", self.tag_ipset, async=True)] )
def setUp(self): super(TestTagIpsetActor, self).setUp() self.m_ipset = Mock(spec=Ipset) self.m_ipset.max_elem = 1234 self.m_ipset.set_name = "felix-a_set_name" self.m_ipset.temp_set_name = "felix-a_set_name-tmp" self.tag_ipset = TagIpset("tag-123", "IPv4", max_elem=1024) self.tag_ipset._ipset = self.m_ipset self.m_mgr = Mock() self.tag_ipset._manager = self.m_mgr self.tag_ipset._id = "tag-123"