Migrate remaining session tests