Merge pull request #408 from tonyhallett/fix-updated-line-numbers-out…

…-of-range Fix updated line numbers out of range
FortuneN · Mar 20, 2024 · 2071f51 · 2071f51
2 parents a5fb411 + 26fcd77
commit 2071f51
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/FineCodeCoverageTests/Editor/DynamicCoverage/BufferLineCoverage_Tests.cs b/FineCodeCoverageTests/Editor/DynamicCoverage/BufferLineCoverage_Tests.cs
@@ -235,30 +235,31 @@ public void Should_Create_New_TextLines_When_Coverage_Changed_And_Not_Off_In_App
 
  [TestCase(true)]
  [TestCase(false)]
- public void Should_Update_TrackedLines_When_Text_Buffer_ChangedOnBackground(bool textLinesChanged)
+ public void Should_Update_TrackedLines_When_Text_Buffer_ChangedOnBackground(bool linesInRange)
  {
  SimpleTextInfoSetUp();
 
- var afterSnapshot = new Mock<ITextSnapshot>().Object;
+ var mockAfterSnapshot = new Mock<ITextSnapshot>();
+ mockAfterSnapshot.SetupGet(textSnapshot => textSnapshot.LineCount).Returns(linesInRange ? 100 : 11);
 
  var newSpan = new Span(1, 2);
  var mockTrackedLines = new Mock<ITrackedLines>();
- var changedLineNumbers = textLinesChanged ? new List<int> { 1, 2 } : new List<int>();
- mockTrackedLines.Setup(trackedLines => trackedLines.GetChangedLineNumbers(afterSnapshot, new List<Span> { newSpan }))
+ var changedLineNumbers = new List<int> { 11, 12 };
+ mockTrackedLines.Setup(trackedLines => trackedLines.GetChangedLineNumbers(mockAfterSnapshot.Object, new List<Span> { newSpan }))
  .Returns(changedLineNumbers);
  autoMoqer.Setup<ITrackedLinesFactory, ITrackedLines>(trackedLinesFactory => trackedLinesFactory.Create(It.IsAny<List<ILine>>(), It.IsAny<ITextSnapshot>(), It.IsAny<Language>()))
  .Returns(mockTrackedLines.Object);
 
 
  var bufferLineCoverage = autoMoqer.Create<BufferLineCoverage>();
 
- mockTextBuffer.Raise(textBuffer => textBuffer.ChangedOnBackground += null, CreateTextContentChangedEventArgs(afterSnapshot, newSpan));
+ mockTextBuffer.Raise(textBuffer => textBuffer.ChangedOnBackground += null, CreateTextContentChangedEventArgs(mockAfterSnapshot.Object, newSpan));
 
  autoMoqer.Verify<IEventAggregator>(
  eventAggregator => eventAggregator.SendMessage(
- It.Is<CoverageChangedMessage>(message => message.AppliesTo == "filepath" && message.BufferLineCoverage == bufferLineCoverage && message.ChangedLineNumbers == changedLineNumbers)
+ It.Is<CoverageChangedMessage>(message => message.AppliesTo == "filepath" && message.BufferLineCoverage == bufferLineCoverage && message.ChangedLineNumbers.SequenceEqual(changedLineNumbers))
  , null
- ), Times.Exactly(textLinesChanged ? 1 : 0));
+ ), Times.Exactly(linesInRange ? 1 : 0));
 
  }
 

diff --git a/SharedProject/Editor/DynamicCoverage/Management/BufferLineCoverage.cs b/SharedProject/Editor/DynamicCoverage/Management/BufferLineCoverage.cs
@@ -139,7 +139,8 @@ private void TextBuffer_ChangedOnBackground(object sender, TextContentChangedEve
 
  private void UpdateTrackedLines(TextContentChangedEventArgs e)
  {
- IEnumerable<int> changedLineNumbers = this.trackedLines.GetChangedLineNumbers(e.After, e.Changes.Select(change => change.NewSpan).ToList());
+ IEnumerable<int> changedLineNumbers = this.trackedLines.GetChangedLineNumbers(e.After, e.Changes.Select(change => change.NewSpan).ToList())
+ .Where(changedLine => changedLine >= 0 && changedLine < e.After.LineCount);
  this.SendCoverageChangedMessageIfChanged(changedLineNumbers);
  }