def garbage_collect_old_service_chains(desired_chains): current_paasta_chains = { chain for chain in iptables.all_chains() if chain.startswith('PAASTA.') } for chain in current_paasta_chains - set(desired_chains): iptables.delete_chain(chain)
def test_all_chains(mock_Table): chain1 = mock.Mock() chain1.name = "INPUT" chain2 = mock.Mock() chain2.name = "OUTPUT" mock_Table.return_value = mock.Mock(chains=[chain1, chain2]) assert iptables.all_chains() == {"INPUT", "OUTPUT"}
def test_all_chains(mock_Table): chain1 = mock.Mock() chain1.name = 'INPUT' chain2 = mock.Mock() chain2.name = 'OUTPUT' mock_Table.return_value = mock.Mock(chains=[ chain1, chain2, ]) assert iptables.all_chains() == {'INPUT', 'OUTPUT'}