public void TestDeadlockWhenRemovingAnActiveDataSource()
{
var logFile = new Mock <ILogSource>();
var dataSource = new Mock <IDataSource>();
dataSource.Setup(x => x.UnfilteredLogSource).Returns(logFile.Object);
var collection = new BookmarkCollection(_bookmarks.Object, TimeSpan.Zero);
logFile.Setup(x => x.RemoveListener(It.IsAny <ILogSourceListener>()))
.Callback((ILogSourceListener unused) =>
{
// In order to produce the deadlock, we have to simulate what's happening in reality.
// Any ILogFile implementation will hold a lock both while invoking listeners and while
// Removing them. Waiting on the OnLogFileModified() call simulates exactly that...
var task = new Task(() =>
{
collection.OnLogFileModified(logFile.Object, LogSourceModification.Removed(0, 10));
}, TaskCreationOptions.LongRunning);
task.Start();
task.Wait();
});
collection.AddDataSource(dataSource.Object);
var removeTask = Task.Factory.StartNew(() => collection.RemoveDataSource(dataSource.Object));
removeTask.Wait(TimeSpan.FromSeconds(1)).Should().BeTrue("because RemoveDataSource should not block at all");
logFile.Verify(x => x.RemoveListener(collection), Times.Once);
}
private IDataSource AddDataSource(DataSource settings)
{
lock (_syncRoot)
{
IDataSource dataSource;
if (!string.IsNullOrEmpty(settings.LogFileFolderPath))
{
dataSource = new FolderDataSource(_taskScheduler, _logFileFactory, _filesystem, settings);
}
else if (!string.IsNullOrEmpty(settings.File))
{
dataSource = new SingleDataSource(_logFileFactory, _taskScheduler, settings, _maximumWaitTime);
}
else
{
if (settings.DisplayName == null)
{
settings.DisplayName = "Merged Data Source";
}
dataSource = new MergedDataSource(_taskScheduler, settings, _maximumWaitTime);
}
_dataSources.Add(dataSource);
_dataSourceIds.Add(dataSource.Id);
_bookmarks.AddDataSource(dataSource);
return(dataSource);
}
}
public void TestAddSameBookmarkTwice()
{
var collection = new BookmarkCollection(_bookmarks.Object, TimeSpan.Zero);
collection.AddDataSource(_dataSource.Object);
_logSource.AddEntry("", LevelFlags.Error);
_logSource.AddEntry("", LevelFlags.Error);
var bookmark = collection.TryAddBookmark(_dataSource.Object, 1);
collection.TryAddBookmark(_dataSource.Object, 1).Should().BeNull();
collection.Count.Should().Be(1);
collection.Bookmarks.Should().BeEquivalentTo(new object[] { bookmark });
}
public void TestAddOneBookmark()
{
var collection = new BookmarkCollection(TimeSpan.Zero);
collection.AddDataSource(_dataSource.Object);
_logFile.AddEntry("", LevelFlags.Error);
_logFile.AddEntry("", LevelFlags.Error);
var bookmark = collection.TryAddBookmark(_dataSource.Object, 1);
bookmark.Should().NotBeNull();
bookmark.Index.Should().Be(new LogLineIndex(1));
collection.Count.Should().Be(1);
collection.Bookmarks.Should().BeEquivalentTo(new object[] { bookmark });
}
public void TestAddTwoBookmarks()
{
var collection = new BookmarkCollection(_bookmarks.Object, TimeSpan.Zero);
collection.AddDataSource(_dataSource.Object);
_logSource.AddEntry("", LevelFlags.Error);
_logSource.AddEntry("", LevelFlags.Error);
var bookmark1 = collection.TryAddBookmark(_dataSource.Object, 1);
var bookmark2 = collection.TryAddBookmark(_dataSource.Object, 0);
bookmark2.Should().NotBeNull();
bookmark2.Index.Should().Be(new LogLineIndex(0));
collection.Count.Should().Be(2);
collection.Bookmarks.Should().BeEquivalentTo(new object[] { bookmark1, bookmark2 });
}
public void TestRemoveBookmark()
{
var collection = new BookmarkCollection(_bookmarks.Object, TimeSpan.Zero);
collection.AddDataSource(_dataSource.Object);
_logSource.AddEntry("", LevelFlags.Error);
_logSource.AddEntry("", LevelFlags.Error);
var bookmark = collection.TryAddBookmark(_dataSource.Object, 1);
_bookmarks.Verify(x => x.SaveAsync(), Times.Once);
collection.RemoveBookmark(bookmark);
_bookmarks.Verify(x => x.SaveAsync(), Times.Exactly(2));
collection.Bookmarks.Should().BeEmpty();
}
public void TestConstruction2()
{
var bookmarks = new List <BookmarkSettings>();
bookmarks.Add(new BookmarkSettings(DataSourceId.CreateNew(), new LogLineIndex(42)));
bookmarks.Add(new BookmarkSettings(DataSourceId.CreateNew(), new LogLineIndex(101)));
_bookmarks.Setup(x => x.All).Returns(bookmarks);
var collection = new BookmarkCollection(_bookmarks.Object, TimeSpan.Zero);
collection.Bookmarks.Should().BeEmpty();
_dataSource.Setup(x => x.Id).Returns(bookmarks[1].DataSourceId);
collection.AddDataSource(_dataSource.Object);
collection.Bookmarks.Should().HaveCount(1);
collection.Bookmarks[0].DataSource.Should().BeSameAs(_dataSource.Object);
collection.Bookmarks[0].Index.Should().Be(new LogLineIndex(101));
}
public void TestAddDataSourceDuplicateBookmarks()
{
var dataSourceId = DataSourceId.CreateNew();
_bookmarks.Setup(x => x.All).Returns(new[]
{
new BookmarkSettings(dataSourceId, new LogLineIndex(42)),
new BookmarkSettings(dataSourceId, new LogLineIndex(42)),
});
var collection = new BookmarkCollection(_bookmarks.Object, TimeSpan.Zero);
var dataSource = new Mock <IDataSource>();
dataSource.Setup(x => x.UnfilteredLogSource).Returns(new InMemoryLogSource());
dataSource.Setup(x => x.Id).Returns(dataSourceId);
new Action(() => collection.AddDataSource(dataSource.Object)).Should().NotThrow();
collection.Bookmarks.Should().Equal(new object[]
{
new Bookmark(dataSource.Object, new LogLineIndex(42))
}, "because even though there are two bookmarks in the settings object, they describe the same log line and thus only one bookmark should have been added in the end");
}
public void TestClearBookmarks()
{
var collection = new BookmarkCollection(_bookmarks.Object, TimeSpan.Zero);
collection.AddDataSource(_dataSource.Object);
_logSource.AddEntry("", LevelFlags.Error);
_logSource.AddEntry("", LevelFlags.Error);
var bookmark1 = collection.TryAddBookmark(_dataSource.Object, 0);
var bookmark2 = collection.TryAddBookmark(_dataSource.Object, 1);
collection.Count.Should().Be(2);
collection.Bookmarks.Should().Equal(new[] { bookmark1, bookmark2 });
_bookmarks.Verify(x => x.SaveAsync(), Times.Exactly(2));
collection.Clear();
collection.Count.Should().Be(0);
collection.Bookmarks.Should().BeEmpty();
_bookmarks.Verify(x => x.SaveAsync(), Times.Exactly(3));
}
