|
16 | 16 |
|
17 | 17 | package com.alibaba.fluss.security.auth.sasl.gssapi; |
18 | 18 |
|
| 19 | +import static org.assertj.core.api.Assertions.assertThat; |
| 20 | +import static org.mockito.ArgumentMatchers.any; |
| 21 | +import static org.mockito.ArgumentMatchers.eq; |
| 22 | +import static org.mockito.Mockito.mock; |
| 23 | +import static org.mockito.Mockito.mockStatic; |
| 24 | +import static org.mockito.Mockito.when; |
| 25 | + |
19 | 26 | import com.alibaba.fluss.security.auth.sasl.jaas.LoginManager; |
20 | 27 | import com.alibaba.fluss.security.auth.sasl.jaas.SaslServerFactory; |
21 | 28 |
|
22 | 29 | import org.junit.jupiter.api.Test; |
23 | 30 | import org.mockito.MockedStatic; |
24 | 31 |
|
25 | | -import javax.security.auth.Subject; |
26 | | -import javax.security.auth.login.AppConfigurationEntry; |
27 | | -import javax.security.sasl.Sasl; |
28 | | -import javax.security.sasl.SaslServer; |
29 | | - |
30 | 32 | import java.security.PrivilegedExceptionAction; |
31 | 33 | import java.util.Collections; |
32 | 34 | import java.util.List; |
33 | 35 | import java.util.Map; |
34 | 36 |
|
35 | | -import static org.assertj.core.api.Assertions.assertThat; |
36 | | -import static org.mockito.ArgumentMatchers.any; |
37 | | -import static org.mockito.ArgumentMatchers.eq; |
38 | | -import static org.mockito.Mockito.mock; |
39 | | -import static org.mockito.Mockito.mockStatic; |
40 | | -import static org.mockito.Mockito.when; |
| 37 | +import javax.security.auth.Subject; |
| 38 | +import javax.security.auth.login.AppConfigurationEntry; |
| 39 | +import javax.security.sasl.Sasl; |
| 40 | +import javax.security.sasl.SaslServer; |
41 | 41 |
|
42 | 42 | /** |
43 | 43 | * Unit tests for {@link SaslServerFactory} to verify it can create a SASL server for the mechanism |
|
48 | 48 | * a privileged context. |
49 | 49 | */ |
50 | 50 | public class GssapiSaslServerTest { |
| 51 | + String mechanism = "GSSAPI"; |
| 52 | + String service = "fluss"; |
| 53 | + String host = "localhost"; |
51 | 54 |
|
52 | 55 | @Test |
53 | 56 | public void testCreateSaslServerForGssapi() throws Exception { |
54 | 57 |
|
55 | | - // Mock the dependencies: LoginManager, SaslServer, and Subject |
| 58 | + // Mock the dependencies: LoginManager, SaslServer |
56 | 59 | LoginManager mockLoginManager = mock(LoginManager.class); |
57 | 60 | SaslServer mockSaslServer = mock(SaslServer.class); |
58 | 61 |
|
@@ -82,17 +85,13 @@ public void testCreateSaslServerForGssapi() throws Exception { |
82 | 85 | saslMock.when( |
83 | 86 | () -> |
84 | 87 | Sasl.createSaslServer( |
85 | | - eq("GSSAPI"), |
86 | | - eq("fluss"), |
87 | | - eq("localhost"), |
88 | | - eq(props), |
89 | | - any())) |
| 88 | + eq(mechanism), eq(service), eq(host), eq(props), any())) |
90 | 89 | .thenReturn(mockSaslServer); |
91 | 90 |
|
92 | 91 | // Call SaslServerFactory to create a SASL server for "GSSAPI" and verify the result |
93 | 92 | SaslServer saslServer = |
94 | 93 | SaslServerFactory.createSaslServer( |
95 | | - "GSSAPI", "localhost", props, mockLoginManager, jaasConfig); |
| 94 | + mechanism, host, props, mockLoginManager, jaasConfig); |
96 | 95 |
|
97 | 96 | assertThat(saslServer).isNotNull(); |
98 | 97 | assertThat(saslServer).isEqualTo(mockSaslServer); |
|
0 commit comments