Enhance stub tests to highlight the update and replace behaviors