public async Task TraceIdEnvironmentVariableIsSet() { using (var bootstrap = new LambdaBootstrap(_testFunction.BaseHandlerAsync, null)) { bootstrap.Client = _testRuntimeApiClient; Assert.Null(_environmentVariables.GetEnvironmentVariable(LambdaEnvironment.EnvVarTraceId)); await bootstrap.InvokeOnceAsync(); Assert.NotNull(_testRuntimeApiClient.LastTraceId); Assert.Equal(_testRuntimeApiClient.LastTraceId, _environmentVariables.GetEnvironmentVariable(LambdaEnvironment.EnvVarTraceId)); } Assert.False(_testInitializer.InitializerWasCalled); Assert.True(_testFunction.HandlerWasCalled); }
public void SetsExecutionEnvironmentButNotTwice() { var expectedValueRegex = new Regex($"{LambdaExecutionEnvironment}_amazonlambdaruntimesupport_[0-9]+\\.[0-9]+\\.[0-9]+"); _environmentVariables.SetEnvironmentVariable(LambdaEnvironment.EnvVarExecutionEnvironment, LambdaExecutionEnvironment); var lambdaEnvironment = new LambdaEnvironment(_environmentVariables); Assert.True(expectedValueRegex.IsMatch(lambdaEnvironment.ExecutionEnvironment)); Assert.True(expectedValueRegex.IsMatch(_environmentVariables.GetEnvironmentVariable(LambdaEnvironment.EnvVarExecutionEnvironment))); // Make sure that creating another LambdaEnvironment instance won't change the value. lambdaEnvironment = new LambdaEnvironment(_environmentVariables); Assert.True(expectedValueRegex.IsMatch(lambdaEnvironment.ExecutionEnvironment)); Assert.True(expectedValueRegex.IsMatch(_environmentVariables.GetEnvironmentVariable(LambdaEnvironment.EnvVarExecutionEnvironment))); }