using System.IO;
using SharpCompress.Common;
using SharpCompress.Compressors.Xz;
using Xunit;

namespace SharpCompress.Test.Xz;

public class XzHeaderTests : XzTestsBase
{
    [Fact]
    public void ChecksMagicNumber()
    {
        var bytes = (byte[])Compressed.Clone();
        bytes[3]++;
        using Stream badMagicNumberStream = new MemoryStream(bytes);
        using var br = new BinaryReader(badMagicNumberStream);
        var header = new XZHeader(br);
        var ex = Assert.Throws<InvalidFormatException>(() =>
        {
            header.Process();
        });
        Assert.Equal("Invalid XZ Stream", ex.Message);
    }

    [Fact]
    public void CorruptHeaderThrows()
    {
        var bytes = (byte[])Compressed.Clone();
        bytes[8]++;
        using Stream badCrcStream = new MemoryStream(bytes);
        using var br = new BinaryReader(badCrcStream);
        var header = new XZHeader(br);
        var ex = Assert.Throws<InvalidFormatException>(() =>
        {
            header.Process();
        });
        Assert.Equal("Stream header corrupt", ex.Message);
    }

    [Fact]
    public void BadVersionIfCrcOkButStreamFlagUnknown()
    {
        var bytes = (byte[])Compressed.Clone();
        byte[] streamFlags = [0x00, 0xF4];
        var crc = Crc32.Compute(streamFlags).ToLittleEndianBytes();
        streamFlags.CopyTo(bytes, 6);
        crc.CopyTo(bytes, 8);
        using Stream badFlagStream = new MemoryStream(bytes);
        using var br = new BinaryReader(badFlagStream);
        var header = new XZHeader(br);
        var ex = Assert.Throws<InvalidFormatException>(() =>
        {
            header.Process();
        });
        Assert.Equal("Unknown XZ Stream Version", ex.Message);
    }

    [Fact]
    public void ProcessesBlockCheckType()
    {
        using var br = new BinaryReader(CompressedStream);
        var header = new XZHeader(br);
        header.Process();
        Assert.Equal(CheckType.CRC64, header.BlockCheckType);
    }

    [Fact]
    public void CanCalculateBlockCheckSize()
    {
        using var br = new BinaryReader(CompressedStream);
        var header = new XZHeader(br);
        header.Process();
        Assert.Equal(8, header.BlockCheckSize);
    }

    [Fact]
    public void ProcessesStreamHeaderFromFactory()
    {
        var header = XZHeader.FromStream(CompressedStream);
        Assert.Equal(CheckType.CRC64, header.BlockCheckType);
    }
}
