The SERVER-79735 refactor removed two lines which assigned the current shard primary to a local variable which was never used. The purpose of these lines was actually to wait for the new primary before continuing with the test, we should bring this behavior back.