svn merge -c 1452338 FIXES: HADOOP-9352. Expose UGI.setLoginUser for tests (daryn)
git-svn-id: https://svn.apache.org/repos/asf/hadoop/common/branches/branch-2@1452339 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
a5985e4b0e
commit
221b871f0c
|
@ -1065,6 +1065,8 @@ Release 0.23.7 - UNRELEASED
|
||||||
HADOOP-9336. Allow UGI of current connection to be queried. (Daryn Sharp
|
HADOOP-9336. Allow UGI of current connection to be queried. (Daryn Sharp
|
||||||
via kihwal)
|
via kihwal)
|
||||||
|
|
||||||
|
HADOOP-9352. Expose UGI.setLoginUser for tests (daryn)
|
||||||
|
|
||||||
OPTIMIZATIONS
|
OPTIMIZATIONS
|
||||||
|
|
||||||
BUG FIXES
|
BUG FIXES
|
||||||
|
|
|
@ -65,6 +65,8 @@ import org.apache.hadoop.security.token.TokenIdentifier;
|
||||||
import org.apache.hadoop.util.Shell;
|
import org.apache.hadoop.util.Shell;
|
||||||
import org.apache.hadoop.util.Time;
|
import org.apache.hadoop.util.Time;
|
||||||
|
|
||||||
|
import com.google.common.annotations.VisibleForTesting;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* User and group information for Hadoop.
|
* User and group information for Hadoop.
|
||||||
* This class wraps around a JAAS Subject and provides methods to determine the
|
* This class wraps around a JAAS Subject and provides methods to determine the
|
||||||
|
@ -702,7 +704,8 @@ public class UserGroupInformation {
|
||||||
|
|
||||||
@InterfaceAudience.Private
|
@InterfaceAudience.Private
|
||||||
@InterfaceStability.Unstable
|
@InterfaceStability.Unstable
|
||||||
synchronized static void setLoginUser(UserGroupInformation ugi) {
|
@VisibleForTesting
|
||||||
|
public synchronized static void setLoginUser(UserGroupInformation ugi) {
|
||||||
// if this is to become stable, should probably logout the currently
|
// if this is to become stable, should probably logout the currently
|
||||||
// logged in ugi if it's different
|
// logged in ugi if it's different
|
||||||
loginUser = ugi;
|
loginUser = ugi;
|
||||||
|
|
|
@ -608,4 +608,11 @@ public class TestUserGroupInformation {
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test(timeout=1000)
|
||||||
|
public void testSetLoginUser() throws IOException {
|
||||||
|
UserGroupInformation ugi = UserGroupInformation.createRemoteUser("test-user");
|
||||||
|
UserGroupInformation.setLoginUser(ugi);
|
||||||
|
assertEquals(ugi, UserGroupInformation.getLoginUser());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue