/// <exception cref="System.Exception"/> public virtual void TestGetGroups() { Org.Mockito.Mockito.When(admin.GetGroupsForUser(Matchers.Eq("admin"))).ThenReturn (new string[] { "group1", "group2" }); TextWriter origOut = System.Console.Out; TextWriter @out = Org.Mockito.Mockito.Mock <TextWriter>(); Runtime.SetOut(@out); try { string[] args = new string[] { "-getGroups", "admin" }; NUnit.Framework.Assert.AreEqual(0, rmAdminCLI.Run(args)); Org.Mockito.Mockito.Verify(admin).GetGroupsForUser(Matchers.Eq("admin")); Org.Mockito.Mockito.Verify(@out).WriteLine(Matchers.ArgThat(new _ArgumentMatcher_190 ())); } finally { Runtime.SetOut(origOut); } }
/// <exception cref="System.IO.IOException"/> private int GetGroups(string[] usernames) { // Get groups users belongs to ResourceManagerAdministrationProtocol adminProtocol = CreateAdminProtocol(); if (usernames.Length == 0) { usernames = new string[] { UserGroupInformation.GetCurrentUser().GetUserName() }; } foreach (string username in usernames) { StringBuilder sb = new StringBuilder(); sb.Append(username + " :"); foreach (string group in adminProtocol.GetGroupsForUser(username)) { sb.Append(" "); sb.Append(group); } System.Console.Out.WriteLine(sb); } return(0); }
GetGroupsForUser(RpcController controller, YarnServerResourceManagerServiceProtos.GetGroupsForUserRequestProto request) { string user = request.GetUser(); try { string[] groups = real.GetGroupsForUser(user); YarnServerResourceManagerServiceProtos.GetGroupsForUserResponseProto.Builder responseBuilder = YarnServerResourceManagerServiceProtos.GetGroupsForUserResponseProto.NewBuilder (); foreach (string group in groups) { responseBuilder.AddGroups(group); } return((YarnServerResourceManagerServiceProtos.GetGroupsForUserResponseProto)responseBuilder .Build()); } catch (IOException e) { throw new ServiceException(e); } }