// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
#nullable disable
 
using System.Text;
using System.Threading;
using Microsoft.CodeAnalysis.Host;
using Microsoft.CodeAnalysis.Shared.Extensions;
using Microsoft.CodeAnalysis.Test.Utilities;
using Microsoft.CodeAnalysis.Text;
using Roslyn.Utilities;
using Xunit;
 
namespace Microsoft.CodeAnalysis.UnitTests;
 
[UseExportProvider]
public sealed class SourceTextSerializationTests
{
    [Fact]
    public void TestSourceTextSerialization()
    {
        using var workspace = new AdhocWorkspace();
        var textService = Assert.IsType<TextFactoryService>(workspace.Services.GetService<ITextFactoryService>());
 
        var maxSize = SourceTextExtensions.SourceTextLengthThreshold * 3;
        var sb = new StringBuilder(0, maxSize);
 
        for (var i = 0; i < maxSize; i++)
        {
            var originalText = CreateSourceText(sb, i);
 
            using var stream = SerializableBytes.CreateWritableStream();
 
            using (var writer = new ObjectWriter(stream, leaveOpen: true))
            {
                originalText.WriteTo(writer, CancellationToken.None);
            }
 
            stream.Position = 0;
 
            using var reader = ObjectReader.TryGetReader(stream);
            var recovered = SourceTextExtensions.ReadFrom(textService, reader, originalText.Encoding, originalText.ChecksumAlgorithm, CancellationToken.None);
 
            Assert.Equal(originalText.ToString(), recovered.ToString());
        }
    }
 
    private static SourceText CreateSourceText(StringBuilder sb, int size)
    {
        for (var i = sb.Length; i < size; i++)
        {
            sb.Append((char)('0' + (i % 10)));
        }
 
        return SourceText.From(sb.ToString());
    }
}