public void Should_be_possible_finalize_the_collectRequest_if_the_number_of_execution_exceeds_of_limits()
{
IDocumentSession fakeSession = provider.GetSession();
CollectRequest collectRequest = this.GetCollectRequest();
oval_definitions definitions = new OvalDocumentLoader().GetFakeOvalDefinitions("definitionsWithLocalVariable.xml");
var objList = new List <ObjectType>();
objList.Add(new family_object()
{
id = "oval:org.mitre.oval:obj:1000"
});
objList.Add(new file_object()
{
id = "oval:org.mitre.oval:obj:5000"
});
definitions.objects = objList.ToArray();
var newDefinitiondoc = new DefinitionDocument()
{
OriginalId = "01", Text = definitions.GetDefinitionsXml()
};
session.Store(newDefinitiondoc);
collectRequest.OvalDefinitionsId = newDefinitiondoc.Oid;
CollectExecutionManagerFactory factory = new CollectExecutionManagerFactory(collectRequest, session);
List <string> registryObjectsThatNotProcessInFirstTime = new List <string>();
List <string> familyObjectsThatNotProcessInFirstTime = new List <string>();
CollectExecutionManager executionManager = factory.CreateExecutionManagerForTheSuccessScenario(this.GetResultForRegistry(collectRequest, registryObjectsThatNotProcessInFirstTime),
this.GetResultForFamily(collectRequest, familyObjectsThatNotProcessInFirstTime));
session.Store(collectRequest);
var newExecution1 = new CollectExecution()
{
RequestId = collectRequest.Oid
};
session.Store(newExecution1);
var newExecution2 = new CollectExecution()
{
RequestId = collectRequest.Oid
};
session.Store(newExecution2);
var newExecution3 = new CollectExecution()
{
RequestId = collectRequest.Oid
};
session.Store(newExecution3);
executionManager.ExecuteCollect(fakeSession, collectRequest, FamilyEnumeration.windows);
session.SaveChanges();
CollectRequest collectRequestAfterExecute = session.Load <CollectRequest>(collectRequest.Oid.ToString());
Assert.IsNotNull(collectRequest);
Assert.AreEqual(4, collectRequestAfterExecute.GetNumberOfExecutions(session));
Assert.IsTrue(collectRequestAfterExecute.isClosed());
}
public void Should_be_possible_to_get_a_number_of_times_that_was_executed_the_collectRequest()
{
CollectRequestFactory factory = new CollectRequestFactory();
IDocumentSession fakeSession = this.GetSession();
CollectRequest collectRequest = factory.CreateCollectRequest(fakeSession).Item2;
Assert.IsTrue(collectRequest.GetNumberOfExecutions(fakeSession) == 0, "this request collect still not executed");
//add two executions
CollectExecution firstExecution = factory.CreateCollect(fakeSession);
firstExecution.RequestId = collectRequest.Oid;
firstExecution.ProbeExecutions.Add(factory.CreateProbeExecution(fakeSession, "registry"));
firstExecution.ProbeExecutions.Add(factory.CreateProbeExecution(fakeSession, "family"));
CollectExecution secondExecution = factory.CreateCollect(fakeSession);
secondExecution.RequestId = collectRequest.Oid;
secondExecution.ProbeExecutions.Add(factory.CreateProbeExecution(fakeSession, "registry"));
fakeSession.SaveChanges();
Assert.AreEqual(2, collectRequest.GetNumberOfExecutions(fakeSession));
}
private void EndCollect(IDocumentSession session, CollectRequest collectRequest, CollectExecution collectExecution)
{
if (collectTimeOut.IsExceededTheMaxOfExecutionsDefined(collectRequest.GetNumberOfExecutions(session)))
{
collectRequest.Close();
}
else
{
if (!collectRequest.isClosed())
{
if (!collectExecution.ExistsExecutionsWithError())
{
collectRequest.TryClose(session);
}
else
{
collectRequest.Reopen();
}
}
}
collectExecution.SetDateEndCollect();
session.SaveChanges();
}
private void CheckTheDefaulStateOfRequestCollectAfterOneExecution(CollectRequest requestCollectAfterExecute)
{
Assert.IsNotNull(requestCollectAfterExecute);
Assert.AreEqual(1, requestCollectAfterExecute.GetNumberOfExecutions(session));
}