Checkstyle fix

Author: sobychacko

models/spring-ai-anthropic/src/test/java/org/springframework/ai/anthropic/client/AnthropicChatClientMethodFunctionCallbackIT.java

@@ -36,7 +36,7 @@
 import org.springframework.util.ReflectionUtils;
 
 import static org.assertj.core.api.Assertions.assertThat;
-import static org.junit.Assert.assertThrows;
+import static org.assertj.core.api.AssertionsForClassTypes.assertThatThrownBy;
 
 @SpringBootTest(classes = AnthropicTestConfiguration.class, properties = "spring.ai.retry.on-http-codes=429")
 @EnabledIfEnvironmentVariable(named = "ANTHROPIC_API_KEY", matches = ".+")
@@ -47,66 +47,6 @@ class AnthropicChatClientMethodFunctionCallbackIT {
 
 	public static Map<String, Object> arguments = new ConcurrentHashMap<>();
 
-	@Autowired
-	ChatModel chatModel;
-
-	record MyRecord(String foo, String bar) {
-	}
-
-	public enum Unit {
-
-		CELSIUS, FAHRENHEIT
-
-	}
-
-	public static class TestFunctionClass {
-
-		public static void argumentLessReturnVoid() {
-			arguments.put("method called", "argumentLessReturnVoid");
-		}
-
-		public static String getWeatherStatic(String city, Unit unit) {
-
-			logger.info("City: " + city + " Unit: " + unit);
-
-			arguments.put("city", city);
-			arguments.put("unit", unit);
-
-			double temperature = 0;
-			if (city.contains("Paris")) {
-				temperature = 15;
-			}
-			else if (city.contains("Tokyo")) {
-				temperature = 10;
-			}
-			else if (city.contains("San Francisco")) {
-				temperature = 30;
-			}
-
-			return "temperature: " + temperature + " unit: " + unit;
-		}
-
-		public String getWeatherNonStatic(String city, Unit unit) {
-			return getWeatherStatic(city, unit);
-		}
-
-		public String getWeatherWithContext(String city, Unit unit, ToolContext context) {
-			arguments.put("tool", context.getContext().get("tool"));
-			return getWeatherStatic(city, unit);
-		}
-
-		public void turnLight(String roomName, boolean on) {
-			arguments.put("roomName", roomName);
-			arguments.put("on", on);
-			logger.info("Turn light in room: {} to: {}", roomName, on);
-		}
-
-		public void turnLivingRoomLightOn() {
-			arguments.put("turnLivingRoomLightOn", true);
-		}
-
-	}
-
 	@BeforeEach
 	void beforeEach() {
 		arguments.clear();
@@ -120,7 +60,7 @@ void methodGetWeatherStatic() {
 		String response = ChatClient.create(this.chatModel).prompt()
 				.user("What's the weather like in San Francisco, Tokyo, and Paris?  Use Celsius.")
 				.functions(MethodFunctionCallback.builder()
-					.method(method)						
+					.method(method)
 					.description("Get the weather in location")
 					.build())
 				.call()
@@ -144,7 +84,7 @@ void methodTurnLightNoResponse() {
 				.user("Turn light on in the living room.")
 				.functions(MethodFunctionCallback.builder()
 					.functionObject(targetObject)
-					.method(method)						
+					.method(method)
 					.description("Can turn lights on or off by room name")
 					.build())
 				.call()
@@ -170,7 +110,7 @@ void methodGetWeatherNonStatic() {
 				.user("What's the weather like in San Francisco, Tokyo, and Paris?  Use Celsius.")
 				.functions(MethodFunctionCallback.builder()
 					.functionObject(targetObject)
-					.method(method)						
+					.method(method)
 					.description("Get the weather in location")
 					.build())
 				.call()
@@ -195,7 +135,7 @@ void methodGetWeatherToolContext() {
 				.user("What's the weather like in San Francisco, Tokyo, and Paris?  Use Celsius.")
 				.functions(MethodFunctionCallback.builder()
 					.functionObject(targetObject)
-					.method(method)						
+					.method(method)
 					.description("Get the weather in location")
 					.build())
 				.toolContext(Map.of("tool", "value"))
@@ -218,18 +158,18 @@ void methodGetWeatherToolContextButNonContextMethod() {
 				Unit.class);
 
 		// @formatter:off
-		assertThrows("Configured method does not accept ToolContext as input parameter!",IllegalArgumentException.class, () -> {
-			ChatClient.create(this.chatModel).prompt()
+		assertThatThrownBy(() -> ChatClient.create(this.chatModel).prompt()
 				.user("What's the weather like in San Francisco, Tokyo, and Paris?  Use Celsius.")
 				.functions(MethodFunctionCallback.builder()
-					.functionObject(targetObject)
-					.method(method)						
-					.description("Get the weather in location")
-					.build())
+						.functionObject(targetObject)
+						.method(method)
+						.description("Get the weather in location")
+						.build())
 				.toolContext(Map.of("tool", "value"))
 				.call()
-				.content();
-		});
+				.content())
+				.isInstanceOf(IllegalArgumentException.class)
+				.hasMessage("Configured method does not accept ToolContext as input parameter!");
 		// @formatter:on
 	}
 
@@ -245,7 +185,7 @@ void methodNoParameters() {
 				.user("Turn light on in the living room.")
 				.functions(MethodFunctionCallback.builder()
 					.functionObject(targetObject)
-					.method(method)						
+					.method(method)
 					.description("Can turn lights on in the Living Room")
 					.build())
 				.call()
@@ -257,4 +197,64 @@ void methodNoParameters() {
 		assertThat(arguments).containsEntry("turnLivingRoomLightOn", true);
 	}
 
+	@Autowired
+	ChatModel chatModel;
+
+	record MyRecord(String foo, String bar) {
+	}
+
+	public enum Unit {
+
+		CELSIUS, FAHRENHEIT
+
+	}
+
+	public static class TestFunctionClass {
+
+		public static void argumentLessReturnVoid() {
+			arguments.put("method called", "argumentLessReturnVoid");
+		}
+
+		public static String getWeatherStatic(String city, Unit unit) {
+
+			logger.info("City: " + city + " Unit: " + unit);
+
+			arguments.put("city", city);
+			arguments.put("unit", unit);
+
+			double temperature = 0;
+			if (city.contains("Paris")) {
+				temperature = 15;
+			}
+			else if (city.contains("Tokyo")) {
+				temperature = 10;
+			}
+			else if (city.contains("San Francisco")) {
+				temperature = 30;
+			}
+
+			return "temperature: " + temperature + " unit: " + unit;
+		}
+
+		public String getWeatherNonStatic(String city, Unit unit) {
+			return getWeatherStatic(city, unit);
+		}
+
+		public String getWeatherWithContext(String city, Unit unit, ToolContext context) {
+			arguments.put("tool", context.getContext().get("tool"));
+			return getWeatherStatic(city, unit);
+		}
+
+		public void turnLight(String roomName, boolean on) {
+			arguments.put("roomName", roomName);
+			arguments.put("on", on);
+			logger.info("Turn light in room: {} to: {}", roomName, on);
+		}
+
+		public void turnLivingRoomLightOn() {
+			arguments.put("turnLivingRoomLightOn", true);
+		}
+
+	}
+
 }