Added surefire and failsafe plugins

This commit is contained in:
Amith Koujalgi
2023-12-14 17:41:23 +05:30
parent f6af7025a8
commit b2d76970dc
4 changed files with 29 additions and 2 deletions

View File

@@ -101,7 +101,7 @@ public class TestRealAPIs {
String model = OllamaModelType.LLAMA2;
String prompt = "some prompt text";
try {
when(ollamaAPI.ask(model, prompt)).thenReturn(new OllamaResult("", 0));
when(ollamaAPI.ask(model, prompt)).thenReturn(new OllamaResult("", 0,200));
ollamaAPI.ask(model, prompt);
verify(ollamaAPI, times(1)).ask(model, prompt);
} catch (IOException | OllamaBaseException | InterruptedException e) {