Armistead luego sirvió en Fort Towson, Arkansas, Fort Washita cerca de la frontera de Oklahoma. Sirviendo en la guerra de México, fue nombrado capitán brevet para Contreras y Churubusco, herido en Chapultepec, y fue nombrado mayor brevet para Molino del Rey y Chapultepec.