def test_get_ha_nodes(self, mock_config, mock_unit_get, mock_get_host_ip, mock_get_ipv6_addr): mock_get_host_ip.side_effect = lambda host: host def unit_get(key): return {'private-address': '10.0.0.1'}.get(key) mock_unit_get.side_effect = unit_get def config(key): return {'prefer-ipv6': False}.get(key) mock_config.side_effect = config nodes = utils.get_ha_nodes() self.assertEqual(nodes, {'hanode/0-cid': '10.0.0.1', 'hanode/1-cid': '10.0.0.2'}) self.assertTrue(mock_get_host_ip.called) self.assertFalse(mock_get_ipv6_addr.called)