Add new robustness extension tests