上传图片接口Magic Bytes校验集成测试失败求助

核心问题分析 #

你的测试代码存在三个关键错误，直接导致Magic Bytes校验失败：

1. 错误使用文本流写入二进制文件
   StreamWriter是用于处理文本数据的工具，用它写入图片这类二进制文件会破坏原始字节结构，导致文件的Magic Bytes被篡改。

2. ContentType与实际文件类型不匹配
   测试中设置的ContentType是image/jpeg，但实际加载的是PNG格式的文件（test_image.PNG）。如果MagicBytes字典中image/jpeg对应的是JPG的魔数（FF D8），而PNG的魔数是89 50 4E 47，这会导致要么MagicBytes.TryGetValue找不到对应项返回false，要么魔数对比完全不匹配。

3. 内存流操作冗余且破坏原始文件内容
   代码中重复调用writer.Write(FileContent)向内存流追加额外内容，同时重复设置OpenReadStream的返回值，导致最终流中的数据不是原始图片的完整字节。

修正后的测试代码 #

private MultipartFormDataContent SetupFileImageTests(string fileName)
{
    // 加载真实的PNG文件
    using var realFile = File.OpenRead(ImageHelper.GetImagePath());
    var ms = new MemoryStream();
    
    // 使用二进制方式复制文件内容，避免破坏字节结构
    realFile.CopyTo(ms);
    ms.Position = 0;

    var mockedFile = new Mock<IFormFile>();
    mockedFile.Setup(f => f.OpenReadStream()).Returns(ms);
    mockedFile.Setup(f => f.FileName).Returns(fileName);
    mockedFile.Setup(f => f.Length).Returns(ms.Length);
    mockedFile.Setup(_ => _.CopyToAsync(It.IsAny<Stream>(), It.IsAny<CancellationToken>()))
        .Returns((Stream stream, CancellationToken token) => ms.CopyToAsync(stream, token))
        .Verifiable();

    // 匹配实际文件类型设置ContentType
    var fileContent = new StreamContent(mockedFile.Object.OpenReadStream())
    {
        Headers =
        {
            ContentLength = mockedFile.Object.Length,
            ContentType = new MediaTypeHeaderValue("image/png")
        }
    };

    return new MultipartFormDataContent
    {
        {fileContent, "file", fileName}
    };
}

额外校验建议 #

1. 确认MagicBytes字典中包含image/png对应的PNG魔数（new byte[] { 0x89, 0x50, 0x4E, 0x47 }）。
2. 校验代码中stream.Read不能保证一次性读取到足够的字节数，建议改为循环读取确保获取完整的Magic Bytes：

int bytesRead = 0;
while (bytesRead < magicBytes.Length)
{
    int read = stream.Read(fileMagicBytes, bytesRead, magicBytes.Length - bytesRead);
    if (read == 0) break; // 文件长度不足，直接返回false
    bytesRead += read;
}
bool haveValidMagicBytes = bytesRead == magicBytes.Length && magicBytes.SequenceEqual(fileMagicBytes);

内容的提问来源于stack exchange，提问作者Gilles De Vylder