iSharkFly-Docs
/
java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

BAEL-7142, fix the tests to not call external client

This commit is contained in:
Ekatereana 2023-11-27 17:05:05 +02:00
parent c8dafdf0e4
commit e1116c86bd
3 changed files with 45 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
spring-ai/src/test/java/com/baeldung/spring/ai/web/ExceptionTranslatorIntegrationTest.java
View File

@ -1,45 +0,0 @@
package com.baeldung.spring.ai.web;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.ai.client.AiClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.DynamicPropertyRegistry;
import org.springframework.test.context.DynamicPropertySource;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import static java.lang.String.format;
import static org.hamcrest.Matchers.containsStringIgnoringCase;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
@AutoConfigureMockMvc
@RunWith(SpringRunner.class)
@SpringBootTest
public class ExceptionTranslatorIntegrationTest {
@Autowired
private MockMvc mockMvc;
@Autowired
private ObjectMapper objectMapper;
@Autowired
private AiClient aiClient;
@DynamicPropertySource
static void setupTestcontainerProperties(DynamicPropertyRegistry registry) {
registry.add("spring.ai.openai.api-key", () -> "incorrect_token");
}
@Test
public void givenGetCatHaiku_whenCallingAiClientWithIncorrectToken_thenUnauthorized() throws Exception {
mockMvc.perform(get("/ai/cathaiku"))
.andExpect(status().isUnauthorized())
.andExpect(jsonPath("$.title").value(ExceptionTranslator.OPEN_AI_CLIENT_RAISED_EXCEPTION));
}
}

50
spring-ai/src/test/java/com/baeldung/spring/ai/web/PoetryControllerIntegrationTest.java
View File

@ -1,16 +1,29 @@
package com.baeldung.spring.ai.web;
import com.baeldung.spring.ai.dto.PoetryDto;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.theokanning.openai.OpenAiError;
import com.theokanning.openai.OpenAiHttpException;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.ai.client.AiClient;
import org.springframework.ai.client.AiResponse;
import org.springframework.ai.client.Generation;
import org.springframework.ai.prompt.Prompt;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.boot.test.mock.mockito.SpyBean;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import java.util.List;
import java.util.Optional;
import static org.hamcrest.Matchers.*;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.when;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
@ -19,31 +32,56 @@ import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.
@SpringBootTest
public class PoetryControllerIntegrationTest {
public static final String GENERATED_HAIKU = "generated haiku";
public static final String TEST_TITLE = "test_title";
public static final String SOME_POETRY = "some poetry";
public static final String TEST_GENRE = "test_genre";
public static final String TEST_THEME = "test_theme";
@Autowired
private MockMvc mockMvc;
@Autowired
private ObjectMapper objectMapper;
@Autowired
@MockBean
private AiClient aiClient;
@Test
public void givenGetCatHaiku_whenCallingAiClient_thenCorrect() throws Exception {
when(aiClient.generate(anyString())).thenReturn(GENERATED_HAIKU);
mockMvc.perform(get("/ai/cathaiku"))
.andExpect(status().isOk())
.andExpect(content().string(containsStringIgnoringCase("cat")));
.andExpect(content().string(GENERATED_HAIKU));
}
@Test
public void givenGetPoetryWithGenreAndTheme_whenCallingAiClient_thenCorrect() throws Exception {
PoetryDto poetryDto = new PoetryDto(
TEST_TITLE,
SOME_POETRY,
TEST_GENRE,
TEST_THEME
);
when(aiClient.generate((Prompt) any(Prompt.class))).thenReturn(
new AiResponse(List.of(new Generation(objectMapper.writeValueAsString(poetryDto)))));
String genre = "lyric";
String theme = "coffee";
mockMvc.perform(get("/ai/poetry?genre={genre}&theme={theme}", genre, theme))
.andExpect(status().isOk())
.andExpect(jsonPath("$.genre").value(containsStringIgnoringCase(genre)))
.andExpect(jsonPath("$.theme").value(containsStringIgnoringCase(theme)))
.andExpect(jsonPath("$.poetry").isNotEmpty())
.andExpect(jsonPath("$.title").exists());
.andExpect(jsonPath("$.genre").value(poetryDto.genre()))
.andExpect(jsonPath("$.theme").value(poetryDto.theme()))
.andExpect(jsonPath("$.poetry").value(poetryDto.poetry()))
.andExpect(jsonPath("$.title").value(poetryDto.title()));
}
@Test
public void givenGetCatHaiku_whenCallingAiClientWithIncorrectToken_thenUnauthorized() throws Exception {
when(aiClient.generate(anyString())).thenThrow(new OpenAiHttpException(
new OpenAiError(new OpenAiError.OpenAiErrorDetails()),
new Exception(),
401));
mockMvc.perform(get("/ai/cathaiku"))
.andExpect(status().isUnauthorized())
.andExpect(jsonPath("$.title").value(ExceptionTranslator.OPEN_AI_CLIENT_RAISED_EXCEPTION));
}
}

2
spring-ai/src/test/resources/application.yml
View File

@ -1,3 +1,3 @@
spring:
ai:
openai.api-key: ${OPEN_AI_KEY}
openai.api-key: "mock-key-token"