tests: add missing tests for streaming chat message

2024-06-29 22:07:49 -05:00
parent 77ceb49c77
commit 5edc0b5fff
16 changed files with 1183 additions and 7 deletions
@@ -0,0 +1,43 @@
+namespace AnthropicClient.Tests.Unit.Models;
+
+public class MessageDeltaTests : SerializationTest
+{
+  private readonly string _testJson = @"{
+    ""stop_reason"": ""max_tokens"",
+    ""stop_sequence"": ""max_tokens""
+  }";
+
+  [Fact]
+  public void Constructor_WhenCalled_ItShouldInitializeProperties()
+  {
+    var expectedStopReason = "max_tokens";
+    var expectedStopSequence = "max_tokens";
+
+    var messageDelta = new MessageDelta(expectedStopReason, expectedStopSequence);
+
+    messageDelta.StopReason.Should().Be(expectedStopReason);
+    messageDelta.StopSequence.Should().Be(expectedStopSequence);
+  }
+
+  [Fact]
+  public void JsonSerialization_WhenSerialized_ItShouldHaveExpectedShape()
+  {
+    var messageDelta = new MessageDelta("max_tokens", "max_tokens");
+
+    var actualJson = Serialize(messageDelta);
+
+    JsonAssert.Equal(_testJson, actualJson);
+  }
+
+  [Fact]
+  public void JsonDeserialization_WhenDeserialized_ItShouldHaveExpectedProperties()
+  {
+    var expectedStopReason = "max_tokens";
+    var expectedStopSequence = "max_tokens";
+
+    var messageDelta = Deserialize<MessageDelta>(_testJson);
+
+    messageDelta!.StopReason.Should().Be(expectedStopReason);
+    messageDelta.StopSequence.Should().Be(expectedStopSequence);
+  }
+}