Ensure entity type is registered for EntityEntry

Turnerj · Turnerj · commit a163b11e4f49 · 2021-10-31T15:32:21.000+10:30
Fixes #261
diff --git a/src/MongoFramework/Infrastructure/EntityEntry.cs b/src/MongoFramework/Infrastructure/EntityEntry.cs
@@ -1,5 +1,6 @@
 ﻿using System;
 using MongoDB.Bson;
+using MongoFramework.Infrastructure.Mapping;
 
 namespace MongoFramework.Infrastructure
 {
@@ -33,6 +34,9 @@ public EntityEntry(object entity, EntityEntryState state)
 			Entity = entity;
 			EntityType = entity.GetType();
 
+			// Ensure the entity type is registered before we potentially serialize the entity to a BSON document
+			EntityMapping.TryRegisterType(EntityType, out _);
+
 			if (state == EntityEntryState.NoChanges)
 			{
 				OriginalValues = Entity.ToBsonDocument();

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,6 @@`
`1`	`1`	`using System;`
`2`	`2`	`using MongoDB.Bson;`
	`3`	`+using MongoFramework.Infrastructure.Mapping;`
`3`	`4`
`4`	`5`	`namespace MongoFramework.Infrastructure`
`5`	`6`	`{`
`@@ -33,6 +34,9 @@ public EntityEntry(object entity, EntityEntryState state)`
`33`	`34`	`Entity = entity;`
`34`	`35`	`EntityType = entity.GetType();`
`35`	`36`
	`37`	`+ // Ensure the entity type is registered before we potentially serialize the entity to a BSON document`
	`38`	`+ EntityMapping.TryRegisterType(EntityType, out _);`
	`39`	`+`
`36`	`40`	`if (state == EntityEntryState.NoChanges)`
`37`	`41`	`{`
`38`	`42`	`OriginalValues = Entity.ToBsonDocument();`