2020package io .bootique .jdbc .junit5 ;
2121
2222import io .bootique .jdbc .junit5 .connector .DbConnector ;
23- import io .bootique .jdbc .junit5 .sql .ExecStatementBuilder ;
24- import io .bootique .jdbc .junit5 .sql .InsertBuilder ;
2523import io .bootique .jdbc .junit5 .dataset .TableDataSet ;
2624import io .bootique .jdbc .junit5 .matcher .TableMatcher ;
2725import io .bootique .jdbc .junit5 .metadata .DbColumnMetadata ;
26+ import io .bootique .jdbc .junit5 .metadata .DbMetadata ;
2827import io .bootique .jdbc .junit5 .metadata .DbTableMetadata ;
2928import io .bootique .jdbc .junit5 .metadata .TableFQName ;
29+ import io .bootique .jdbc .junit5 .metadata .flavors .DbFlavor ;
30+ import io .bootique .jdbc .junit5 .sql .InsertBuilder ;
3031import org .junit .jupiter .api .Test ;
3132
33+ import javax .sql .DataSource ;
3234import java .util .Arrays ;
3335import java .util .List ;
3436import java .util .stream .Collectors ;
3537
3638import static java .util .Arrays .asList ;
3739import static org .junit .jupiter .api .Assertions .*;
38- import static org .mockito .Mockito .mock ;
39- import static org .mockito .Mockito .when ;
4040
4141public class TableTest {
4242
4343 private Table createTable () {
4444
45- ExecStatementBuilder mockExecBuilder = mock (ExecStatementBuilder .class );
46-
47- DbConnector mockConnector = mock (DbConnector .class );
48- when (mockConnector .execStatement ()).thenReturn (mockExecBuilder );
45+ DataSource dataSource = new TestDataSource ();
46+ DbConnector connector = new DbConnector (dataSource , DbMetadata .create (dataSource , new TestDbFlavor ()));
4947
5048 DbColumnMetadata [] columns = new DbColumnMetadata []{
5149 new DbColumnMetadata ("a" , DbColumnMetadata .NO_TYPE , false , true ),
@@ -54,7 +52,7 @@ private Table createTable() {
5452 };
5553
5654 DbTableMetadata tableMetadata = new DbTableMetadata (new TableFQName (null , null , "t" ), columns );
57- return new Table (mockConnector , tableMetadata );
55+ return new Table (connector , tableMetadata );
5856 }
5957
6058 @ Test
@@ -90,4 +88,31 @@ public void csvDataSet() {
9088 assertEquals (0 , ds .size ());
9189 }
9290
91+ static class TestDbFlavor implements DbFlavor {
92+
93+ @ Override
94+ public String getIdentifierQuote () {
95+ return "" ;
96+ }
97+
98+ @ Override
99+ public boolean supportsParamsMetadata () {
100+ return false ;
101+ }
102+
103+ @ Override
104+ public boolean supportsBatchUpdates () {
105+ return false ;
106+ }
107+
108+ @ Override
109+ public boolean supportsCatalogs () {
110+ return false ;
111+ }
112+
113+ @ Override
114+ public boolean supportsSchemas () {
115+ return false ;
116+ }
117+ }
93118}
0 commit comments