Add agent persistence tests