Fix exception on calling DBCDStorage.Save() twice

wowdev · Aug 23, 2024 · 7e6de71 · 7e6de71
1 parent cab0d12
commit 7e6de71
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/DBCD.Tests/WritingTest.cs b/DBCD.Tests/WritingTest.cs
@@ -66,6 +66,15 @@ public void TestWritingNewRowDb2WithArrayField()
             Assert.AreEqual(116146, outputStorage.Count);
         }
 
+        [TestMethod]
+        public void TestSavingSameStorageTwice()
+        {
+            DBCD dbcd = new(wagoDBCProvider, githubDBDProvider);
+            IDBCDStorage storage = dbcd.Load("AlliedRace", "9.2.7.45745");
+            storage.Save(Path.Join(OutputPath, "AlliedRace.db2"));
+            storage.Save(Path.Join(OutputPath, "AlliedRace.db2"));
+        }
+
         [TestMethod]
         public void TestWritingAllDB2s()
         {

diff --git a/DBCD/DBCDStorage.cs b/DBCD/DBCDStorage.cs
@@ -190,7 +190,8 @@ public void Save(string filename)
             foreach (var (id, record) in new SortedDictionary<int, DBCDRow>(this))
                 storage.Add(id, record.AsType<T>());
 #endif
-            storage?.Save(filename);
+            storage.Save(filename);
+            storage.Clear();
         }
 
         public DBCDRow ConstructRow(int index) {