fix azure blob stream lenght

Author: TRybina132

ManagedCode.Storage.Azure/BlobStream.cs

@@ -8,6 +8,7 @@ namespace ManagedCode.Storage.Azure;
 
 public class BlobStream : Stream
 {
+    // TODO: add same thing to google and aws
     private const string MetadataLengthKey = "STREAM_LENGTH";
     private const int PageSizeInBytes = 512;
     public const int DefaultBufferSize = 1024 * 1024 * 4;
@@ -39,6 +40,7 @@ public override long Length
         {
             var realLength = 0L;
             var metadata = _pageBlob.GetProperties().Value.Metadata;
+            var contentLenght = _pageBlob.GetProperties().Value.ContentLength;
             if (metadata.TryGetValue(MetadataLengthKey, out var length))
             {
                 if (long.TryParse(length, out realLength))
@@ -47,8 +49,8 @@ public override long Length
                 }
             }
 
-            SetLengthInternal(realLength);
-            return realLength;
+            SetLengthInternal(contentLenght);
+            return contentLenght;
         }
     }
 

ManagedCode.Storage.Tests/Common/FileHelper.cs

@@ -25,6 +25,34 @@ public static LocalFile GenerateLocalFile(string fileName, int byteSize)
         return localFile;
     }
 
+    public static LocalFile GenerateLocalFileWithData(LocalFile file, int sizeInBytes)
+    {
+        using (var fileStream = file.FileStream)
+        {
+            Random random = new Random();
+            byte[] buffer = new byte[1024]; // Buffer for writing in chunks
+
+            while (sizeInBytes > 0)
+            {
+                int bytesToWrite = (int) Math.Min(sizeInBytes, buffer.Length);
+
+                for (int i = 0; i < bytesToWrite; i++)
+                {
+                    buffer[i] = (byte) random.Next(65, 91); // 'A' to 'Z'
+                    if (random.Next(2) == 0)
+                    {
+                        buffer[i] = (byte) random.Next(97, 123); // 'a' to 'z'
+                    }
+                }
+
+                fileStream.Write(buffer, 0, bytesToWrite);
+                sizeInBytes -= bytesToWrite;
+            }
+        }
+
+        return file;
+    }
+
     public static IFormFile GenerateFormFile(string fileName, int byteSize)
     {
         var localFile = GenerateLocalFile(fileName, byteSize);

ManagedCode.Storage.Tests/Storages/Azure/AzureBlobStreamTests.cs

@@ -1,6 +1,58 @@
+using System;
+using System.IO;
+using System.Threading.Tasks;
+using FluentAssertions;
+using ManagedCode.Storage.Azure;
+using ManagedCode.Storage.Core.Models;
+using ManagedCode.Storage.Tests.Common;
+using Microsoft.Extensions.DependencyInjection;
+using Testcontainers.Azurite;
+using Xunit;
+
 namespace ManagedCode.Storage.Tests.Storages.Azure;
 
-public class AzureBlobStreamTests
+public class AzureBlobStreamTests : StreamTests<AzuriteContainer>
 {
+    protected override AzuriteContainer Build()
+    {
+        return new AzuriteBuilder()
+            .WithImage("mcr.microsoft.com/azure-storage/azurite:3.26.0")
+            .Build();
+    }
+
+    protected override ServiceProvider ConfigureServices()
+    {
+        return AzureConfigurator.ConfigureServices(Container.GetConnectionString());
+    }
 
+    [Fact]
+    public async Task ReadStream_WhenFileExists_ReturnData()
+    {
+        // Arrange
+        var directory = "test-directory";
+        var fileName = Guid.NewGuid().ToString();
+        await using var localFile = LocalFile.FromRandomNameWithExtension(".txt");
+        FileHelper.GenerateLocalFileWithData(localFile, 10);
+        var storage = (IAzureStorage) Storage;
+        
+        UploadOptions options = new() { FileName = localFile.Name, Directory = directory };
+        var result = await storage.UploadAsync(localFile.FileInfo.OpenRead(), options);
+
+        await using var blobStream = storage.GetBlobStream(result.Value.FullName);
+
+        var chunkSize = localFile.FileInfo.Length;
+        byte[] fileChunk1 = new byte[chunkSize];
+        byte[] fileChunk2 = new byte[chunkSize];
+        
+        // Act
+        using var streamReader = new StreamReader(blobStream);
+        var content = await streamReader.ReadToEndAsync();
+        
+        // Assert
+        content.Should().NotBeNullOrEmpty();
+        using var fileReader = new StreamReader(localFile.FileStream);
+        var fileContent = await fileReader.ReadToEndAsync();
+        fileContent.Should().NotBeNullOrEmpty();
+        content.Should().Be(fileContent);
+    }
 }

ManagedCode.Storage.Tests/Storages/StreamTests.cs

@@ -8,9 +8,5 @@ namespace ManagedCode.Storage.Tests.Storages;
 public abstract class StreamTests<T> : BaseContainer<T>
     where T : IContainer
 {
-    [Fact]
-    public async Task ReadStream_WhenFileExists_ReturnData()
-    {
-        
-    }
+    
 }