Stevan/anthropic-client
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tests: add tests for creating tools with cache control set

Browse Source
This commit is contained in:
Stevan Freeborn
2024-08-18 14:19:55 -05:00
parent 539d735e6b
commit 42d6ee4d34
1 changed files with 138 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/AnthropicClient.Tests/Unit/Models/ToolTests.cs
+138
View File
@@ -70,6 +70,17 @@ public class ToolTests : SerializationTest
tool.Name.Should().HaveLength(64);
}
[Fact]
public void Constructor_WhenCalledAndGivenCacheControl_ItShouldInitializeCacheControl()
{
var method = () => true;
var function = new AnthropicFunction(method.Method);
var cacheControl = new EphemeralCacheControl();
var tool = new Tool("name", "description", function, cacheControl);
tool.CacheControl.Should().BeSameAs(cacheControl);
}
[Theory]
[InlineData(typeof(TestClass), null)]
[InlineData(typeof(TestClass), "")]
@@ -119,6 +130,29 @@ public class ToolTests : SerializationTest
);
}
[Fact]
public void CreateFromStaticMethod_WhenCalledWithCacheControl_ItShouldReturnTool()
{
var cacheControl = new EphemeralCacheControl();
var tool = Tool.CreateFromStaticMethod("test name", "description", typeof(TestClass), nameof(TestClass.TestStaticMethod), cacheControl);
var expectedSchema = new JsonObject()
{
["type"] = "object",
};
tool.Name.Should().Be("test_name");
tool.DisplayName.Should().Be("test name");
tool.Description.Should().Be("description");
tool.Function.Method.Name.Should().Be(nameof(TestClass.TestStaticMethod));
tool.Function.Instance.Should().BeNull();
tool.InputSchema.Should().BeEquivalentTo(
expectedSchema,
t => t.IgnoringCyclicReferences()
);
tool.CacheControl.Should().BeSameAs(cacheControl);
}
[Theory]
[InlineData(null)]
[InlineData("")]
@@ -177,6 +211,30 @@ public class ToolTests : SerializationTest
);
}
[Fact]
public void CreateFromInstanceMethod_WhenCalledWithCacheControl_ItShouldReturnTool()
{
var instance = new TestClass();
var cacheControl = new EphemeralCacheControl();
var tool = Tool.CreateFromInstanceMethod("test name", "description", instance, nameof(instance.TestInstanceMethod), cacheControl);
var expectedSchema = new JsonObject()
{
["type"] = "object",
};
tool.Name.Should().Be("test_name");
tool.DisplayName.Should().Be("test name");
tool.Description.Should().Be("description");
tool.Function.Method.Name.Should().Be(nameof(TestClass.TestInstanceMethod));
tool.Function.Instance.Should().Be(instance);
tool.InputSchema.Should().BeEquivalentTo(
expectedSchema,
t => t.IgnoringCyclicReferences()
);
tool.CacheControl.Should().BeSameAs(cacheControl);
}
[Fact]
public void CreateFromParameterlessFunction_WhenGivenNullFunction_ItShouldThrowArgumentNullException()
{
@@ -206,6 +264,29 @@ public class ToolTests : SerializationTest
);
}
[Fact]
public void CreateFromParameterlessFunction_WhenCalledWithCacheControl_ItShouldReturnTool()
{
var func = () => true;
var cacheControl = new EphemeralCacheControl();
var tool = Tool.CreateFromFunction("test name", "description", func, cacheControl);
var expectedSchema = new JsonObject()
{
["type"] = "object",
};
tool.Name.Should().Be("test_name");
tool.DisplayName.Should().Be("test name");
tool.Description.Should().Be("description");
tool.Function.Method.Should().BeSameAs(func.Method);
tool.InputSchema.Should().BeEquivalentTo(
expectedSchema,
t => t.IgnoringCyclicReferences()
);
tool.CacheControl.Should().BeSameAs(cacheControl);
}
[Fact]
public void CreateFromFunctionWithParameter_WhenGivenNullFunction_ItShouldThrowArgumentNullException()
{
@@ -246,6 +327,40 @@ public class ToolTests : SerializationTest
);
}
[Fact]
public void CreateFromFunctionWithParameter_WhenCalledWithCacheControl_ItShouldReturnTool()
{
var func = (string s) => true;
var cacheControl = new EphemeralCacheControl();
var tool = Tool.CreateFromFunction("test name", "description", func, cacheControl);
var expectedSchema = new JsonObject()
{
["type"] = "object",
["properties"] = new JsonObject()
{
["s"] = new JsonObject()
{
["type"] = "string",
},
},
["required"] = new JsonArray()
{
"s",
},
};
tool.Name.Should().Be("test_name");
tool.DisplayName.Should().Be("test name");
tool.Description.Should().Be("description");
tool.Function.Method.Should().BeSameAs(func.Method);
tool.InputSchema.Should().BeEquivalentTo(
expectedSchema,
t => t.IgnoringCyclicReferences()
);
tool.CacheControl.Should().BeSameAs(cacheControl);
}
[Fact]
public void CreateFromClass_WhenCalledWithToolWhoseNameIsNull_ItShouldThrowException()
{
@@ -306,6 +421,29 @@ public class ToolTests : SerializationTest
t => t.IgnoringCyclicReferences()
);
}
[Fact]
public void CreateFromClass_WhenCalledWithProperToolAndCacheControl_ItShouldReturnTool()
{
var cacheControl = new EphemeralCacheControl();
var tool = Tool.CreateFromClass<ProperTool>(cacheControl);
var expectedSchema = new JsonObject()
{
["type"] = "object",
};
tool.Name.Should().Be("Name");
tool.DisplayName.Should().Be("Name");
tool.Description.Should().Be("Description");
tool.Function.Method.Name.Should().Be(nameof(ProperTool.GetWeather));
tool.Function.Instance.Should().BeOfType<ProperTool>();
tool.InputSchema.Should().BeEquivalentTo(
expectedSchema,
t => t.IgnoringCyclicReferences()
);
tool.CacheControl.Should().BeSameAs(cacheControl);
}
}
class TestClass