yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-11-11 00:11:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes #6526 - Reset the Fernet Mock in SecretManager in tests (#6527)

Browse Source
This commit is contained in:
Suresh Srinivas 2022-08-02 17:35:06 -07:00 committed by GitHub
parent 9760da6d2f
commit 4cc6f09062
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
catalog-rest-service/src/test/java/org/openmetadata/catalog/secrets/LocalSecretsManagerTest.java
View File

@ -16,13 +16,14 @@ import static org.openmetadata.catalog.services.connections.metadata.OpenMetadat
import static org.openmetadata.catalog.services.connections.metadata.OpenMetadataServerConnection.AuthProvider.OPENMETADATA; import static org.openmetadata.catalog.services.connections.metadata.OpenMetadataServerConnection.AuthProvider.OPENMETADATA;
import java.util.stream.Stream; import java.util.stream.Stream;
import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith; import org.junit.jupiter.api.extension.ExtendWith;
import org.junit.jupiter.params.ParameterizedTest; import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments; import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource; import org.junit.jupiter.params.provider.MethodSource;
import org.mockito.Mock; import org.mockito.Mockito;
import org.mockito.junit.jupiter.MockitoExtension; import org.mockito.junit.jupiter.MockitoExtension;
import org.openmetadata.catalog.airflow.AirflowConfiguration; import org.openmetadata.catalog.airflow.AirflowConfiguration;
import org.openmetadata.catalog.airflow.AuthConfiguration; import org.openmetadata.catalog.airflow.AuthConfiguration;
@ -43,19 +44,23 @@ public class LocalSecretsManagerTest {
private static final boolean DECRYPT = false; private static final boolean DECRYPT = false;
private static final String ENCRYPTED_VALUE = "fernet:abcdef"; private static final String ENCRYPTED_VALUE = "fernet:abcdef";
private static final String DECRYPTED_VALUE = "123456"; private static final String DECRYPTED_VALUE = "123456";
private static LocalSecretsManager secretsManager;
@Mock private Fernet fernet; @BeforeAll
static void setUp() {
private LocalSecretsManager secretsManager;
@BeforeEach
void setUp() {
secretsManager = LocalSecretsManager.getInstance(); secretsManager = LocalSecretsManager.getInstance();
Fernet fernet = Mockito.mock(Fernet.class);
lenient().when(fernet.decrypt(anyString())).thenReturn(DECRYPTED_VALUE); lenient().when(fernet.decrypt(anyString())).thenReturn(DECRYPTED_VALUE);
lenient().when(fernet.encrypt(anyString())).thenReturn(ENCRYPTED_VALUE); lenient().when(fernet.encrypt(anyString())).thenReturn(ENCRYPTED_VALUE);
secretsManager.setFernet(fernet); secretsManager.setFernet(fernet);
} }
@AfterAll
static void teardown() {
// At the end of the test, remove mocked fernet instance so other tests run fine
secretsManager.setFernet(Fernet.getInstance());
}
@Test @Test
void testIsLocalSecretsManager() { void testIsLocalSecretsManager() {
assertTrue(secretsManager.isLocal()); assertTrue(secretsManager.isLocal());