Fix: ensure generated abstract class access modifier matches interface

DecoratorGenerator.UnitTests/Tests.cs

@@ -5,6 +5,7 @@
 using SampleLibrary.Deep.Nesteds;
 using System.Reflection;
 using System.Text;
+using TestLibrary;
 using VerifyCS = DecoratorGenerator.UnitTests.CSharpSourceGeneratorVerifier<DecoratorGenerator.Main>;
 
 namespace DecoratorGenerator.UnitTests;
@@ -41,6 +42,29 @@
         }.RunAsync();
     }
 
+    [Test]
+    public async Task OneInterface_Internal() {
+        var source = await ReadCSharpFile<IInternalType>(true);
+        var generated = await ReadCSharpFile<InternalTypeDecorator>(true);
+
+        await new VerifyCS.Test
+        {
+            TestState = {
+                ReferenceAssemblies = ReferenceAssemblies.Net.Net90,
+                AdditionalReferences =
+                {
+                    implementationAssembly,
+                    GetAssembly("TestLibrary")
+                },
+                Sources = { source },
+                GeneratedSources =
+                {
+                    (typeof(Main), "InternalTypeDecorator.generated.cs", SourceText.From(generated, Encoding.UTF8, SourceHashAlgorithm.Sha256)),
+                },
+            },
+        }.RunAsync();
+    }
+
     [Test]
     public async Task OneInterface_Properties() {
         var source = await ReadCSharpFile<ILionProperties>(true);
@@ -152,7 +176,7 @@
                 {
                     implementationAssembly,
                     GetAssembly("TestLibrary"),
                     Assembly.GetAssembly(typeof(DynamoDBContext)),
                 },
                 Sources = { wrapperList, source },
                 GeneratedSources =
@@ -182,19 +206,19 @@
     private static async Task<string> ReadFile(bool isTestLibrary, string searchPattern) {
         var currentDirectory = GetCurrentDirectory();
 
         var targetDirectory = isTestLibrary ? GetTestLibraryDirectory(currentDirectory) : currentDirectory;
 
         var file = targetDirectory.GetFiles(searchPattern).First();
 
         using var fileReader = new StreamReader(file.OpenRead());
         return await fileReader.ReadToEndAsync();
     }
 
     private static DirectoryInfo? GetCurrentDirectory() {
         return Directory.GetParent(Directory.GetParent(AppDomain.CurrentDomain.BaseDirectory).Parent.Parent.FullName);
     }
 
     private static DirectoryInfo GetTestLibraryDirectory(DirectoryInfo currentDirectory) {
         return currentDirectory.Parent.GetDirectories("TestLibrary").First();
     }
 }

DecoratorGenerator/OutputGenerator.cs

@@ -8,6 +8,7 @@ namespace DecoratorGenerator
     internal static class OutputGenerator
     {
         public static (string source, string className) GenerateOutputs(INamedTypeSymbol @interface) {
+            var interfaceAccessModifier = @interface.DeclaredAccessibility.ToString().ToLower();
             var className = $"{@interface.Name.Substring(1)}Decorator";
             var formattedConstraintTypes = FormatInterfaceConstraintTypes(@interface);
             var formattedConstraints = CreateFormattedConstraints(@interface.TypeParameters);
@@ -25,7 +26,7 @@ public static (string source, string className) GenerateOutputs(INamedTypeSymbol
 #nullable restore
 namespace {@interface.ContainingNamespace.ToDisplayString()};
 
-public abstract class {className}{formattedConstraintTypes} : {@interface.Name}{formattedConstraintTypes}{(formattedConstraints != string.Empty ? $@" {formattedConstraints}" : string.Empty)}
+{interfaceAccessModifier} abstract class {className}{formattedConstraintTypes} : {@interface.Name}{formattedConstraintTypes}{(formattedConstraints != string.Empty ? $@" {formattedConstraints}" : string.Empty)}
 {{
     private {@interface.Name}{formattedConstraintTypes} {targetFieldName};
 

TestLibrary/IInternalType.cs

@@ -0,0 +1,9 @@
+using DecoratorGenerator;
+
+namespace TestLibrary;
+
+[Decorate]
+internal interface IInternalType
+{
+
+}

TestLibrary/InternalTypeDecorator.generated.cs

@@ -0,0 +1,16 @@
+// <auto-generated/>
+#nullable restore
+namespace TestLibrary;
+
+internal abstract class InternalTypeDecorator : IInternalType
+{
+    private IInternalType internalType;
+
+    protected InternalTypeDecorator(IInternalType internalType) {
+        this.internalType = internalType;
+    }
+
+
+
+
+}