Skip to content

Commit f8031f9

Browse files
committed
fix tests, resolve comments
1 parent 3f6d3fb commit f8031f9

File tree

2 files changed

+11
-7
lines changed

2 files changed

+11
-7
lines changed

sdk/Sdk.Generators/FunctionMetadataProviderGenerator.cs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,6 @@ public void Execute(GeneratorExecutionContext context)
2424
return;
2525
}
2626

27-
// attempt to parse user compilation
28-
var p = new Parser(context);
29-
3027
context.AnalyzerConfigOptions.GlobalOptions.TryGetValue(Constants.BuildProperties.EnableSourceGenProp, out var sourceGenSwitch);
3128

3229
bool.TryParse(sourceGenSwitch, out bool enableSourceGen);
@@ -36,6 +33,9 @@ public void Execute(GeneratorExecutionContext context)
3633
return;
3734
}
3835

36+
// attempt to parse user compilation
37+
var p = new Parser(context);
38+
3939
IReadOnlyList<GeneratorFunctionMetadata> functionMetadataInfo = p.GetFunctionMetadataInfo(receiver.CandidateMethods);
4040

4141
// Proceed to generate the file if function metadata info was successfully returned
@@ -44,7 +44,7 @@ public void Execute(GeneratorExecutionContext context)
4444
Emitter e = new();
4545
string result = e.Emit(functionMetadataInfo, context.CancellationToken);
4646

47-
context.AddSource(Constants.FileNames.GeneratedFunctionMetadataFileName, SourceText.From(result, Encoding.UTF8));
47+
context.AddSource(Constants.FileNames.GeneratedFunctionMetadata, SourceText.From(result, Encoding.UTF8));
4848
}
4949
}
5050

test/Sdk.Generator.Tests/TestHelpers.cs

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,16 +34,20 @@ public static Task RunTestAsync<TSourceGenerator>(
3434
Sources = { inputSource },
3535
AdditionalReferences =
3636
{
37-
typeof(WorkerExtensionStartupAttribute).Assembly,
38-
},
39-
},
37+
typeof(WorkerExtensionStartupAttribute).Assembly
38+
}
39+
}
4040
};
4141

4242
if (expectedOutputSource != null && expectedFileName != null)
4343
{
4444
test.TestState.GeneratedSources.Add((typeof(TSourceGenerator), expectedFileName, SourceText.From(expectedOutputSource, Encoding.UTF8)));
4545
}
4646

47+
// Enable SourceGen MSBuild Property for testing
48+
string config = $"is_global = true{Environment.NewLine}build_property.FunctionsMetadataSourceGen_Enabled = {true}";
49+
test.TestState.AnalyzerConfigFiles.Add(("/.globalconfig", config));
50+
4751
foreach (var item in extensionAssemblyReferences)
4852
{
4953
test.TestState.AdditionalReferences.Add(item);

0 commit comments

Comments
 (0)