Implement file support with models, interface, and comprehensive tests
Co-authored-by: StevanFreeborn <65925598+StevanFreeborn@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
co-authored by
StevanFreeborn
StevanFreeborn
parent
5d5b60486f
commit
84ce38a586
@@ -0,0 +1,219 @@
|
||||
namespace AnthropicClient.Tests.Integration;
|
||||
|
||||
public class AnthropicApiClientFileTests : IntegrationTest
|
||||
{
|
||||
[Fact]
|
||||
public async Task CreateFileAsync_WhenCalled_ItShouldReturnFileResponse()
|
||||
{
|
||||
var fileResponseJson = @"{
|
||||
""type"": ""file"",
|
||||
""id"": ""file_abc123"",
|
||||
""filename"": ""example.txt"",
|
||||
""content_type"": ""text/plain"",
|
||||
""size_bytes"": 1024,
|
||||
""created_at"": ""2024-03-15T10:30:00Z""
|
||||
}";
|
||||
|
||||
_mockHttpMessageHandler
|
||||
.WhenCreateFileRequest()
|
||||
.Respond(HttpStatusCode.OK, "application/json", fileResponseJson);
|
||||
|
||||
var content = "Hello World"u8.ToArray();
|
||||
var request = new FileRequest(content, "example.txt", "text/plain");
|
||||
|
||||
var result = await Client.CreateFileAsync(request);
|
||||
|
||||
result.IsSuccess.Should().BeTrue();
|
||||
result.Value.Should().NotBeNull();
|
||||
result.Value!.Id.Should().Be("file_abc123");
|
||||
result.Value.Filename.Should().Be("example.txt");
|
||||
result.Value.ContentType.Should().Be("text/plain");
|
||||
result.Value.SizeBytes.Should().Be(1024);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ListFilesAsync_WhenCalled_ItShouldReturnFilesPage()
|
||||
{
|
||||
var filesResponseJson = @"{
|
||||
""data"": [
|
||||
{
|
||||
""type"": ""file"",
|
||||
""id"": ""file_abc123"",
|
||||
""filename"": ""example.txt"",
|
||||
""content_type"": ""text/plain"",
|
||||
""size_bytes"": 1024,
|
||||
""created_at"": ""2024-03-15T10:30:00Z""
|
||||
}
|
||||
],
|
||||
""has_more"": false,
|
||||
""first_id"": ""file_abc123"",
|
||||
""last_id"": ""file_abc123""
|
||||
}";
|
||||
|
||||
_mockHttpMessageHandler
|
||||
.WhenListFilesRequest()
|
||||
.Respond(HttpStatusCode.OK, "application/json", filesResponseJson);
|
||||
|
||||
var result = await Client.ListFilesAsync();
|
||||
|
||||
result.IsSuccess.Should().BeTrue();
|
||||
result.Value.Should().NotBeNull();
|
||||
result.Value!.Data.Should().HaveCount(1);
|
||||
result.Value.Data[0].Id.Should().Be("file_abc123");
|
||||
result.Value.HasMore.Should().BeFalse();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task GetFileAsync_WhenCalled_ItShouldReturnFileResponse()
|
||||
{
|
||||
var fileResponseJson = @"{
|
||||
""type"": ""file"",
|
||||
""id"": ""file_abc123"",
|
||||
""filename"": ""example.txt"",
|
||||
""content_type"": ""text/plain"",
|
||||
""size_bytes"": 1024,
|
||||
""created_at"": ""2024-03-15T10:30:00Z""
|
||||
}";
|
||||
|
||||
_mockHttpMessageHandler
|
||||
.WhenGetFileRequest("file_abc123")
|
||||
.Respond(HttpStatusCode.OK, "application/json", fileResponseJson);
|
||||
|
||||
var result = await Client.GetFileAsync("file_abc123");
|
||||
|
||||
result.IsSuccess.Should().BeTrue();
|
||||
result.Value.Should().NotBeNull();
|
||||
result.Value!.Id.Should().Be("file_abc123");
|
||||
result.Value.Filename.Should().Be("example.txt");
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task DownloadFileAsync_WhenCalled_ItShouldReturnFileContent()
|
||||
{
|
||||
var fileContent = "Hello World"u8.ToArray();
|
||||
|
||||
var httpResponseMessage = new HttpResponseMessage(HttpStatusCode.OK)
|
||||
{
|
||||
Content = new ByteArrayContent(fileContent)
|
||||
};
|
||||
httpResponseMessage.Content.Headers.ContentType = new System.Net.Http.Headers.MediaTypeHeaderValue("text/plain");
|
||||
httpResponseMessage.Content.Headers.ContentDisposition = new System.Net.Http.Headers.ContentDispositionHeaderValue("attachment")
|
||||
{
|
||||
FileName = "\"example.txt\""
|
||||
};
|
||||
|
||||
_mockHttpMessageHandler
|
||||
.WhenDownloadFileRequest("file_abc123")
|
||||
.Respond(_ => httpResponseMessage);
|
||||
|
||||
var result = await Client.DownloadFileAsync("file_abc123");
|
||||
|
||||
result.IsSuccess.Should().BeTrue();
|
||||
result.Value.Should().NotBeNull();
|
||||
result.Value!.Content.Should().BeEquivalentTo(fileContent);
|
||||
result.Value.Filename.Should().Be("example.txt");
|
||||
result.Value.ContentType.Should().Be("text/plain");
|
||||
result.Value.SizeBytes.Should().Be(fileContent.Length);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task DeleteFileAsync_WhenCalled_ItShouldReturnDeleteResponse()
|
||||
{
|
||||
var deleteResponseJson = @"{
|
||||
""type"": ""file_deleted"",
|
||||
""id"": ""file_abc123"",
|
||||
""deleted"": true
|
||||
}";
|
||||
|
||||
_mockHttpMessageHandler
|
||||
.WhenDeleteFileRequest("file_abc123")
|
||||
.Respond(HttpStatusCode.OK, "application/json", deleteResponseJson);
|
||||
|
||||
var result = await Client.DeleteFileAsync("file_abc123");
|
||||
|
||||
result.IsSuccess.Should().BeTrue();
|
||||
result.Value.Should().NotBeNull();
|
||||
result.Value!.Id.Should().Be("file_abc123");
|
||||
result.Value.Deleted.Should().BeTrue();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task CreateFileAsync_WhenCalledAndErrorReturned_ItShouldHandleError()
|
||||
{
|
||||
var errorJson = @"{
|
||||
""type"": ""error"",
|
||||
""error"": {
|
||||
""type"": ""invalid_request_error"",
|
||||
""message"": ""File too large""
|
||||
}
|
||||
}";
|
||||
|
||||
_mockHttpMessageHandler
|
||||
.WhenCreateFileRequest()
|
||||
.Respond(HttpStatusCode.BadRequest, "application/json", errorJson);
|
||||
|
||||
var content = "Hello World"u8.ToArray();
|
||||
var request = new FileRequest(content, "example.txt", "text/plain");
|
||||
|
||||
var result = await Client.CreateFileAsync(request);
|
||||
|
||||
result.IsSuccess.Should().BeFalse();
|
||||
result.Error.Should().BeOfType<AnthropicError>();
|
||||
result.Error.Error.Should().BeOfType<InvalidRequestError>();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ListAllFilesAsync_WhenCalled_ItShouldReturnAllPages()
|
||||
{
|
||||
var firstPageJson = @"{
|
||||
""data"": [
|
||||
{
|
||||
""type"": ""file"",
|
||||
""id"": ""file_abc123"",
|
||||
""filename"": ""example1.txt"",
|
||||
""content_type"": ""text/plain"",
|
||||
""size_bytes"": 1024,
|
||||
""created_at"": ""2024-03-15T10:30:00Z""
|
||||
}
|
||||
],
|
||||
""has_more"": true,
|
||||
""first_id"": ""file_abc123"",
|
||||
""last_id"": ""file_abc123""
|
||||
}";
|
||||
|
||||
var secondPageJson = @"{
|
||||
""data"": [
|
||||
{
|
||||
""type"": ""file"",
|
||||
""id"": ""file_def456"",
|
||||
""filename"": ""example2.txt"",
|
||||
""content_type"": ""text/plain"",
|
||||
""size_bytes"": 2048,
|
||||
""created_at"": ""2024-03-15T11:30:00Z""
|
||||
}
|
||||
],
|
||||
""has_more"": false,
|
||||
""first_id"": ""file_def456"",
|
||||
""last_id"": ""file_def456""
|
||||
}";
|
||||
|
||||
_mockHttpMessageHandler
|
||||
.WhenListFilesRequest()
|
||||
.Respond(HttpStatusCode.OK, "application/json", firstPageJson);
|
||||
|
||||
_mockHttpMessageHandler
|
||||
.WhenListFilesRequest()
|
||||
.Respond(HttpStatusCode.OK, "application/json", secondPageJson);
|
||||
|
||||
var pages = new List<AnthropicResult<Page<AnthropicFile>>>();
|
||||
await foreach (var page in Client.ListAllFilesAsync(1))
|
||||
{
|
||||
pages.Add(page);
|
||||
}
|
||||
|
||||
pages.Should().HaveCount(2);
|
||||
pages.All(p => p.IsSuccess).Should().BeTrue();
|
||||
pages[0].Value!.Data[0].Id.Should().Be("file_abc123");
|
||||
pages[1].Value!.Data[0].Id.Should().Be("file_def456");
|
||||
}
|
||||
}
|
||||
@@ -20,6 +20,7 @@ public static class MockHttpMessageHandlerExtensions
|
||||
private static readonly string CountTokensEndpoint = $"{BaseUrl}/messages/count_tokens";
|
||||
private static readonly string MessageBatchesEndpoint = $"{BaseUrl}/messages/batches";
|
||||
private static readonly string ModelsEndpoint = $"{BaseUrl}/models";
|
||||
private static readonly string FilesEndpoint = $"{BaseUrl}/files";
|
||||
|
||||
private static MockedRequest SetupBaseRequest(
|
||||
this MockHttpMessageHandler mockHttpMessageHandler,
|
||||
@@ -103,4 +104,34 @@ public static class MockHttpMessageHandlerExtensions
|
||||
return mockHttpMessageHandler
|
||||
.SetupBaseRequest(HttpMethod.Delete, $"{MessageBatchesEndpoint}/{batchId}");
|
||||
}
|
||||
|
||||
public static MockedRequest WhenCreateFileRequest(this MockHttpMessageHandler mockHttpMessageHandler)
|
||||
{
|
||||
return mockHttpMessageHandler
|
||||
.SetupBaseRequest(HttpMethod.Post, FilesEndpoint);
|
||||
}
|
||||
|
||||
public static MockedRequest WhenListFilesRequest(this MockHttpMessageHandler mockHttpMessageHandler)
|
||||
{
|
||||
return mockHttpMessageHandler
|
||||
.SetupBaseRequest(HttpMethod.Get, FilesEndpoint);
|
||||
}
|
||||
|
||||
public static MockedRequest WhenGetFileRequest(this MockHttpMessageHandler mockHttpMessageHandler, string fileId)
|
||||
{
|
||||
return mockHttpMessageHandler
|
||||
.SetupBaseRequest(HttpMethod.Get, $"{FilesEndpoint}/{fileId}");
|
||||
}
|
||||
|
||||
public static MockedRequest WhenDownloadFileRequest(this MockHttpMessageHandler mockHttpMessageHandler, string fileId)
|
||||
{
|
||||
return mockHttpMessageHandler
|
||||
.SetupBaseRequest(HttpMethod.Get, $"{FilesEndpoint}/{fileId}/content");
|
||||
}
|
||||
|
||||
public static MockedRequest WhenDeleteFileRequest(this MockHttpMessageHandler mockHttpMessageHandler, string fileId)
|
||||
{
|
||||
return mockHttpMessageHandler
|
||||
.SetupBaseRequest(HttpMethod.Delete, $"{FilesEndpoint}/{fileId}");
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user