/*
 * The Alluxio Open Foundation licenses this work under the Apache License, version 2.0
 * (the "License"). You may not use this work except in compliance with the License, which is
 * available at www.apache.org/licenses/LICENSE-2.0
 *
 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied, as more fully set forth in the License.
 *
 * See the NOTICE file distributed with this work for information regarding copyright ownership.
 */

package alluxio.security;

import alluxio.Configuration;
import alluxio.PropertyKey;

import org.powermock.reflect.Whitebox;

import java.io.IOException;

/**
 * Utility methods for the tests using {@link LoginUser}.
 */
public final class LoginUserTestUtils {

  private LoginUserTestUtils() {} // prevent instantiation

  /**
   * Resets the singleton {@link LoginUser} to null.
   */
  public static void resetLoginUser() {
    synchronized (LoginUser.class) {
      Whitebox.setInternalState(LoginUser.class, "sLoginUser", (String) null);
    }
  }

  /**
   * Resets the {@link LoginUser} and re-login with new user.
   *
   * @param user the new user
   * @throws IOException if login fails
   */
  public static void resetLoginUser(String user) throws IOException {
    synchronized (LoginUser.class) {
      resetLoginUser();
      Configuration.set(PropertyKey.SECURITY_LOGIN_USERNAME, user);
      LoginUser.get();
    }
  }
}

@@ -1327,114 +1327,79 @@
     
-resetLoginUser();%0A      Configuration.set(PropertyKey.SECURITY_LOGIN_USERNAME, user);%0A      LoginUser.get(
+Whitebox.setInternalState(LoginUser.class, %22sLoginUser%22, new User(user)
 );%0A 

Update LoginUserTestUtils: setting configuration without restoring it is bad. Use whitebox directly to modify the value