Added classes and methods for testing issue 2127 DiskWriterQueue is n…

…ot reliable when handling back-to-back insertions
BenderNK · Dec 23, 2021 · c5db78e · c5db78e
1 parent 00d28bf
commit c5db78e
Showing 1 changed file with 99 additions and 0 deletions.
diff --git a/LiteDB.Tests/Issues/Issue2127_Tests.cs b/LiteDB.Tests/Issues/Issue2127_Tests.cs
@@ -0,0 +1,99 @@
+using Xunit;
+using System;
+using System.IO;
+using System.Text;
+using System.Threading;
+
+namespace LiteDB.Tests.Issues
+{
+    public class Issue2127_Tests
+    {
+        public class ReproTests
+        {
+            [Fact]
+            public void InsertItemBackToBack_Test()
+            {
+                var databaseDirectory = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "DatabaseLocation");
+                var databasePath = Path.Combine(databaseDirectory, "SampleDatabase.db");
+                var databaseLogPath = Path.Combine(databaseDirectory, "SampleDatabase-log.db");
+
+                // Repeat in a loop to hit that scenario where queue processing thread exists the loop but the task still remains incomplete.
+                for (var i = 0; i < 100; i++)
+                {
+                    if (Directory.Exists(databaseDirectory))
+                        Directory.Delete(databaseDirectory, true);
+
+                    Directory.CreateDirectory(databaseDirectory);
+
+                    using var subject = new ExampleItemRepository(databasePath);
+
+                    var item1 = new ExampleItem
+                    {
+                        Id = Guid.NewGuid(),
+                        SomeProperty = Guid.NewGuid().ToString()
+                    };
+                    var item2 = new ExampleItem
+                    {
+                        Id = Guid.NewGuid(),
+                        SomeProperty = Guid.NewGuid().ToString()
+                    };
+
+                    subject.Insert(item1);
+                    subject.Insert(item2);
+
+                    // Allow items to be processed.
+                    Thread.Sleep(1000);
+
+                    var liteDbPath = databasePath;
+                    var liteDbLogPath = databaseLogPath;
+                    var copiedLiteDbPath = liteDbPath + ".copy";
+                    var copiedLiteDbLogPath = liteDbLogPath + ".copy";
+                    File.Copy(liteDbPath, copiedLiteDbPath);
+                    File.Copy(liteDbLogPath, copiedLiteDbLogPath);
+                    var liteDbContent = File.ReadAllText(copiedLiteDbPath, Encoding.UTF8);
+                    liteDbContent += File.ReadAllText(copiedLiteDbLogPath, Encoding.UTF8);
+
+                    Assert.True(liteDbContent.Contains(item1.SomeProperty, StringComparison.OrdinalIgnoreCase), $"Could not find item 1 property. {item1.SomeProperty}, Iteration: {i}");
+                    Assert.True(liteDbContent.Contains(item2.SomeProperty, StringComparison.OrdinalIgnoreCase), $"Could not find item 2 property. {item2.SomeProperty}, Iteration: {i}");
+                }
+            }
+        }
+
+        public class ExampleItem
+        {
+            [BsonId]
+            public Guid Id { get; set; }
+
+            public string SomeProperty { get; set; }
+        }
+
+        public sealed class ExampleItemRepository : IDisposable
+        {
+            public const string DatabaseFileName = "SampleDb";
+
+            private readonly LiteDatabase _liteDb;
+
+            public ExampleItemRepository(string databasePath)
+            {
+                var connectionString = new ConnectionString
+                {
+                    Filename = databasePath,
+                    Connection = ConnectionType.Direct
+                };
+
+                _liteDb = new LiteDatabase(connectionString);
+            }
+
+            public void Insert(ExampleItem item)
+            {
+                var collection = _liteDb.GetCollection<ExampleItem>();
+                collection.Insert(item);
+            }
+
+            public void Dispose()
+            {
+                _liteDb?.Dispose();
+            }
+        }
+    }
+}