public void When_it_was_not_possible_to_establish_a_connection_to_target_the_status_of_collect_result_must_be_error() { IDocumentSession fakeSession = provider.GetSession(); var fakeRequestCollect = this.GetCollectRequest(); var definitions = new OvalDocumentLoader().GetFakeOvalDefinitions("definitionsWithLocalVariable.xml"); var newDefinitiondoc = new DefinitionDocument() { OriginalId = "01", Text = definitions.GetDefinitionsXml() }; session.Store(newDefinitiondoc); fakeRequestCollect.OvalDefinitionsId = newDefinitiondoc.Oid; var factory = new CollectExecutionManagerFactory(fakeRequestCollect, session); var executionManager = factory .CreateExecutionManagerWithCannotConnectToHostExceptionScenario( this.GetResultForRegistry(fakeRequestCollect, new List <String>()), this.GetResultForFamily(fakeRequestCollect, new string[] { "oval:org.mitre.oval:obj:1000" }.ToList()), true); //fakeRequestCollect.Collects.Add(new CollectExecution(session)); //fakeRequestCollect.Collects.Add(new CollectExecution(session)); //fakeRequestCollect.Collects.Add(new CollectExecution(session)); session.SaveChanges(); executionManager.ExecuteCollect(fakeSession, fakeRequestCollect, FamilyEnumeration.windows); var requestCollectAfterExecution = session.Load <CollectRequest>(fakeRequestCollect.Oid); Assert.AreEqual(CollectStatus.Error, requestCollectAfterExecution.Result.Status); }
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()); }