-
Notifications
You must be signed in to change notification settings - Fork 379
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
xiaojiebao
committed
Feb 6, 2024
1 parent
bc7d1e2
commit fde3c14
Showing
7 changed files
with
204 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
102 changes: 102 additions & 0 deletions
102
...om/datastrato/gravitino/storage/relational/mysql/session/TestSqlSessionFactoryHelper.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,102 @@ | ||
| /* | ||
| * Copyright 2024 Datastrato Pvt Ltd. | ||
| * This software is licensed under the Apache License version 2. | ||
| */ | ||
|
|
||
| package com.datastrato.gravitino.storage.relational.mysql.session; | ||
|
|
||
| import static com.datastrato.gravitino.Configs.DEFAULT_ENTITY_RELATIONAL_STORE; | ||
| import static com.datastrato.gravitino.Configs.ENTITY_RELATIONAL_STORE; | ||
| import static com.datastrato.gravitino.Configs.ENTITY_STORE; | ||
| import static com.datastrato.gravitino.Configs.MYSQL_ENTITY_STORE_DRIVER_NAME_KEY; | ||
| import static com.datastrato.gravitino.Configs.MYSQL_ENTITY_STORE_URL_KEY; | ||
| import static com.datastrato.gravitino.Configs.MYSQL_ENTITY_STORE_USERNAME_KEY; | ||
| import static com.datastrato.gravitino.Configs.RELATIONAL_ENTITY_STORE; | ||
| import static org.junit.jupiter.api.Assertions.assertEquals; | ||
| import static org.junit.jupiter.api.Assertions.assertNotNull; | ||
| import static org.junit.jupiter.api.Assertions.assertThrows; | ||
|
|
||
| import com.datastrato.gravitino.Config; | ||
| import java.io.File; | ||
| import java.io.IOException; | ||
| import java.sql.SQLException; | ||
| import java.util.UUID; | ||
| import org.apache.commons.dbcp2.BasicDataSource; | ||
| import org.junit.jupiter.api.AfterAll; | ||
| import org.junit.jupiter.api.AfterEach; | ||
| import org.junit.jupiter.api.BeforeAll; | ||
| import org.junit.jupiter.api.BeforeEach; | ||
| import org.junit.jupiter.api.Test; | ||
| import org.mockito.Mockito; | ||
|
|
||
| public class TestSqlSessionFactoryHelper { | ||
| private static final String MYSQL_STORE_PATH = | ||
| "/tmp/gravitino_test_entityStore_" + UUID.randomUUID().toString().replace("-", ""); | ||
| private static final String DB_DIR = MYSQL_STORE_PATH + "/testdb"; | ||
|
|
||
| private static Config config; | ||
|
|
||
| @BeforeAll | ||
| public static void setUp() { | ||
| File dir = new File(DB_DIR); | ||
| if (dir.exists() || !dir.isDirectory()) { | ||
| dir.delete(); | ||
| } | ||
| dir.mkdirs(); | ||
|
|
||
| config = Mockito.mock(Config.class); | ||
| Mockito.when(config.get(ENTITY_STORE)).thenReturn(RELATIONAL_ENTITY_STORE); | ||
| Mockito.when(config.get(ENTITY_RELATIONAL_STORE)).thenReturn(DEFAULT_ENTITY_RELATIONAL_STORE); | ||
| Mockito.when(config.getRawString(MYSQL_ENTITY_STORE_URL_KEY)) | ||
| .thenReturn(String.format("jdbc:h2:%s;DB_CLOSE_DELAY=-1;MODE=MYSQL", DB_DIR)); | ||
| Mockito.when(config.getRawString(MYSQL_ENTITY_STORE_USERNAME_KEY)).thenReturn("sa"); | ||
| Mockito.when(config.getRawString(MYSQL_ENTITY_STORE_DRIVER_NAME_KEY)) | ||
| .thenReturn("org.h2.Driver"); | ||
| } | ||
|
|
||
| @BeforeEach | ||
| public void init() { | ||
| SqlSessionFactoryHelper.setSqlSessionFactory(null); | ||
| } | ||
|
|
||
| @AfterEach | ||
| public void cleanUp() { | ||
| SqlSessionFactoryHelper.setSqlSessionFactory(null); | ||
| } | ||
|
|
||
| @AfterAll | ||
| public static void tearDown() throws IOException { | ||
| File dir = new File(DB_DIR); | ||
| if (dir.exists()) { | ||
| dir.delete(); | ||
| } | ||
| } | ||
|
|
||
| @Test | ||
| public void testGetInstance() { | ||
| SqlSessionFactoryHelper instance = SqlSessionFactoryHelper.getInstance(); | ||
| assertNotNull(instance); | ||
| } | ||
|
|
||
| @Test | ||
| public void testInit() throws SQLException { | ||
| SqlSessionFactoryHelper.getInstance().init(config); | ||
| assertNotNull(SqlSessionFactoryHelper.getInstance().getSqlSessionFactory()); | ||
| BasicDataSource dataSource = | ||
| (BasicDataSource) | ||
| SqlSessionFactoryHelper.getInstance() | ||
| .getSqlSessionFactory() | ||
| .getConfiguration() | ||
| .getEnvironment() | ||
| .getDataSource(); | ||
| assertEquals("org.h2.Driver", dataSource.getDriverClassName()); | ||
| assertEquals(config.getRawString(MYSQL_ENTITY_STORE_URL_KEY), dataSource.getUrl()); | ||
| } | ||
|
|
||
| @Test | ||
| public void testGetSqlSessionFactoryWithoutInit() { | ||
| assertThrows( | ||
| IllegalStateException.class, | ||
| () -> SqlSessionFactoryHelper.getInstance().getSqlSessionFactory()); | ||
| } | ||
| } |
83 changes: 83 additions & 0 deletions
83
.../test/java/com/datastrato/gravitino/storage/relational/mysql/session/TestSqlSessions.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,83 @@ | ||
| /* | ||
| * Copyright 2024 Datastrato Pvt Ltd. | ||
| * This software is licensed under the Apache License version 2. | ||
| */ | ||
|
|
||
| package com.datastrato.gravitino.storage.relational.mysql.session; | ||
|
|
||
| import static com.datastrato.gravitino.Configs.DEFAULT_ENTITY_RELATIONAL_STORE; | ||
| import static com.datastrato.gravitino.Configs.ENTITY_RELATIONAL_STORE; | ||
| import static com.datastrato.gravitino.Configs.ENTITY_STORE; | ||
| import static com.datastrato.gravitino.Configs.MYSQL_ENTITY_STORE_DRIVER_NAME_KEY; | ||
| import static com.datastrato.gravitino.Configs.MYSQL_ENTITY_STORE_URL_KEY; | ||
| import static com.datastrato.gravitino.Configs.MYSQL_ENTITY_STORE_USERNAME_KEY; | ||
| import static com.datastrato.gravitino.Configs.RELATIONAL_ENTITY_STORE; | ||
| import static org.junit.jupiter.api.Assertions.assertNotNull; | ||
| import static org.junit.jupiter.api.Assertions.assertNull; | ||
|
|
||
| import com.datastrato.gravitino.Config; | ||
| import java.io.File; | ||
| import java.io.IOException; | ||
| import java.util.UUID; | ||
| import org.apache.ibatis.session.SqlSession; | ||
| import org.junit.jupiter.api.AfterAll; | ||
| import org.junit.jupiter.api.BeforeAll; | ||
| import org.junit.jupiter.api.Test; | ||
| import org.mockito.Mockito; | ||
|
|
||
| public class TestSqlSessions { | ||
| private static final String MYSQL_STORE_PATH = | ||
| "/tmp/gravitino_test_entityStore_" + UUID.randomUUID().toString().replace("-", ""); | ||
| private static final String DB_DIR = MYSQL_STORE_PATH + "/testdb"; | ||
|
|
||
| @BeforeAll | ||
| public static void setUp() { | ||
| File dir = new File(DB_DIR); | ||
| if (dir.exists() || !dir.isDirectory()) { | ||
| dir.delete(); | ||
| } | ||
| dir.mkdirs(); | ||
|
|
||
| Config config = Mockito.mock(Config.class); | ||
| Mockito.when(config.get(ENTITY_STORE)).thenReturn(RELATIONAL_ENTITY_STORE); | ||
| Mockito.when(config.get(ENTITY_RELATIONAL_STORE)).thenReturn(DEFAULT_ENTITY_RELATIONAL_STORE); | ||
| Mockito.when(config.getRawString(MYSQL_ENTITY_STORE_URL_KEY)) | ||
| .thenReturn(String.format("jdbc:h2:%s;DB_CLOSE_DELAY=-1;MODE=MYSQL", DB_DIR)); | ||
| Mockito.when(config.getRawString(MYSQL_ENTITY_STORE_USERNAME_KEY)).thenReturn("sa"); | ||
| Mockito.when(config.getRawString(MYSQL_ENTITY_STORE_DRIVER_NAME_KEY)) | ||
| .thenReturn("org.h2.Driver"); | ||
| SqlSessionFactoryHelper.getInstance().init(config); | ||
| } | ||
|
|
||
| @AfterAll | ||
| public static void tearDown() throws IOException { | ||
| File dir = new File(DB_DIR); | ||
| if (dir.exists()) { | ||
| dir.delete(); | ||
| } | ||
| } | ||
|
|
||
| @Test | ||
| public void testOpenAndCloseSqlSession() { | ||
| SqlSession session = SqlSessions.getSqlSession(); | ||
| assertNotNull(session); | ||
| SqlSessions.closeSqlSession(); | ||
| assertNull(SqlSessions.getSessions().get()); | ||
| } | ||
|
|
||
| @Test | ||
| public void testOpenAndCommitAndCloseSqlSession() { | ||
| SqlSession session = SqlSessions.getSqlSession(); | ||
| assertNotNull(session); | ||
| SqlSessions.commitAndCloseSqlSession(); | ||
| assertNull(SqlSessions.getSessions().get()); | ||
| } | ||
|
|
||
| @Test | ||
| public void testOpenAndRollbackAndCloseSqlSession() { | ||
| SqlSession session = SqlSessions.getSqlSession(); | ||
| assertNotNull(session); | ||
| SqlSessions.rollbackAndCloseSqlSession(); | ||
| assertNull(SqlSessions.getSessions().get()); | ||
| } | ||
| } |