// Copyright 2026 Alibaba Group Holding Ltd.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

using OpenSandbox.Core;
using OpenSandbox.Models;
using Xunit;

namespace OpenSandbox.Tests;

public class EndpointCacheTests
{
    private static Endpoint Ep(string addr) => new() { EndpointAddress = addr, Headers = new Dictionary<string, string>() };

    [Fact]
    public void Get_ReturnsNull_OnMiss()
    {
        var cache = new EndpointCache(maxSize: 10, ttlSeconds: 60);
        var key = new EndpointCacheKey("sb-1", 8080, false);
        Assert.Null(cache.Get(key));
    }

    [Fact]
    public void Get_ReturnsEndpoint_AfterPut()
    {
        var cache = new EndpointCache(maxSize: 10, ttlSeconds: 60);
        var key = new EndpointCacheKey("sb-1", 8080, false);
        cache.Put(key, Ep("localhost:8080"));
        Assert.Equal("localhost:8080", cache.Get(key)?.EndpointAddress);
    }

    [Fact]
    public void Invalidate_RemovesAllEntriesForSandbox()
    {
        var cache = new EndpointCache(maxSize: 10, ttlSeconds: 60);
        cache.Put(new EndpointCacheKey("sb-1", 8080, false), Ep("a"));
        cache.Put(new EndpointCacheKey("sb-1", 18080, false), Ep("b"));
        cache.Put(new EndpointCacheKey("sb-2", 8080, false), Ep("c"));

        cache.Invalidate("sb-1");

        Assert.Null(cache.Get(new EndpointCacheKey("sb-1", 8080, false)));
        Assert.Null(cache.Get(new EndpointCacheKey("sb-1", 18080, false)));
        Assert.NotNull(cache.Get(new EndpointCacheKey("sb-2", 8080, false)));
    }

    [Fact]
    public async Task GetOrFetchAsync_DeduplicatesConcurrentRequests()
    {
        var cache = new EndpointCache(maxSize: 10, ttlSeconds: 60);
        var key = new EndpointCacheKey("sb-1", 8080, false);
        var fetchCount = 0;

        async Task<Endpoint> Fetcher()
        {
            Interlocked.Increment(ref fetchCount);
            await Task.Delay(50);
            return Ep("result");
        }

        var tasks = Enumerable.Range(0, 5)
            .Select(_ => cache.GetOrFetchAsync(key, Fetcher))
            .ToArray();

        var results = await Task.WhenAll(tasks);

        Assert.Equal(1, fetchCount);
        Assert.All(results, r => Assert.Equal("result", r.EndpointAddress));
    }

    [Fact]
    public async Task GetOrFetchAsync_ReturnsCachedValue_WithoutCallingFetcher()
    {
        var cache = new EndpointCache(maxSize: 10, ttlSeconds: 60);
        var key = new EndpointCacheKey("sb-1", 8080, false);
        cache.Put(key, Ep("cached"));
        var called = false;

        var result = await cache.GetOrFetchAsync(key, () =>
        {
            called = true;
            return Task.FromResult(Ep("fetched"));
        });

        Assert.Equal("cached", result.EndpointAddress);
        Assert.False(called);
    }

    [Fact]
    public async Task GetOrFetchAsync_DoesNotCache_OnError()
    {
        var cache = new EndpointCache(maxSize: 10, ttlSeconds: 60);
        var key = new EndpointCacheKey("sb-1", 8080, false);

        await Assert.ThrowsAsync<InvalidOperationException>(() =>
            cache.GetOrFetchAsync(key, () => throw new InvalidOperationException("network error")));

        Assert.Equal(0, cache.Count);
    }

    [Fact]
    public void LruEviction_WhenAtCapacity()
    {
        var cache = new EndpointCache(maxSize: 3, ttlSeconds: 60);
        cache.Put(new EndpointCacheKey("sb-0", 8080, false), Ep("h0"));
        cache.Put(new EndpointCacheKey("sb-1", 8080, false), Ep("h1"));
        cache.Put(new EndpointCacheKey("sb-2", 8080, false), Ep("h2"));

        // Access sb-0 to make it recently used
        cache.Get(new EndpointCacheKey("sb-0", 8080, false));
        // Insert 4th — sb-1 should be evicted
        cache.Put(new EndpointCacheKey("sb-3", 8080, false), Ep("h3"));

        Assert.Null(cache.Get(new EndpointCacheKey("sb-1", 8080, false)));
        Assert.NotNull(cache.Get(new EndpointCacheKey("sb-0", 8080, false)));
    }
}