Merge pull request #111 from Sevitas/107_custom_type_provider_customi…

…zation fixes #107 - Enable further customization of the CustomTypeProvider
kontent-ai · Aug 12, 2021 · a8f1852 · a8f1852
2 parents 83a6202 + 770b3ed
commit a8f1852
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/Kentico.Kontent.ModelGenerator.Core/TypeProviderCodeGenerator.cs b/src/Kentico.Kontent.ModelGenerator.Core/TypeProviderCodeGenerator.cs
@@ -63,17 +63,17 @@ namespace {_namespace}
 {{
     public class {CLASS_NAME} : ITypeProvider
     {{
-        private static readonly Dictionary<Type, string> _codenames = new Dictionary<Type, string>
+        protected static readonly Dictionary<Type, string> _codenames = new Dictionary<Type, string>
         {{
 {CreateCodenameDictionaryValues()}
         }};
 
-        public Type GetType(string contentType)
+        public virtual Type GetType(string contentType)
         {{
             return _codenames.Keys.FirstOrDefault(type => GetCodename(type).Equals(contentType));
         }}
 
-        public string GetCodename(Type contentType)
+        public virtual string GetCodename(Type contentType)
         {{
             return _codenames.TryGetValue(contentType, out var codename) ? codename : null;
         }}

diff --git a/test/Kentico.Kontent.ModelGenerator.Tests/Assets/CustomTypeProvider_CompiledCode.txt b/test/Kentico.Kontent.ModelGenerator.Tests/Assets/CustomTypeProvider_CompiledCode.txt
@@ -7,18 +7,18 @@ namespace KenticoKontentModels
 {
 	public class CustomTypeProvider : ITypeProvider
 	{
-		private static readonly Dictionary<Type, string> _codenames = new Dictionary<Type, string>
+		protected static readonly Dictionary<Type, string> _codenames = new Dictionary<Type, string>
 		{
 			{typeof(Article), "article"},
 			{typeof(Office), "office"}
 		};
 
-		public Type GetType(string contentType)
+		public virtual Type GetType(string contentType)
 		{
 			return _codenames.Keys.FirstOrDefault(type => GetCodename(type).Equals(contentType));
 		}
 
-		public string GetCodename(Type contentType)
+		public virtual string GetCodename(Type contentType)
 		{
 			return _codenames.TryGetValue(contentType, out var codename) ? codename : null;
 		}