// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System;
using System.IO;
using System.Threading.Tasks;
using Azure.Identity;
using Azure.Storage.Files.DataLake;
using Microsoft.Extensions.AI.Evaluation.Reporting.Storage;
using Xunit;
 
namespace Microsoft.Extensions.AI.Evaluation.Reporting.Tests;
 
public class AzureResponseCacheTests : ResponseCacheTester, IAsyncLifetime
{
    private static readonly DataLakeFileSystemClient? _fsClient;
 
    static AzureResponseCacheTests()
    {
        if (Settings.Current.Configured)
        {
            var credential = new ChainedTokenCredential(new AzureCliCredential(), new DefaultAzureCredential());
            _fsClient = new(
                new Uri(
                    baseUri: new Uri(Settings.Current.StorageAccountEndpoint),
                    relativeUri: Settings.Current.StorageContainerName),
                credential);
        }
    }
 
    private readonly DataLakeDirectoryClient? _dirClient;
 
    public AzureResponseCacheTests()
    {
        _dirClient = _fsClient?.GetDirectoryClient(Path.GetRandomFileName());
    }
 
    public Task InitializeAsync() => Task.CompletedTask;
 
    public async Task DisposeAsync()
    {
        if (Settings.Current.Configured)
        {
            await CreateResponseCacheProvider().ResetAsync();
        }
    }
 
    internal override bool IsConfigured => Settings.Current.Configured;
 
    internal override IEvaluationResponseCacheProvider CreateResponseCacheProvider()
        => new AzureStorageResponseCacheProvider(_dirClient!);
 
    internal override IEvaluationResponseCacheProvider CreateResponseCacheProvider(Func<DateTime> provideDateTime)
        => new AzureStorageResponseCacheProvider(_dirClient!, provideDateTime);
}