ContainerAutoData

Author: Noggog

CSharpExt.UnitTests/AutoFixture/ContainerAutoDataTests.cs

@@ -0,0 +1,39 @@
+using System.IO.Abstractions;
+using Autofac;
+using FluentAssertions;
+using Noggog;
+using Noggog.Testing.AutoFixture;
+
+namespace CSharpExt.UnitTests.AutoFixture;
+
+public class ContainerAutoDataTests
+{
+    public class TypicalModule : Module
+    {
+        protected override void Load(ContainerBuilder builder)
+        {
+            builder.RegisterType<Something>()
+                .AsImplementedInterfaces().SingleInstance();
+        }
+    }
+
+    public interface ISomething
+    {
+    }
+
+    public class Something : ISomething
+    {
+        
+    }
+
+    [Theory, ContainerAutoData(typeof(TypicalModule))]
+    public void Typical(ISomething something, int i)
+    {
+        something.Should().BeOfType<Something>();
+    }
+
+    [Theory, ContainerAutoData(typeof(TypicalModule))]
+    public void Missing(IFileSystem missing, int i)
+    {
+    }
+}

Noggog.Testing/AutoFixture/ContainerAutoData.cs

@@ -0,0 +1,22 @@
+using AutoFixture;
+using AutoFixture.Xunit2;
+
+namespace Noggog.Testing.AutoFixture;
+
+public class ContainerAutoData : AutoDataAttribute
+{
+    public ContainerAutoData(
+        Type ContainerType,
+        TargetFileSystem FileSystem = TargetFileSystem.Fake,
+        bool OmitAutoProperties = false)
+        : base(() =>
+        {
+            var ret = new Fixture();
+            ret.Customize(new DefaultCustomization(FileSystem));
+            ret.OmitAutoProperties = OmitAutoProperties;
+            ret.Customize(new ContainerAutoDataCustomization(ContainerType));
+            return ret;
+        })
+    {
+    }
+}

Noggog.Testing/AutoFixture/ContainerAutoDataCustomization.cs

@@ -0,0 +1,53 @@
+using Autofac;
+using Autofac.Core;
+using AutoFixture;
+using AutoFixture.Kernel;
+
+namespace Noggog.Testing.AutoFixture;
+
+public class ContainerAutoDataCustomization : ICustomization
+{
+    private readonly IContainer _container;
+    
+    public ContainerAutoDataCustomization(Type module)
+    {
+        var builder = new ContainerBuilder();
+        builder.RegisterModule((IModule)Activator.CreateInstance(module)!);
+        _container = builder.Build();
+    }
+    
+    public void Customize(IFixture fixture)
+    {
+        fixture.Customizations.Add(new ContainerSpecimenBuilder(_container));
+        fixture.Register<IContainer>(() => _container);
+    }
+
+    class ContainerSpecimenBuilder : ISpecimenBuilder
+    {
+        private readonly IContainer _container;
+        private readonly HashSet<Type> _blacklist = new();
+
+        public ContainerSpecimenBuilder(IContainer container)
+        {
+            _container = container;
+        }
+        
+        public object Create(object request, ISpecimenContext context)
+        {
+            if (request is Type t && !_blacklist.Contains(t))
+            {
+                try
+                {
+                    return _container.Resolve(t);
+                }
+                catch (Exception e)
+                {
+                    _blacklist.Add(t);
+                    return new NoSpecimen();
+                }
+            }
+            
+            return new NoSpecimen();
+        }
+    }
+}