Add upgrade test in E2E test