tests: work on adding integration tests

2024-07-01 23:59:24 -05:00
parent c2788c3893
commit ebae27077e
6 changed files with 180 additions and 2 deletions
@@ -0,0 +1,34 @@
+namespace AnthropicClient.Tests.Integration;
+
+public class AnthropicApiClientTests : IntegrationTest
+{
+  [Theory]
+  [ClassData(typeof(ErrorTestData))]
+  public async Task CreateChatMessageAsync_WhenCalledAndErrorReturned_ItShouldHandleError(
+    HttpStatusCode statusCode,
+    string content,
+    Type errorType
+  )
+  {
+    _mockHttpMessageHandler
+      .WhenCreateMessageRequest()
+      .Respond(
+        statusCode,
+        "application/json",
+        content
+      );
+
+    var request = new ChatMessageRequest(
+      model: AnthropicModels.Claude3Haiku,
+      messages: [new(MessageRole.User, [new TextContent("Hello!")])]
+    );
+
+    var result = await Client.CreateChatMessageAsync(request);
+
+    result.IsSuccess.Should().BeFalse();
+    result.Error.Should().BeOfType<AnthropicError>();
+    
+    var actualErrorType = result.Error.Error!.GetType();
+    actualErrorType.Should().Be(errorType);
+  }
+}