diff --git a/tests/Aspire.Hosting.Python.Tests/AddPythonProjectTests.cs b/tests/Aspire.Hosting.Python.Tests/AddPythonProjectTests.cs index 06556e6615..318556c3fb 100644 --- a/tests/Aspire.Hosting.Python.Tests/AddPythonProjectTests.cs +++ b/tests/Aspire.Hosting.Python.Tests/AddPythonProjectTests.cs @@ -105,6 +105,28 @@ public async Task PythonResourceFinishesSuccessfully() Directory.Delete(projectDirectory, true); } + [Fact] + [RequiresTools(["python"])] + public async Task PythonResourceSupportsWithReference() + { + var (projectDirectory, _, scriptName) = CreateTempPythonProject(outputHelper); + + using var builder = CreateTestDistributedApplicationBuilder(); + + var externalResource = builder.AddConnectionString("connectionString"); + builder.Configuration["ConnectionStrings:connectionString"] = "test"; + + var pyproj = builder.AddPythonProject("pyproj", projectDirectory, scriptName) + .WithReference(externalResource); + + var environmentVariables = await pyproj.Resource.GetEnvironmentVariableValuesAsync(DistributedApplicationOperation.Run); + + Assert.Equal("test", environmentVariables["ConnectionStrings__connectionString"]); + + // If we don't throw, clean up the directories. + Directory.Delete(projectDirectory, true); + } + [Fact] [RequiresTools(["python"])] public async Task AddPythonProject_SetsResourcePropertiesCorrectly()