public virtual void TestGetClusterNodeLabelsWithLocalAccess()
{
YarnClient client = Org.Mockito.Mockito.Mock <YarnClient>();
Org.Mockito.Mockito.When(client.GetClusterNodeLabels()).ThenReturn(ImmutableSet.Of
("remote1", "remote2"));
ClusterCLI cli = new ClusterCLI();
cli.SetClient(client);
cli.SetSysOutPrintStream(sysOut);
cli.SetSysErrPrintStream(sysErr);
ClusterCLI.localNodeLabelsManager = Org.Mockito.Mockito.Mock <CommonNodeLabelsManager
>();
Org.Mockito.Mockito.When(ClusterCLI.localNodeLabelsManager.GetClusterNodeLabels()
).ThenReturn(ImmutableSet.Of("local1", "local2"));
int rc = cli.Run(new string[] { ClusterCLI.Cmd, "-" + ClusterCLI.ListLabelsCmd, "-"
+ ClusterCLI.DirectlyAccessNodeLabelStore });
NUnit.Framework.Assert.AreEqual(0, rc);
ByteArrayOutputStream baos = new ByteArrayOutputStream();
PrintWriter pw = new PrintWriter(baos);
// it should return local* instead of remote*
pw.Write("Node Labels: local1,local2");
pw.Close();
Org.Mockito.Mockito.Verify(sysOut).WriteLine(baos.ToString("UTF-8"));
}
public virtual void TestGetEmptyClusterNodeLabels()
{
YarnClient client = Org.Mockito.Mockito.Mock <YarnClient>();
Org.Mockito.Mockito.When(client.GetClusterNodeLabels()).ThenReturn(new HashSet <string
>());
ClusterCLI cli = new ClusterCLI();
cli.SetClient(client);
cli.SetSysOutPrintStream(sysOut);
cli.SetSysErrPrintStream(sysErr);
int rc = cli.Run(new string[] { ClusterCLI.Cmd, "-" + ClusterCLI.ListLabelsCmd });
NUnit.Framework.Assert.AreEqual(0, rc);
ByteArrayOutputStream baos = new ByteArrayOutputStream();
PrintWriter pw = new PrintWriter(baos);
pw.Write("Node Labels: ");
pw.Close();
Org.Mockito.Mockito.Verify(sysOut).WriteLine(baos.ToString("UTF-8"));
}
/// <exception cref="Org.Apache.Hadoop.Yarn.Exceptions.YarnException"/>
/// <exception cref="System.IO.IOException"/>
public override ICollection <string> GetClusterNodeLabels()
{
return(client.GetClusterNodeLabels());
}