测试 .net 内核 web api Z594C103F2C6E04C3D8AB059F031E0C 使用 XUnitCoreTestHost 文件上传。

Question

I have this controller to upload files:

[HttpPost]
public async Task<IActionResult> Post([FromForm] FileInformation fileInfo)
{
    int newFileVersion = 1;

    if (fileInfo == null || fileInfo.Files == null || fileInfo.Files.Count == 0)
        return BadRequest("File(s) not found");

    try
    {
        foreach (var locFile in fileInfo.Files)
        {
            //check for file extension, if not there, return an error
            var fileExtension = Path.GetExtension(locFile.FileName);
            if (string.IsNullOrEmpty(fileExtension))
                return BadRequest("Files must include file extension");


            var valid = await fileUtilities.IsFileValid(locFile);

            var newFileName = string.Concat(Guid.NewGuid().ToString(),valid.fileExtension);

            var newFileLocation = Path.Combine(config.GetSection("StoredFilePath").Value, newFileName);
           

            if (!valid.FileExtensionFound)
            {
                return BadRequest($"Error {valid.FileExtensionFoundError}");
            }
            if (!valid.FileSizeAllowed)
            {
                return BadRequest($"Error: {valid.FileSizeAllowedError}");
            }


            //check for an existing file in the database.  If there is one, increment the file version before the save
            var currentFile = await fileUtilities.FileExists(fileInfo, locFile);

            if (currentFile != null)
            {
                newFileVersion = currentFile.Version + 1;
            }
      
            //save to the file system
            using (var stream = new FileStream(newFileLocation, FileMode.OpenOrCreate, FileAccess.ReadWrite))
            {
                await locFile.CopyToAsync(stream);
            }

            //save to the db.  Check to see if the file exists first.  If it does, do an insert, if not, return an error
            if (System.IO.File.Exists(newFileLocation))
            {
                FileUploads upload = new FileUploads
                {
                    EntityId = fileInfo.EntityId,
                    FileName = locFile.FileName,
                    ItemId = fileInfo.ItemId.ToString(),
                    NewFileName = newFileName,
                    ValidFile = true,
                    Version = newFileVersion
                };
                context.FileUploads.Add(upload);
                context.SaveChanges();
                //TODO: fire event the file has been saved provide Id key to find the record
                //upload.Id;
            }
            else
            {
                return BadRequest("Error: File Could not be saved");
            }

        }
    }
    catch (Exception ex)
    {
        return BadRequest("Failure to upload files.");
    }
    return Ok("File Uploaded");
}

I am trying to write a unit text using XUnit and the Microsoft.AspNetCore.TestHost and I can't figure out how to write the test for the file upload. Here is one test I'm using to get a badrequest:

[Theory]
[InlineData("POST", "")]
public async Task LibrarianUploadFile_Error(string method, string? data = null)
{
    //arrange
    var request = new HttpRequestMessage(new HttpMethod(method), $"/api/librarian");

    //act
    var response = await _client.SendAsync(request);

    //assert
    Assert.Equal(HttpStatusCode.BadRequest, response.StatusCode);
}

This test is passing. I have written all the other test for the GET methods without a problem, I just don't know how to do this test. I was thinking I could have a test file on my local test machine to upload, I just don't know how to construct the POST.

Here is the my constructor for the Class:

private readonly HttpClient _client;

public LibrarianUploadFiles()
{
    var configuration = new ConfigurationBuilder()
       .AddJsonFile("appsettings.json")
       .Build();

    var server = new TestServer(new WebHostBuilder()
        .UseConfiguration(configuration)
        .UseStartup<Startup>());
    _client = server.CreateClient();
}

This creates the client I need to GET, POST, etc.

Answer 1

Here is the final solution to the problem. Using the answer from @gpaoli and a little looking around I got this code to post and pass the test.

 //arrange
        var request = new HttpRequestMessage(new HttpMethod(method), [url to post]);

        using var form = new MultipartFormDataContent();
        using var fileContent = new ByteArrayContent(await File.ReadAllBytesAsync(@"c:\files\test.txt"));
        fileContent.Headers.ContentType = MediaTypeHeaderValue.Parse("multipart/form-data");
        form.Add(fileContent, "Files", "test.txt");
        form.Add(new StringContent("Partner1"), "EntityId");
        form.Add(new StringContent(Guid.NewGuid().ToString()), "ItemId");

        //act
        HttpResponseMessage response;
        
        response = await _client.PostAsync($"/api/librarian", form);

        //assert

        Assert.Equal(HttpStatusCode.OK, response.StatusCode);

Couple of problems I had was adding the stringContent and making sure that the File, and string content was named correctly so the controller could map the values.

Answer 2

Do you need make post similar this:

// Act
HttpResponseMessage response;

using (var file = File.OpenRead(@"path\fileName.txt"))
using (var content = new StreamContent(file))
using (var formData = new MultipartFormDataContent())
{
    formData.Add(content, "fileInfo", "fileName.txt");

    response = await client.PostAsync(url, formData);
}

// Assert
response.EnsureSuccessStatusCode();