def test_ip_pool_remove(self, m_client, m_sleep): """ Test mainline function of ip_pool_remove. """ net1 = IPNetwork("1.2.3.0/24") m_client.remove_ip_pool.side_effect = HostAffinityClaimedError m_pool = IPPool(net1) m_pool.cidr = net1.ip m_client.get_ip_pool_config.return_value = m_pool self.assertRaises(SystemExit, pool.ip_pool_remove, [str(net1)], 4) m_client.get_ip_pool_config.assert_called_once_with(4, net1) m_client.set_ip_pool_config.assert_called_once_with(4, m_pool) self.assertEqual(m_pool.disabled, True) m_client.release_pool_affinities.assert_called_once_with(m_pool) m_client.remove_ip_pool.assert_called_once_with(4, net1.ip)