private void RetainEvents(IEnumerable <CustomEventWireModel> customEvents)
{
customEvents = customEvents.ToList();
_agentHealthReporter.ReportCustomEventsRecollected(customEvents.Count());
foreach (var customEvent in customEvents)
{
if (customEvent != null)
{
_customEvents.Add(new PrioritizedNode <CustomEventWireModel>(customEvent));
}
}
}
public void Events_are_retained_after_harvest_if_response_equals_retain()
{
// Arrange
var sentEventCount = int.MinValue;
Mock.Arrange(() => _dataTransportService.Send(Arg.IsAny <IEnumerable <CustomEventWireModel> >()))
.Returns <IEnumerable <CustomEventWireModel> >(events =>
{
sentEventCount = events.Count();
return(DataTransportResponseStatus.Retain);
});
_customEventAggregator.Collect(new CustomEventWireModel(0.1f, GetCustomEventAttribs()));
_customEventAggregator.Collect(new CustomEventWireModel(0.2f, GetCustomEventAttribs()));
_harvestAction();
sentEventCount = int.MinValue; // reset
// Act
_harvestAction();
// Assert
Assert.AreEqual(2, sentEventCount);
Mock.Assert(() => _agentHealthReporter.ReportCustomEventsRecollected(2));
}
