Джонстон участвовал в сражении при Контрерас, в сражении при Чурубуско, при Молино-дель-Рей, в штурме Чапультепека и в штурме Мехико, во время которого был ранен.