Added logic to also save a new FileAttachment record DynamicsValue/fa…

…ke-xrm-easy#157
DynamicsValue · Aug 8, 2024 · ff7f54b · ff7f54b
1 parent 4f226f0
commit ff7f54b
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 3 deletions.
diff --git a/src/FakeXrmEasy.Core/FileStorage/Db/InMemoryFileDb.cs b/src/FakeXrmEasy.Core/FileStorage/Db/InMemoryFileDb.cs
@@ -4,6 +4,7 @@
 using System.Linq;
 using FakeXrmEasy.Core.Db;
 using FakeXrmEasy.Core.FileStorage.Db.Exceptions;
+using Microsoft.Xrm.Sdk;
 
 namespace FakeXrmEasy.Core.FileStorage.Db
 {
@@ -12,6 +13,8 @@ internal class InMemoryFileDb : IInMemoryFileDbInternal
         private readonly ConcurrentDictionary<string, FileUploadSession> _uncommittedFileUploads;
         private readonly ConcurrentDictionary<string, FileAttachment> _files;
         private readonly InMemoryDb _db;
+
+        internal const string FILE_ATTACHMENT_TABLE_NAME = "fileattachment";
 
         internal InMemoryFileDb(InMemoryDb db)
         {
@@ -101,8 +104,32 @@ public string CommitFileUploadSession(CommitFileUploadSessionProperties commitPr
                     _files.TryRemove(fileAttachment.Id, out var removedFileAttachment);
                 }
 
+                SaveFileAttachment(fileUploadSession, fileAttachment);
+
                 return fileAttachment.Id;
             }
         }
+
+        private void SaveFileAttachment(FileUploadSession fileUploadSession, FileAttachment fileAttachment)
+        {
+            //Add file attachment
+            if (!_db.ContainsTable(FILE_ATTACHMENT_TABLE_NAME))
+            {
+                InMemoryTable fileAttachmentTable;
+                _db.AddTable(FILE_ATTACHMENT_TABLE_NAME, out fileAttachmentTable);
+            }
+
+            var filesTable = _db.GetTable(FILE_ATTACHMENT_TABLE_NAME);
+            filesTable.Add(new Entity(FILE_ATTACHMENT_TABLE_NAME)
+            {
+                Id = new Guid(fileAttachment.Id)
+            });
+
+            //Update associated record
+            var table = _db.GetTable(fileUploadSession.Properties.Target.LogicalName);
+            var record = table.GetById(fileUploadSession.Properties.Target.Id);
+            record[fileUploadSession.Properties.FileAttributeName] = fileAttachment.Id;
+            record[$"{fileUploadSession.Properties.FileAttributeName}_name"] = fileAttachment.FileName;
+        }
     }
 }
diff --git a/tests/FakeXrmEasy.Core.Tests/FileStorage/Db/InMemoryFileDbTests.cs b/tests/FakeXrmEasy.Core.Tests/FileStorage/Db/InMemoryFileDbTests.cs
@@ -100,6 +100,21 @@ public void Should_init_and_commit_file()
             Assert.Equal(commitProperties.FileName, createdFile.FileName);
             Assert.Equal(commitProperties.MimeType, createdFile.MimeType);
             Assert.Equal(new byte[] { 1, 2, 3, 4 }, createdFile.Content);
+
+            //Uncommited session is removed
+            Assert.Null(_fileDb.GetFileUploadSession(fileContinuationToken));
+
+            //File attribute is updated with file id and file name
+            var entityAfter = _db.GetTable(_entity.LogicalName).GetById(_entity.Id);
+            Assert.Equal(createdFile.Id, entityAfter[_fileUploadProperties.FileAttributeName]);
+            Assert.Equal(createdFile.FileName, entityAfter[$"{_fileUploadProperties.FileAttributeName}_name"]);
+
+            //New file attachment record is created
+            var fileAttachment = _db
+                .GetTable(InMemoryFileDb.FILE_ATTACHMENT_TABLE_NAME)
+                .GetById(new Guid(createdFile.Id));
+
+            Assert.NotNull(fileAttachment);
         }
 
         [Fact]
@@ -148,9 +163,6 @@ public void Should_init_and_upload_multiple_blocks_concurrently_and_commit_file(
                 Assert.Equal(Convert.ToByte(30 + i + 1), createdFile.Content[i * 4 + 2]);
                 Assert.Equal(Convert.ToByte(40 + i + 1), createdFile.Content[i * 4 + 3]);
             }
-
-            //Uncommited session is removed
-            Assert.Null(_fileDb.GetFileUploadSession(fileContinuationToken));
         }
 
         [Fact]