Проверка при выборе цели атаки

Мне нужно заставить юнита атаковать любого своего союзника, находящегося поблизости. Возникает такая проблема: допустим, юнит, который должен атаковать - пехотинец. А два ближайших к нему союзника - стрелок и грифон. С вероятностью 50% от будет атаковать стрелка, но с той же вероятностью он попытается атаковать грифона, воздушную боевую единицу, и тогда система не сработает.

При выборе цели я отсеиваю всех неподходящих юнитов - мёртвых, неуязвимых и. т. д. Можно ли также определить, способен ли выбранный воин атаковать свою цель, как в случае с пехотинцем и грифоном?