1313import static org .junit .jupiter .api .Assertions .fail ;
1414import static org .mockito .Mockito .mock ;
1515import static org .mockito .Mockito .when ;
16+
17+ import java .sql .Connection ;
18+ import java .sql .DatabaseMetaData ;
19+
1620import org .eclipse .daanse .jdbc .db .dialect .db .postgresql .PostgreSqlDialect ;
1721import org .eclipse .daanse .olap .api .exception .OlapRuntimeException ;
18- import org .eclipse .daanse .jdbc .db .api .meta .DatabaseInfo ;
19- import org .eclipse .daanse .jdbc .db .api .meta .IdentifierInfo ;
20- import org .eclipse .daanse .jdbc .db .api .meta .MetaInfo ;
2122import org .junit .jupiter .api .Test ;
2223
2324/**
@@ -55,7 +56,7 @@ void testSucces_CodeSetContainsOnlyCodeForPostgresDialect()
5556 throws Exception
5657 {
5758 PostgreSqlDialect postgreSqlDialect = new PostgreSqlDialect (
58- mockMetaInfo (
59+ mockConnection (
5960 POSTGRESQL_PRODUCT_NAME ,
6061 POSTGRESQL_PRODUCT_VERSION ));
6162 codeSet = new SqlQuery .CodeSet ();
@@ -82,7 +83,7 @@ void testSucces_CodeSetContainsCodeForBothPostgresAndGenericDialects()
8283 throws Exception
8384 {
8485 PostgreSqlDialect postgreSqlDialect = new PostgreSqlDialect (
85- mockMetaInfo (
86+ mockConnection (
8687 POSTGRESQL_PRODUCT_NAME ,
8788 POSTGRESQL_PRODUCT_VERSION ));
8889 codeSet = new SqlQuery .CodeSet ();
@@ -111,7 +112,7 @@ void testSucces_CodeSetContainsCodeForBothPostgresAndGenericDialects()
111112 throws Exception
112113 {
113114 PostgreSqlDialect postgreSqlDialect = new PostgreSqlDialect (
114- mockMetaInfo (
115+ mockConnection (
115116 POSTGRESQL_PRODUCT_NAME ,
116117 POSTGRESQL_PRODUCT_VERSION ));
117118 codeSet = new SqlQuery .CodeSet ();
@@ -138,7 +139,7 @@ void testSucces_CodeSetContainsOnlyCodeForGenericlDialect()
138139 throws Exception
139140 {
140141 PostgreSqlDialect postgreSqlDialect = new PostgreSqlDialect (
141- mockMetaInfo (
142+ mockConnection (
142143 POSTGRESQL_PRODUCT_NAME ,
143144 POSTGRESQL_PRODUCT_VERSION ));
144145 codeSet = new SqlQuery .CodeSet ();
@@ -164,7 +165,7 @@ void testMondrianExceptionThrown_WhenCodeSetContainsNOCodeForDialect()
164165 throws Exception
165166 {
166167 PostgreSqlDialect postgreSqlDialect = new PostgreSqlDialect (
167- mockMetaInfo (
168+ mockConnection (
168169 POSTGRESQL_PRODUCT_NAME ,
169170 POSTGRESQL_PRODUCT_VERSION ));
170171 codeSet = new SqlQuery .CodeSet ();
@@ -180,19 +181,17 @@ void testMondrianExceptionThrown_WhenCodeSetContainsNOCodeForDialect()
180181 }
181182 }
182183
183- private MetaInfo mockMetaInfo (
184+ private Connection mockConnection (
184185 String dbProductName , String dbProductVersion ) throws Exception
185186 {
186- MetaInfo metaInfoMock = mock (MetaInfo .class );
187- DatabaseInfo databaseInfo = mock (DatabaseInfo .class );
188- IdentifierInfo identifierInfo = mock (IdentifierInfo .class );
189- when (metaInfoMock .databaseInfo ()).thenReturn (databaseInfo );
190- when (metaInfoMock .identifierInfo ()).thenReturn (identifierInfo );
191- when (databaseInfo .databaseProductName ()).thenReturn (
187+ Connection connection = mock (Connection .class );
188+ DatabaseMetaData metaData = mock (DatabaseMetaData .class );
189+ when (connection .getMetaData ()).thenReturn (metaData );
190+ when (metaData .getDatabaseProductName ()).thenReturn (
192191 dbProductName != null ? dbProductName : EMPTY_NAME );
193- when (databaseInfo . databaseProductVersion ()).thenReturn (
192+ when (metaData . getDatabaseProductVersion ()).thenReturn (
194193 dbProductVersion != null ? dbProductVersion : EMPTY_NAME );
195- return metaInfoMock ;
194+ return connection ;
196195 }
197196
198197}
0 commit comments