diff --git a/Source/Mockolate.SourceGenerators/Sources/Sources.ReturnsAsyncExtensions.cs b/Source/Mockolate.SourceGenerators/Sources/Sources.ReturnsAsyncExtensions.cs
index 992e5c60..cf691390 100644
--- a/Source/Mockolate.SourceGenerators/Sources/Sources.ReturnsAsyncExtensions.cs
+++ b/Source/Mockolate.SourceGenerators/Sources/Sources.ReturnsAsyncExtensions.cs
@@ -33,7 +33,7 @@ namespace Mockolate;
"\t/// Registers the for this method.")
.AppendLine();
sb.Append("\t/// ").AppendLine();
- sb.Append("\tpublic static IReturnMethodSetup, ").Append(types)
+ sb.Append("\tpublic static IReturnMethodSetupReturnBuilder, ").Append(types)
.Append("> ReturnsAsync(this IReturnMethodSetup, ")
.Append(types).Append("> setup, TReturn returnValue)").AppendLine();
sb.Append("\t\t=> setup.Returns(Task.FromResult(returnValue));").AppendLine();
@@ -43,7 +43,7 @@ namespace Mockolate;
"\t/// Registers an to setup the return value for this method.")
.AppendLine();
sb.Append("\t/// ").AppendLine();
- sb.Append("\tpublic static IReturnMethodSetup, ").Append(types)
+ sb.Append("\tpublic static IReturnMethodSetupReturnBuilder, ").Append(types)
.Append("> ReturnsAsync(this IReturnMethodSetup, ")
.Append(types).Append("> setup, Func callback)").AppendLine();
sb.Append("\t\t=> setup.Returns(() => Task.FromResult(callback()));").AppendLine();
@@ -53,7 +53,7 @@ namespace Mockolate;
"\t/// Registers an to setup the return value for this method.")
.AppendLine();
sb.Append("\t/// ").AppendLine();
- sb.Append("\tpublic static IReturnMethodSetup, ").Append(types)
+ sb.Append("\tpublic static IReturnMethodSetupReturnBuilder, ").Append(types)
.Append("> ReturnsAsync(this IReturnMethodSetup, ")
.Append(types).Append("> setup, Func<").Append(types).Append(", TReturn> callback)").AppendLine();
sb.Append("\t\t=> setup.Returns((").Append(variables).Append(") => Task.FromResult(callback(")
@@ -72,7 +72,7 @@ namespace Mockolate;
"\t/// Registers the for this method.")
.AppendLine();
sb.Append("\t/// ").AppendLine();
- sb.Append("\tpublic static IReturnMethodSetup, ").Append(types)
+ sb.Append("\tpublic static IReturnMethodSetupReturnBuilder, ").Append(types)
.Append("> ReturnsAsync(this IReturnMethodSetup, ").Append(types)
.Append("> setup, TReturn returnValue)").AppendLine();
@@ -83,7 +83,7 @@ namespace Mockolate;
"\t/// Registers an to setup the return value for this method.")
.AppendLine();
sb.Append("\t/// ").AppendLine();
- sb.Append("\tpublic static IReturnMethodSetup, ").Append(types)
+ sb.Append("\tpublic static IReturnMethodSetupReturnBuilder, ").Append(types)
.Append("> ReturnsAsync(this IReturnMethodSetup, ").Append(types)
.Append("> setup, Func callback)").AppendLine();
@@ -94,7 +94,7 @@ namespace Mockolate;
"\t/// Registers an to setup the return value for this method.")
.AppendLine();
sb.Append("\t/// ").AppendLine();
- sb.Append("\tpublic static IReturnMethodSetup, ").Append(types)
+ sb.Append("\tpublic static IReturnMethodSetupReturnBuilder, ").Append(types)
.Append("> ReturnsAsync(this IReturnMethodSetup, ").Append(types).Append("> setup, Func<")
.Append(types).Append(", TReturn> callback)").AppendLine();
diff --git a/Source/Mockolate/ReturnsAsyncExtensions.cs b/Source/Mockolate/ReturnsAsyncExtensions.cs
index 096fc381..e400297f 100644
--- a/Source/Mockolate/ReturnsAsyncExtensions.cs
+++ b/Source/Mockolate/ReturnsAsyncExtensions.cs
@@ -13,98 +13,98 @@ public static class ReturnsAsyncExtensions
///
/// Registers the for this method.
///
- public static IReturnMethodSetup> ReturnsAsync(this IReturnMethodSetup> setup,
+ public static IReturnMethodSetupReturnBuilder> ReturnsAsync(this IReturnMethodSetup> setup,
TReturn returnValue)
=> setup.Returns(Task.FromResult(returnValue));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup> ReturnsAsync(this IReturnMethodSetup> setup,
+ public static IReturnMethodSetupReturnBuilder> ReturnsAsync(this IReturnMethodSetup> setup,
Func callback)
=> setup.Returns(() => Task.FromResult(callback()));
///
/// Registers the for this method.
///
- public static IReturnMethodSetup, T1> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1> ReturnsAsync(
this IReturnMethodSetup, T1> setup, TReturn returnValue)
=> setup.Returns(Task.FromResult(returnValue));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1> ReturnsAsync(
this IReturnMethodSetup, T1> setup, Func callback)
=> setup.Returns(() => Task.FromResult(callback()));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1> ReturnsAsync(
this IReturnMethodSetup, T1> setup, Func callback)
=> setup.Returns(v1 => Task.FromResult(callback(v1)));
///
/// Registers the for this method.
///
- public static IReturnMethodSetup, T1, T2> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2> ReturnsAsync(
this IReturnMethodSetup, T1, T2> setup, TReturn returnValue)
=> setup.Returns(Task.FromResult(returnValue));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2> ReturnsAsync(
this IReturnMethodSetup, T1, T2> setup, Func callback)
=> setup.Returns(() => Task.FromResult(callback()));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2> ReturnsAsync(
this IReturnMethodSetup, T1, T2> setup, Func callback)
=> setup.Returns((v1, v2) => Task.FromResult(callback(v1, v2)));
///
/// Registers the for this method.
///
- public static IReturnMethodSetup, T1, T2, T3> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3> setup, TReturn returnValue)
=> setup.Returns(Task.FromResult(returnValue));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2, T3> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3> setup, Func callback)
=> setup.Returns(() => Task.FromResult(callback()));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2, T3> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3> setup, Func callback)
=> setup.Returns((v1, v2, v3) => Task.FromResult(callback(v1, v2, v3)));
///
/// Registers the for this method.
///
- public static IReturnMethodSetup, T1, T2, T3, T4> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3, T4> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3, T4> setup, TReturn returnValue)
=> setup.Returns(Task.FromResult(returnValue));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2, T3, T4> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3, T4> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3, T4> setup, Func callback)
=> setup.Returns(() => Task.FromResult(callback()));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2, T3, T4> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3, T4> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3, T4> setup, Func callback)
=> setup.Returns((v1, v2, v3, v4) => Task.FromResult(callback(v1, v2, v3, v4)));
@@ -113,98 +113,98 @@ public static IReturnMethodSetup, T1, T2, T3, T4> ReturnsAsync
/// Registers the for this method.
///
- public static IReturnMethodSetup> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder> ReturnsAsync(
this IReturnMethodSetup> setup, TReturn returnValue)
=> setup.Returns(ValueTask.FromResult(returnValue));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder> ReturnsAsync(
this IReturnMethodSetup> setup, Func callback)
=> setup.Returns(() => ValueTask.FromResult(callback()));
///
/// Registers the for this method.
///
- public static IReturnMethodSetup, T1> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1> ReturnsAsync(
this IReturnMethodSetup, T1> setup, TReturn returnValue)
=> setup.Returns(ValueTask.FromResult(returnValue));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1> ReturnsAsync(
this IReturnMethodSetup, T1> setup, Func callback)
=> setup.Returns(() => ValueTask.FromResult(callback()));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1> ReturnsAsync(
this IReturnMethodSetup, T1> setup, Func callback)
=> setup.Returns(v1 => ValueTask.FromResult(callback(v1)));
///
/// Registers the for this method.
///
- public static IReturnMethodSetup, T1, T2> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2> ReturnsAsync(
this IReturnMethodSetup, T1, T2> setup, TReturn returnValue)
=> setup.Returns(ValueTask.FromResult(returnValue));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2> ReturnsAsync(
this IReturnMethodSetup, T1, T2> setup, Func callback)
=> setup.Returns(() => ValueTask.FromResult(callback()));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2> ReturnsAsync(
this IReturnMethodSetup, T1, T2> setup, Func callback)
=> setup.Returns((v1, v2) => ValueTask.FromResult(callback(v1, v2)));
///
/// Registers the for this method.
///
- public static IReturnMethodSetup, T1, T2, T3> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3> setup, TReturn returnValue)
=> setup.Returns(ValueTask.FromResult(returnValue));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2, T3> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3> setup, Func callback)
=> setup.Returns(() => ValueTask.FromResult(callback()));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2, T3> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3> setup, Func callback)
=> setup.Returns((v1, v2, v3) => ValueTask.FromResult(callback(v1, v2, v3)));
///
/// Registers the for this method.
///
- public static IReturnMethodSetup, T1, T2, T3, T4> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3, T4> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3, T4> setup, TReturn returnValue)
=> setup.Returns(ValueTask.FromResult(returnValue));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2, T3, T4> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3, T4> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3, T4> setup, Func callback)
=> setup.Returns(() => ValueTask.FromResult(callback()));
///
/// Registers an to setup the return value for this method.
///
- public static IReturnMethodSetup, T1, T2, T3, T4> ReturnsAsync(
+ public static IReturnMethodSetupReturnBuilder, T1, T2, T3, T4> ReturnsAsync(
this IReturnMethodSetup, T1, T2, T3, T4> setup, Func callback)
=> setup.Returns((v1, v2, v3, v4) => ValueTask.FromResult(callback(v1, v2, v3, v4)));
#pragma warning restore CA2012 // Use ValueTasks correctly
diff --git a/Tests/Mockolate.Api.Tests/Expected/Mockolate_net10.0.txt b/Tests/Mockolate.Api.Tests/Expected/Mockolate_net10.0.txt
index b8502f93..bee28702 100644
--- a/Tests/Mockolate.Api.Tests/Expected/Mockolate_net10.0.txt
+++ b/Tests/Mockolate.Api.Tests/Expected/Mockolate_net10.0.txt
@@ -163,34 +163,34 @@ namespace Mockolate
}
public static class ReturnsAsyncExtensions
{
- public static Mockolate.Setup.IReturnMethodSetup> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup> setup, TReturn returnValue) { }
- public static Mockolate.Setup.IReturnMethodSetup> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup> setup, TReturn returnValue) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1> setup, TReturn returnValue) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1> setup, TReturn returnValue) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1, T2> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1, T2> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1, T2> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1, T2> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1, T2> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1, T2> setup, TReturn returnValue) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1, T2> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1, T2> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1, T2> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1, T2> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1, T2> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1, T2> setup, TReturn returnValue) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1, T2, T3> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1, T2, T3> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup, T1, T2, T3> ReturnsAsync(this Mockolate.Setup.IReturnMethodSetup, T1, T2, T3> setup, System.Func callback) { }
- public static Mockolate.Setup.IReturnMethodSetup