feat: add model for DocumentContent and DocumentSource
This commit is contained in:
@@ -0,0 +1,119 @@
|
||||
namespace AnthropicClient.Tests.Unit.Models;
|
||||
|
||||
public class DocumentContentTests : SerializationTest
|
||||
{
|
||||
private readonly string _testJson = @"{
|
||||
""source"": {
|
||||
""media_type"": ""application/pdf"",
|
||||
""data"": ""data"",
|
||||
""type"": ""base64""
|
||||
},
|
||||
""type"": ""document""
|
||||
}";
|
||||
|
||||
private readonly string _testJsonWithCacheControl = @"{
|
||||
""source"": {
|
||||
""media_type"": ""application/pdf"",
|
||||
""data"": ""data"",
|
||||
""type"": ""base64""
|
||||
},
|
||||
""cache_control"": { ""type"": ""ephemeral"" },
|
||||
""type"": ""document""
|
||||
}";
|
||||
|
||||
[Fact]
|
||||
public void Constructor_WhenCalled_ItShouldInitializeSource()
|
||||
{
|
||||
var expectedMediaType = "application/pdf";
|
||||
var expectedData = "data";
|
||||
|
||||
var result = new DocumentContent(expectedMediaType, expectedData);
|
||||
|
||||
result.Source.Should().BeEquivalentTo(new DocumentSource(expectedMediaType, expectedData));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void Constructor_WhenCalledAndMediaTypeIsNull_ItShouldThrowArgumentNullException()
|
||||
{
|
||||
var expectedData = "data";
|
||||
|
||||
var action = () => new DocumentContent(null!, expectedData);
|
||||
|
||||
action.Should().Throw<ArgumentNullException>();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void Constructor_WhenCalledAndDataIsNull_ItShouldThrowArgumentNullException()
|
||||
{
|
||||
var expectedMediaType = "application/pdf";
|
||||
|
||||
var action = () => new DocumentContent(expectedMediaType, null!);
|
||||
|
||||
action.Should().Throw<ArgumentNullException>();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void Constructor_WhenCalledWithCacheControl_ItShouldInitializeSourceAndCacheControl()
|
||||
{
|
||||
var expectedMediaType = "application/pdf";
|
||||
var expectedData = "data";
|
||||
var cacheControl = new EphemeralCacheControl();
|
||||
|
||||
var result = new DocumentContent(expectedMediaType, expectedData, cacheControl);
|
||||
|
||||
result.Source.Should().BeEquivalentTo(new DocumentSource(expectedMediaType, expectedData));
|
||||
result.CacheControl.Should().BeSameAs(cacheControl);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void Constructor_WhenCalledWithCacheControlAndMediaTypeIsNull_ItShouldThrowArgumentNullException()
|
||||
{
|
||||
var expectedData = "data";
|
||||
var cacheControl = new EphemeralCacheControl();
|
||||
|
||||
var action = () => new DocumentContent(null!, expectedData, cacheControl);
|
||||
|
||||
action.Should().Throw<ArgumentNullException>();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void Constructor_WhenCalledWithCacheControlAndDataIsNull_ItShouldThrowArgumentNullException()
|
||||
{
|
||||
var expectedMediaType = "application/pdf";
|
||||
var cacheControl = new EphemeralCacheControl();
|
||||
|
||||
var action = () => new DocumentContent(expectedMediaType, null!, cacheControl);
|
||||
|
||||
action.Should().Throw<ArgumentNullException>();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void JsonSerialization_WhenSerialized_ItShouldHaveExpectedShape()
|
||||
{
|
||||
var content = new DocumentContent("application/pdf", "data");
|
||||
|
||||
var actual = Serialize(content);
|
||||
|
||||
JsonAssert.Equal(_testJson, actual);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void JsonSerialization_WhenSerializedWithCacheControl_ItShouldHaveExpectedShape()
|
||||
{
|
||||
var content = new DocumentContent("application/pdf", "data", new EphemeralCacheControl());
|
||||
|
||||
var actual = Serialize(content);
|
||||
|
||||
JsonAssert.Equal(_testJsonWithCacheControl, actual);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void JsonDeserialization_WhenDeserialized_ItShouldHaveExpectedShape()
|
||||
{
|
||||
var expected = new DocumentContent("application/pdf", "data");
|
||||
|
||||
var actual = Deserialize<DocumentContent>(_testJson);
|
||||
|
||||
actual.Should().BeEquivalentTo(expected);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,49 @@
|
||||
namespace AnthropicClient.Tests.Unit.Models;
|
||||
|
||||
public class DocumentSourceTests
|
||||
{
|
||||
[Fact]
|
||||
public void Constructor_WhenCalledWithValidArguments_ItShouldSetProperties()
|
||||
{
|
||||
var mediaType = "application/pdf";
|
||||
var data = "base64data";
|
||||
|
||||
var source = new DocumentSource(mediaType, data);
|
||||
|
||||
source.MediaType.Should().Be(mediaType);
|
||||
source.Data.Should().Be(data);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void Constructor_WhenCalledWithNullMediaType_ItShouldThrowArgumentNullException()
|
||||
{
|
||||
string? mediaType = null;
|
||||
var data = "base64data";
|
||||
|
||||
var action = () => new DocumentSource(mediaType!, data);
|
||||
|
||||
action.Should().Throw<ArgumentNullException>();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void Constructor_WhenCalledWithNullData_ItShouldThrowArgumentNullException()
|
||||
{
|
||||
var mediaType = "application/pdf";
|
||||
string? data = null;
|
||||
|
||||
var action = () => new DocumentSource(mediaType, data!);
|
||||
|
||||
action.Should().Throw<ArgumentNullException>();
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void Constructor_WhenCalled_ItShouldHaveTypePropertySetToBase64()
|
||||
{
|
||||
var mediaType = "application/pdf";
|
||||
var data = "base64data";
|
||||
|
||||
var source = new DocumentSource(mediaType, data);
|
||||
|
||||
source.Type.Should().Be("base64");
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user