diff --git a/TUnit.Mocks.SourceGenerator.Tests/SnapshotTestBase.cs b/TUnit.Mocks.SourceGenerator.Tests/SnapshotTestBase.cs
index 7ae7046061..bb4723cefb 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/SnapshotTestBase.cs
+++ b/TUnit.Mocks.SourceGenerator.Tests/SnapshotTestBase.cs
@@ -47,7 +47,7 @@ protected static string[] RunGenerator(string source, CSharpParseOptions? parseO
).WithReferences(_references.Value);
var generator = new MockGenerator();
- GeneratorDriver driver = CSharpGeneratorDriver.Create(generator);
+ GeneratorDriver driver = CSharpGeneratorDriver.Create([generator.AsSourceGenerator()], parseOptions: parseOptions);
var runResult = driver.RunGenerators(compilation).GetRunResult();
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_Inheriting_Multiple_Interfaces.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_Inheriting_Multiple_Interfaces.verified.txt
index ba17896519..aae5a4002c 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_Inheriting_Multiple_Interfaces.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_Inheriting_Multiple_Interfaces.verified.txt
@@ -173,3 +173,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Async_Methods.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Async_Methods.verified.txt
index 54debdf4bf..8df17a1351 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Async_Methods.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Async_Methods.verified.txt
@@ -408,3 +408,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Events.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Events.verified.txt
index 364ed98c68..7040397997 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Events.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Events.verified.txt
@@ -211,3 +211,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Generic_Methods.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Generic_Methods.verified.txt
index eb876fb232..68f87ae580 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Generic_Methods.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Generic_Methods.verified.txt
@@ -112,3 +112,11 @@ namespace TUnit.Mocks.Generated
}
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Inherited_Static_Abstract_Members.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Inherited_Static_Abstract_Members.verified.txt
index 1eff4d264d..dc8c623623 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Inherited_Static_Abstract_Members.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Inherited_Static_Abstract_Members.verified.txt
@@ -115,3 +115,11 @@ namespace TUnit.Mocks.Generated
}
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Keyword_Parameter_Names.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Keyword_Parameter_Names.verified.txt
index 4c614d4cc2..ccc0455c03 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Keyword_Parameter_Names.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Keyword_Parameter_Names.verified.txt
@@ -273,3 +273,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Mixed_Members.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Mixed_Members.verified.txt
index 8b8c4a4c35..4b83f966ab 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Mixed_Members.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Mixed_Members.verified.txt
@@ -349,3 +349,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Nullable_Reference_Type_Parameters.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Nullable_Reference_Type_Parameters.verified.txt
index 1b42ee5f1b..94d54e07ad 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Nullable_Reference_Type_Parameters.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Nullable_Reference_Type_Parameters.verified.txt
@@ -540,3 +540,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Out_Ref_Parameters.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Out_Ref_Parameters.verified.txt
index 9271350b0a..fdb45bb5ea 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Out_Ref_Parameters.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Out_Ref_Parameters.verified.txt
@@ -274,3 +274,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Overloaded_Methods.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Overloaded_Methods.verified.txt
index 27371502d5..cc74b2811e 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Overloaded_Methods.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Overloaded_Methods.verified.txt
@@ -548,3 +548,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Properties.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Properties.verified.txt
index fdd42f5115..75d2bee750 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Properties.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Properties.verified.txt
@@ -86,3 +86,11 @@ namespace TUnit.Mocks.Generated
}
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_RefStruct_Parameters.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_RefStruct_Parameters.verified.txt
index 85bbb1d4db..4603818ba5 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_RefStruct_Parameters.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_RefStruct_Parameters.verified.txt
@@ -117,3 +117,11 @@ namespace TUnit.Mocks.Generated
}
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Static_Abstract_Members.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Static_Abstract_Members.verified.txt
index 035fcac82b..9a09752e23 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Static_Abstract_Members.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Static_Abstract_Members.verified.txt
@@ -137,3 +137,11 @@ namespace TUnit.Mocks.Generated
}
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Static_Abstract_Transitive_Return_Type.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Static_Abstract_Transitive_Return_Type.verified.txt
index 7cf3821cf7..bee541a03d 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Static_Abstract_Transitive_Return_Type.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Interface_With_Static_Abstract_Transitive_Return_Type.verified.txt
@@ -182,3 +182,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Multi_Method_Interface.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Multi_Method_Interface.verified.txt
index 197f3d5a86..1b4c77e37f 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Multi_Method_Interface.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Multi_Method_Interface.verified.txt
@@ -313,3 +313,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Simple_Interface_With_One_Method.verified.txt b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Simple_Interface_With_One_Method.verified.txt
index 3077ce0d93..4ed109f3f2 100644
--- a/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Simple_Interface_With_One_Method.verified.txt
+++ b/TUnit.Mocks.SourceGenerator.Tests/Snapshots/Simple_Interface_With_One_Method.verified.txt
@@ -165,3 +165,11 @@ namespace TUnit.Mocks.Generated
public void WasNeverCalled(string? message) => _engine.CreateVerification(_memberId, _memberName, _matchers).WasNeverCalled(message);
}
}
+
+
+// ===== FILE SEPARATOR =====
+
+//
+#nullable enable
+
+namespace TUnit.Mocks.Generated;
\ No newline at end of file
diff --git a/TUnit.Mocks.SourceGenerator/MockGenerator.cs b/TUnit.Mocks.SourceGenerator/MockGenerator.cs
index 85c2f4605f..9d0da86c80 100644
--- a/TUnit.Mocks.SourceGenerator/MockGenerator.cs
+++ b/TUnit.Mocks.SourceGenerator/MockGenerator.cs
@@ -10,6 +10,17 @@ public class MockGenerator : IIncrementalGenerator
{
public void Initialize(IncrementalGeneratorInitializationContext context)
{
+ // Always emit the TUnit.Mocks.Generated namespace so that global usings never fail
+ context.RegisterPostInitializationOutput(ctx =>
+ {
+ ctx.AddSource("TUnit.Mocks.Generated.Namespace.g.cs", """
+ //
+ #nullable enable
+
+ namespace TUnit.Mocks.Generated;
+ """);
+ });
+
// Step 1a: Find all Mock.Of() invocations and transform to model arrays
var mockTypes = context.SyntaxProvider
.CreateSyntaxProvider(