public void GetEffectiveCoresCount_RetrunsExpectedResult() { TestEnvironment env = new TestEnvironment(); Assert.Equal(Environment.ProcessorCount, EnvironmentExtensions.GetEffectiveCoresCount(env)); env.Clear(); env.SetEnvironmentVariable(EnvironmentSettingNames.AzureWebsiteSku, ScriptConstants.DynamicSku); Assert.Equal(1, EnvironmentExtensions.GetEffectiveCoresCount(env)); env.Clear(); env.SetEnvironmentVariable(EnvironmentSettingNames.AzureWebsiteSku, ScriptConstants.DynamicSku); env.SetEnvironmentVariable(EnvironmentSettingNames.RoleInstanceId, "dw0SmallDedicatedWebWorkerRole_hr0HostRole-0-VM-1"); Assert.Equal(Environment.ProcessorCount, EnvironmentExtensions.GetEffectiveCoresCount(env)); }
public void IsWarmUpRequest_ReturnsExpectedValue() { var environment = new TestEnvironment(); var hostEnvironment = new ScriptWebHostEnvironment(environment); var request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/warmup"); Assert.False(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment)); // Reset environment environment.Clear(); hostEnvironment = new ScriptWebHostEnvironment(environment); environment.SetEnvironmentVariable(EnvironmentSettingNames.AzureWebsitePlaceholderMode, "1"); Assert.False(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment)); environment.SetEnvironmentVariable(EnvironmentSettingNames.AzureWebsiteInstanceId, "12345"); Assert.True(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment)); request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/csharphttpwarmup"); Assert.True(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment)); request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/warmup"); request.Headers.Add(ScriptConstants.AntaresLogIdHeaderName, "xyz123"); Assert.False(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment)); request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/foo"); Assert.False(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment)); // Reset environment environment.Clear(); hostEnvironment = new ScriptWebHostEnvironment(environment); environment.SetEnvironmentVariable(EnvironmentSettingNames.AzureWebsitePlaceholderMode, "1"); Assert.False(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment)); request = HttpTestHelpers.CreateHttpRequest("POST", "http://azure.com/api/warmup"); environment.SetEnvironmentVariable(EnvironmentSettingNames.ContainerName, "TestContainer"); Assert.True(environment.IsLinuxConsumption()); Assert.True(HostWarmupMiddleware.IsWarmUpRequest(request, hostEnvironment, environment)); }
public void Dispose() { _testEnvironment.Clear(); }