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)); }