src/Components/Endpoints/test/TempData/TempDataTest.cs - main - githubmirror/aspnetcore

src/Components/Endpoints/test/TempData/TempDataTest.cs
329 строк8 KB
Daria Tiurina
Fix TempData lazy loading bugs (#65722)
10 мар 2026, 19:03
10 мар 2026, 19:03e55796c
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

namespace Microsoft.AspNetCore.Components.Endpoints;

public class TempDataTest
{
    internal TempData CreateTempData()
    {
        return new TempData(() => new Dictionary<string, object>());
    }

    [Fact]
    public void Indexer_CanSetAndGetValues()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";
        var value = tempData["Key1"];
        Assert.Equal("Value1", value);
    }

    [Fact]
    public void Get_ReturnsValueAndRemovesFromRetainedKeys()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";

        var value = tempData.Get("Key1");

        Assert.Equal("Value1", value);
        var saved = tempData.Save();
        Assert.Empty(saved);
    }

    [Fact]
    public void Get_ReturnsNullForNonExistentKey()
    {
        var tempData = CreateTempData();
        var value = tempData.Get("NonExistent");
        Assert.Null(value);
    }

    [Fact]
    public void Peek_ReturnsValueWithoutRemovingFromRetainedKeys()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";
        var value = tempData.Peek("Key1");
        Assert.Equal("Value1", value);
        value = tempData.Get("Key1");
        Assert.Equal("Value1", value);
    }

    [Fact]
    public void Peek_ReturnsNullForNonExistentKey()
    {
        var tempData = CreateTempData();
        var value = tempData.Peek("NonExistent");
        Assert.Null(value);
    }

    [Fact]
    public void Keep_RetainsAllKeys()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";
        tempData["Key2"] = "Value2";
        _ = tempData.Get("Key1");
        _ = tempData.Get("Key2");

        tempData.Keep();

        var value1 = tempData.Get("Key1");
        var value2 = tempData.Get("Key2");
        Assert.Equal("Value1", value1);
        Assert.Equal("Value2", value2);
    }

    [Fact]
    public void KeepWithKey_RetainsSpecificKey()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";
        tempData["Key2"] = "Value2";
        _ = tempData.Get("Key1");
        _ = tempData.Get("Key2");

        tempData.Keep("Key1");

        var saved = tempData.Save();
        Assert.Single(saved);
        Assert.Equal("Value1", saved["Key1"]);
    }

    [Fact]
    public void KeepWithKey_DoesNothingForNonExistentKey()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";
        _ = tempData.Get("Key1");

        tempData.Keep("NonExistent");

        var value = tempData.Get("NonExistent");
        Assert.Null(value);
    }

    [Fact]
    public void ContainsKey_ReturnsTrueForExistingKey()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";
        var result = tempData.ContainsKey("Key1");
        Assert.True(result);
    }

    [Fact]
    public void ContainsKey_ReturnsFalseForNonExistentKey()
    {
        var tempData = CreateTempData();
        var result = tempData.ContainsKey("NonExistent");
        Assert.False(result);
    }

    [Fact]
    public void Remove_RemovesKeyAndReturnsTrue()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";

        var result = tempData.Remove("Key1");

        Assert.True(result);
        var value = tempData.Get("Key1");
        Assert.Null(value);
    }

    [Fact]
    public void Remove_ReturnsFalseForNonExistentKey()
    {
        var tempData = CreateTempData();
        var result = tempData.Remove("NonExistent");
        Assert.False(result);
    }

    [Fact]
    public void Save_ReturnsOnlyRetainedKeys()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";
        tempData["Key2"] = "Value2";
        tempData["Key3"] = "Value3";
        _ = tempData.Get("Key1");
        _ = tempData.Get("Key2");

        var saved = tempData.Save();

        Assert.Single(saved);
        Assert.Equal("Value3", saved["Key3"]);
    }

    [Fact]
    public void Load_PopulatesDataFromDictionary()
    {
        var tempData = CreateTempData();
        var dataToLoad = new Dictionary<string, object>
        {
            ["Key1"] = "Value1",
            ["Key2"] = "Value2"
        };

        tempData.Load(dataToLoad);

        Assert.Equal("Value1", tempData.Get("Key1"));
        Assert.Equal("Value2", tempData.Get("Key2"));
    }

    [Fact]
    public void Load_ClearsExistingDataBeforeLoading()
    {
        var tempData = CreateTempData();
        tempData["ExistingKey"] = "ExistingValue";
        var dataToLoad = new Dictionary<string, object>
        {
            ["NewKey"] = "NewValue"
        };

        tempData.Load(dataToLoad);

        Assert.False(tempData.ContainsKey("ExistingKey"));
        Assert.True(tempData.ContainsKey("NewKey"));
    }

    [Fact]
    public void Clear_RemovesAllData()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";
        tempData["Key2"] = "Value2";

        tempData.Clear();

        Assert.Null(tempData.Get("Key1"));
        Assert.Null(tempData.Get("Key2"));
    }

    [Fact]
    public void Indexer_IsCaseInsensitive()
    {
        var tempData = CreateTempData();
        tempData["Key1"] = "Value1";
        var value = tempData["KEY1"];
        Assert.Equal("Value1", value);
    }

    [Fact]
    public void Save_DoesNotLoadTempData()
    {
        var loadCalled = false;
        var tempData = new TempData(() =>
        {
            loadCalled = true;
            return new Dictionary<string, object> { ["Key"] = "Value" };
        });
        var saved = tempData.Save();

        Assert.False(loadCalled);
        Assert.False(tempData.WasLoaded);
        Assert.Empty(saved);
    }

    [Fact]
    public void TempData_Loads_WhenLoaded()
    {
        var loadCalled = false;
        var numberOfLoads = 0;
        var tempData = new TempData(() =>
        {
            loadCalled = true;
            numberOfLoads++;
            return new Dictionary<string, object> { ["Key"] = "Value" };
        });

        var value = tempData["Key"];
        _ = tempData.Peek("Key");
        _ = tempData.ContainsKey("Key");

        Assert.True(loadCalled);
        Assert.Equal(1, numberOfLoads);
        Assert.True(tempData.WasLoaded);
        Assert.Equal("Value", value);
    }

    [Fact]
    public void Get_ConsumesValue_WhenFirstAccessTriggersLazyLoad()
    {
        var tempData = new TempData(() => new Dictionary<string, object>
        {
            ["Message"] = "hello"
        });

        var value = tempData.Get("Message");
        var saved = tempData.Save();

        Assert.Equal("hello", value);
        Assert.Empty(saved);
    }

    [Fact]
    public void Indexer_ConsumesValue_WhenFirstAccessTriggersLazyLoad()
    {
        var tempData = new TempData(() => new Dictionary<string, object>
        {
            ["Message"] = "hello"
        });

        var value = tempData["Message"];
        var saved = tempData.Save();

        Assert.Equal("hello", value);
        Assert.Empty(saved);
    }

    [Fact]
    public void Remove_RemovesValue_WhenFirstAccessTriggersLazyLoad()
    {
        var tempData = new TempData(() => new Dictionary<string, object>
        {
            ["Message"] = "hello"
        });

        var result = tempData.Remove("Message");
        var saved = tempData.Save();

        Assert.True(result);
        Assert.Empty(saved);
    }

    [Fact]
    public void Keep_RetainsAllKeys_WhenFirstAccessTriggersLazyLoad()
    {
        var tempData = new TempData(() => new Dictionary<string, object>
        {
            ["Key1"] = "Value1",
            ["Key2"] = "Value2"
        });

        tempData.Keep();
        var saved = tempData.Save();

        Assert.Equal(2, saved.Count);
        Assert.Equal("Value1", saved["Key1"]);
        Assert.Equal("Value2", saved["Key2"]);
    }

    [Fact]
    public void Enumerator_IteratesAllKeys_WhenFirstAccessTriggersLazyLoad()
    {
        var tempData = new TempData(() => new Dictionary<string, object>
        {
            ["Key1"] = "Value1",
            ["Key2"] = "Value2"
        });

        var items = ((IEnumerable<KeyValuePair<string, object>>)tempData).ToList();

        Assert.Equal(2, items.Count);
    }
}