public virtual void TestDefaultSingleQueueMetrics()
{
string queueName = "single";
string user = "******";
QueueMetrics metrics = QueueMetrics.ForQueue(ms, queueName, null, false, conf);
MetricsSource queueSource = QueueSource(ms, queueName);
AppSchedulingInfo app = MockApp(user);
metrics.SubmitApp(user);
MetricsSource userSource = UserSource(ms, queueName, user);
CheckApps(queueSource, 1, 0, 0, 0, 0, 0, true);
metrics.SubmitAppAttempt(user);
CheckApps(queueSource, 1, 1, 0, 0, 0, 0, true);
metrics.SetAvailableResourcesToQueue(Resources.CreateResource(100 * Gb, 100));
metrics.IncrPendingResources(user, 5, Resources.CreateResource(3 * Gb, 3));
// Available resources is set externally, as it depends on dynamic
// configurable cluster/queue resources
CheckResources(queueSource, 0, 0, 0, 0, 0, 100 * Gb, 100, 15 * Gb, 15, 5, 0, 0, 0
);
metrics.RunAppAttempt(app.GetApplicationId(), user);
CheckApps(queueSource, 1, 0, 1, 0, 0, 0, true);
metrics.AllocateResources(user, 3, Resources.CreateResource(2 * Gb, 2), true);
CheckResources(queueSource, 6 * Gb, 6, 3, 3, 0, 100 * Gb, 100, 9 * Gb, 9, 2, 0, 0
, 0);
metrics.ReleaseResources(user, 1, Resources.CreateResource(2 * Gb, 2));
CheckResources(queueSource, 4 * Gb, 4, 2, 3, 1, 100 * Gb, 100, 9 * Gb, 9, 2, 0, 0
, 0);
metrics.FinishAppAttempt(app.GetApplicationId(), app.IsPending(), app.GetUser());
CheckApps(queueSource, 1, 0, 0, 0, 0, 0, true);
metrics.FinishApp(user, RMAppState.Finished);
CheckApps(queueSource, 1, 0, 0, 1, 0, 0, true);
NUnit.Framework.Assert.IsNull(userSource);
}
public virtual void TestMetricsCache()
{
MetricsSystem ms = new MetricsSystemImpl("cache");
ms.Start();
try
{
string p1 = "root1";
string leafQueueName = "root1.leaf";
QueueMetrics p1Metrics = QueueMetrics.ForQueue(ms, p1, null, true, conf);
Queue parentQueue1 = MockitoMaker.Make(MockitoMaker.Stub <Queue>().Returning(p1Metrics
).from.GetMetrics());
QueueMetrics metrics = QueueMetrics.ForQueue(ms, leafQueueName, parentQueue1, true
, conf);
NUnit.Framework.Assert.IsNotNull("QueueMetrics for A shoudn't be null", metrics);
// Re-register to check for cache hit, shouldn't blow up metrics-system...
// also, verify parent-metrics
QueueMetrics alterMetrics = QueueMetrics.ForQueue(ms, leafQueueName, parentQueue1
, true, conf);
NUnit.Framework.Assert.IsNotNull("QueueMetrics for alterMetrics shoudn't be null"
, alterMetrics);
}
finally
{
ms.Shutdown();
}
}
private Queue CreateQueue(string name, Queue parent)
{
QueueMetrics metrics = QueueMetrics.ForQueue(name, parent, false, conf);
ActiveUsersManager activeUsersManager = new ActiveUsersManager(metrics);
Queue queue = Org.Mockito.Mockito.Mock <Queue>();
Org.Mockito.Mockito.When(queue.GetMetrics()).ThenReturn(metrics);
Org.Mockito.Mockito.When(queue.GetActiveUsersManager()).ThenReturn(activeUsersManager
);
return(queue);
}
public virtual void TestQueueAppMetricsForMultipleFailures()
{
string queueName = "single";
string user = "******";
QueueMetrics metrics = QueueMetrics.ForQueue(ms, queueName, null, false, new Configuration
());
MetricsSource queueSource = QueueSource(ms, queueName);
AppSchedulingInfo app = MockApp(user);
metrics.SubmitApp(user);
MetricsSource userSource = UserSource(ms, queueName, user);
CheckApps(queueSource, 1, 0, 0, 0, 0, 0, true);
metrics.SubmitAppAttempt(user);
CheckApps(queueSource, 1, 1, 0, 0, 0, 0, true);
metrics.RunAppAttempt(app.GetApplicationId(), user);
CheckApps(queueSource, 1, 0, 1, 0, 0, 0, true);
metrics.FinishAppAttempt(app.GetApplicationId(), app.IsPending(), app.GetUser());
CheckApps(queueSource, 1, 0, 0, 0, 0, 0, true);
// As the application has failed, framework retries the same application
// based on configuration
metrics.SubmitAppAttempt(user);
CheckApps(queueSource, 1, 1, 0, 0, 0, 0, true);
metrics.RunAppAttempt(app.GetApplicationId(), user);
CheckApps(queueSource, 1, 0, 1, 0, 0, 0, true);
// Suppose say application has failed this time as well.
metrics.FinishAppAttempt(app.GetApplicationId(), app.IsPending(), app.GetUser());
CheckApps(queueSource, 1, 0, 0, 0, 0, 0, true);
// As the application has failed, framework retries the same application
// based on configuration
metrics.SubmitAppAttempt(user);
CheckApps(queueSource, 1, 1, 0, 0, 0, 0, true);
metrics.RunAppAttempt(app.GetApplicationId(), user);
CheckApps(queueSource, 1, 0, 1, 0, 0, 0, true);
// Suppose say application has failed, and there's no more retries.
metrics.FinishAppAttempt(app.GetApplicationId(), app.IsPending(), app.GetUser());
CheckApps(queueSource, 1, 0, 0, 0, 0, 0, true);
metrics.FinishApp(user, RMAppState.Failed);
CheckApps(queueSource, 1, 0, 0, 0, 1, 0, true);
NUnit.Framework.Assert.IsNull(userSource);
}
public virtual void TestCollectAllMetrics()
{
string queueName = "single";
QueueMetrics.ForQueue(ms, queueName, null, false, conf);
MetricsSource queueSource = QueueSource(ms, queueName);
CheckApps(queueSource, 0, 0, 0, 0, 0, 0, true);
try
{
// do not collect all metrics
CheckApps(queueSource, 0, 0, 0, 0, 0, 0, false);
NUnit.Framework.Assert.Fail();
}
catch (Exception e)
{
NUnit.Framework.Assert.IsTrue(e.Message.Contains("Expected exactly one metric for name "
));
}
// collect all metrics
CheckApps(queueSource, 0, 0, 0, 0, 0, 0, true);
}
public virtual void TestTwoLevelWithUserMetrics()
{
string parentQueueName = "root";
string leafQueueName = "root.leaf";
string user = "******";
QueueMetrics parentMetrics = QueueMetrics.ForQueue(ms, parentQueueName, null, true
, conf);
Queue parentQueue = MockitoMaker.Make(MockitoMaker.Stub <Queue>().Returning(parentMetrics
).from.GetMetrics());
QueueMetrics metrics = QueueMetrics.ForQueue(ms, leafQueueName, parentQueue, true
, conf);
MetricsSource parentQueueSource = QueueSource(ms, parentQueueName);
MetricsSource queueSource = QueueSource(ms, leafQueueName);
AppSchedulingInfo app = MockApp(user);
metrics.SubmitApp(user);
MetricsSource userSource = UserSource(ms, leafQueueName, user);
MetricsSource parentUserSource = UserSource(ms, parentQueueName, user);
CheckApps(queueSource, 1, 0, 0, 0, 0, 0, true);
CheckApps(parentQueueSource, 1, 0, 0, 0, 0, 0, true);
CheckApps(userSource, 1, 0, 0, 0, 0, 0, true);
CheckApps(parentUserSource, 1, 0, 0, 0, 0, 0, true);
metrics.SubmitAppAttempt(user);
CheckApps(queueSource, 1, 1, 0, 0, 0, 0, true);
CheckApps(parentQueueSource, 1, 1, 0, 0, 0, 0, true);
CheckApps(userSource, 1, 1, 0, 0, 0, 0, true);
CheckApps(parentUserSource, 1, 1, 0, 0, 0, 0, true);
parentMetrics.SetAvailableResourcesToQueue(Resources.CreateResource(100 * Gb, 100
));
metrics.SetAvailableResourcesToQueue(Resources.CreateResource(100 * Gb, 100));
parentMetrics.SetAvailableResourcesToUser(user, Resources.CreateResource(10 * Gb,
10));
metrics.SetAvailableResourcesToUser(user, Resources.CreateResource(10 * Gb, 10));
metrics.IncrPendingResources(user, 5, Resources.CreateResource(3 * Gb, 3));
CheckResources(queueSource, 0, 0, 0, 0, 0, 100 * Gb, 100, 15 * Gb, 15, 5, 0, 0, 0
);
CheckResources(parentQueueSource, 0, 0, 0, 0, 0, 100 * Gb, 100, 15 * Gb, 15, 5, 0
, 0, 0);
CheckResources(userSource, 0, 0, 0, 0, 0, 10 * Gb, 10, 15 * Gb, 15, 5, 0, 0, 0);
CheckResources(parentUserSource, 0, 0, 0, 0, 0, 10 * Gb, 10, 15 * Gb, 15, 5, 0, 0
, 0);
metrics.RunAppAttempt(app.GetApplicationId(), user);
CheckApps(queueSource, 1, 0, 1, 0, 0, 0, true);
CheckApps(userSource, 1, 0, 1, 0, 0, 0, true);
metrics.AllocateResources(user, 3, Resources.CreateResource(2 * Gb, 2), true);
metrics.ReserveResource(user, Resources.CreateResource(3 * Gb, 3));
// Available resources is set externally, as it depends on dynamic
// configurable cluster/queue resources
CheckResources(queueSource, 6 * Gb, 6, 3, 3, 0, 100 * Gb, 100, 9 * Gb, 9, 2, 3 *
Gb, 3, 1);
CheckResources(parentQueueSource, 6 * Gb, 6, 3, 3, 0, 100 * Gb, 100, 9 * Gb, 9, 2
, 3 * Gb, 3, 1);
CheckResources(userSource, 6 * Gb, 6, 3, 3, 0, 10 * Gb, 10, 9 * Gb, 9, 2, 3 * Gb,
3, 1);
CheckResources(parentUserSource, 6 * Gb, 6, 3, 3, 0, 10 * Gb, 10, 9 * Gb, 9, 2, 3
* Gb, 3, 1);
metrics.ReleaseResources(user, 1, Resources.CreateResource(2 * Gb, 2));
metrics.UnreserveResource(user, Resources.CreateResource(3 * Gb, 3));
CheckResources(queueSource, 4 * Gb, 4, 2, 3, 1, 100 * Gb, 100, 9 * Gb, 9, 2, 0, 0
, 0);
CheckResources(parentQueueSource, 4 * Gb, 4, 2, 3, 1, 100 * Gb, 100, 9 * Gb, 9, 2
, 0, 0, 0);
CheckResources(userSource, 4 * Gb, 4, 2, 3, 1, 10 * Gb, 10, 9 * Gb, 9, 2, 0, 0, 0
);
CheckResources(parentUserSource, 4 * Gb, 4, 2, 3, 1, 10 * Gb, 10, 9 * Gb, 9, 2, 0
, 0, 0);
metrics.FinishAppAttempt(app.GetApplicationId(), app.IsPending(), app.GetUser());
CheckApps(queueSource, 1, 0, 0, 0, 0, 0, true);
CheckApps(parentQueueSource, 1, 0, 0, 0, 0, 0, true);
CheckApps(userSource, 1, 0, 0, 0, 0, 0, true);
CheckApps(parentUserSource, 1, 0, 0, 0, 0, 0, true);
metrics.FinishApp(user, RMAppState.Finished);
CheckApps(queueSource, 1, 0, 0, 1, 0, 0, true);
CheckApps(parentQueueSource, 1, 0, 0, 1, 0, 0, true);
CheckApps(userSource, 1, 0, 0, 1, 0, 0, true);
CheckApps(parentUserSource, 1, 0, 0, 1, 0, 0, true);
}
