Skip to content

Commit 7a79439

Browse files
author
nickchecan
committed
refactor: adapt test from ai client factory for ollama
1 parent 7254e8b commit 7a79439

File tree

3 files changed

+138
-1
lines changed

3 files changed

+138
-1
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,67 @@
1+
package com.developer.nefarious.zjoule.test.chat;
2+
3+
import static org.junit.jupiter.api.Assertions.assertTrue;
4+
import static org.mockito.Mockito.mockStatic;
5+
6+
import org.junit.jupiter.api.AfterEach;
7+
import org.junit.jupiter.api.BeforeEach;
8+
import org.junit.jupiter.api.Test;
9+
import org.mockito.Mock;
10+
import org.mockito.MockedStatic;
11+
import org.mockito.MockitoAnnotations;
12+
13+
import com.developer.nefarious.zjoule.plugin.auth.SessionManager;
14+
import com.developer.nefarious.zjoule.plugin.memory.MemoryOllamaEndpoint;
15+
import com.developer.nefarious.zjoule.plugin.memory.MemoryOllamaModel;
16+
17+
public class AiClientFactoryOllamaTest {
18+
19+
private MockedStatic<SessionManager> mockStaticSessionManager;
20+
21+
private MockedStatic<MemoryOllamaEndpoint> mockStaticMemoryOllamaEndpoint;
22+
23+
private MockedStatic<MemoryOllamaModel> mockStaticMemoryOllamaModel;
24+
25+
@Mock
26+
private MemoryOllamaEndpoint memoryOllamaEndpoint;
27+
28+
@Mock
29+
private MemoryOllamaModel memoryOllamaModel;
30+
31+
@BeforeEach
32+
public void setUp() {
33+
MockitoAnnotations.openMocks(this);
34+
35+
mockStaticSessionManager = mockStatic(SessionManager.class);
36+
mockStaticMemoryOllamaEndpoint = mockStatic(MemoryOllamaEndpoint.class);
37+
mockStaticMemoryOllamaModel = mockStatic(MemoryOllamaModel.class);
38+
39+
mockStaticMemoryOllamaEndpoint.when(MemoryOllamaEndpoint::getInstance).thenReturn(memoryOllamaEndpoint);
40+
mockStaticMemoryOllamaModel.when(MemoryOllamaModel::getInstance).thenReturn(memoryOllamaModel);
41+
42+
mockStaticSessionManager.when(SessionManager::isOllamaSession).thenReturn(true);
43+
mockStaticSessionManager.when(SessionManager::isSapSession).thenReturn(false);
44+
}
45+
46+
@AfterEach
47+
public void tearDown() {
48+
if (mockStaticSessionManager != null) {
49+
mockStaticSessionManager.close();
50+
}
51+
if (mockStaticMemoryOllamaEndpoint != null) {
52+
mockStaticMemoryOllamaEndpoint.close();
53+
}
54+
if (mockStaticMemoryOllamaModel != null) {
55+
mockStaticMemoryOllamaModel.close();
56+
}
57+
}
58+
59+
@Test
60+
public void shouldReturnNullForOthers() {
61+
// Arrange
62+
// Act
63+
// Assert
64+
assertTrue(true);
65+
}
66+
67+
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,70 @@
1+
package com.developer.nefarious.zjoule.test.chat;
2+
3+
import static org.junit.jupiter.api.Assertions.assertNull;
4+
import static org.mockito.Mockito.mockStatic;
5+
6+
import org.junit.jupiter.api.AfterEach;
7+
import org.junit.jupiter.api.BeforeEach;
8+
import org.junit.jupiter.api.Test;
9+
import org.mockito.Mock;
10+
import org.mockito.MockedStatic;
11+
import org.mockito.MockitoAnnotations;
12+
13+
import com.developer.nefarious.zjoule.plugin.auth.SessionManager;
14+
import com.developer.nefarious.zjoule.plugin.chat.AIClientFactory;
15+
import com.developer.nefarious.zjoule.plugin.chat.IAIClient;
16+
import com.developer.nefarious.zjoule.plugin.memory.MemoryOllamaEndpoint;
17+
import com.developer.nefarious.zjoule.plugin.memory.MemoryOllamaModel;
18+
19+
public class AiClientFactoryOthersTest {
20+
21+
private MockedStatic<SessionManager> mockStaticSessionManager;
22+
23+
private MockedStatic<MemoryOllamaEndpoint> mockStaticMemoryOllamaEndpoint;
24+
25+
private MockedStatic<MemoryOllamaModel> mockStaticMemoryOllamaModel;
26+
27+
@Mock
28+
private MemoryOllamaEndpoint memoryOllamaEndpoint;
29+
30+
@Mock
31+
private MemoryOllamaModel memoryOllamaModel;
32+
33+
@BeforeEach
34+
public void setUp() {
35+
MockitoAnnotations.openMocks(this);
36+
37+
mockStaticSessionManager = mockStatic(SessionManager.class);
38+
mockStaticMemoryOllamaEndpoint = mockStatic(MemoryOllamaEndpoint.class);
39+
mockStaticMemoryOllamaModel = mockStatic(MemoryOllamaModel.class);
40+
41+
mockStaticMemoryOllamaEndpoint.when(MemoryOllamaEndpoint::getInstance).thenReturn(memoryOllamaEndpoint);
42+
mockStaticMemoryOllamaModel.when(MemoryOllamaModel::getInstance).thenReturn(memoryOllamaModel);
43+
44+
mockStaticSessionManager.when(SessionManager::isOllamaSession).thenReturn(false);
45+
mockStaticSessionManager.when(SessionManager::isSapSession).thenReturn(false);
46+
}
47+
48+
@AfterEach
49+
public void tearDown() {
50+
if (mockStaticSessionManager != null) {
51+
mockStaticSessionManager.close();
52+
}
53+
if (mockStaticMemoryOllamaEndpoint != null) {
54+
mockStaticMemoryOllamaEndpoint.close();
55+
}
56+
if (mockStaticMemoryOllamaModel != null) {
57+
mockStaticMemoryOllamaModel.close();
58+
}
59+
}
60+
61+
@Test
62+
public void shouldReturnNullForOthers() {
63+
// Arrange
64+
// Act
65+
IAIClient returnValue = AIClientFactory.getClient();
66+
// Assert
67+
assertNull(returnValue);
68+
}
69+
70+
}
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
import com.developer.nefarious.zjoule.plugin.memory.MemoryServiceKey;
2424
import com.developer.nefarious.zjoule.plugin.models.Deployment;
2525

26-
public class AIClientFactorySapTest {
26+
public class AiClientFactorySapTest {
2727

2828
private MockedStatic<SessionManager> mockStaticSessionManager;
2929

0 commit comments

Comments
 (0)