|  | 
|  | 1 | +package g3601_3700.s3601_find_drivers_with_improved_fuel_efficiency | 
|  | 2 | + | 
|  | 3 | +import org.hamcrest.CoreMatchers | 
|  | 4 | +import org.hamcrest.MatcherAssert | 
|  | 5 | +import org.junit.jupiter.api.Test | 
|  | 6 | +import org.zapodot.junit.db.annotations.EmbeddedDatabase | 
|  | 7 | +import org.zapodot.junit.db.annotations.EmbeddedDatabaseTest | 
|  | 8 | +import org.zapodot.junit.db.common.CompatibilityMode | 
|  | 9 | +import java.io.BufferedReader | 
|  | 10 | +import java.io.FileNotFoundException | 
|  | 11 | +import java.io.FileReader | 
|  | 12 | +import java.sql.SQLException | 
|  | 13 | +import java.util.stream.Collectors | 
|  | 14 | +import javax.sql.DataSource | 
|  | 15 | + | 
|  | 16 | +@EmbeddedDatabaseTest( | 
|  | 17 | +    compatibilityMode = CompatibilityMode.MySQL, | 
|  | 18 | +    initialSqls = [ | 
|  | 19 | +        ( | 
|  | 20 | +            "CREATE TABLE drivers (driver_id INTEGER, driver_name VARCHAR(255)); " + | 
|  | 21 | +                "INSERT INTO drivers (driver_id, driver_name) VALUES" + | 
|  | 22 | +                "(1, 'Alice Johnson')," + | 
|  | 23 | +                "(2, 'Bob Smith')," + | 
|  | 24 | +                "(3, 'Carol Davis')," + | 
|  | 25 | +                "(4, 'David Wilson')," + | 
|  | 26 | +                "(5, 'Emma Brown');" + | 
|  | 27 | +                "CREATE TABLE trips (trip_id INTEGER, driver_id INTEGER" + | 
|  | 28 | +                ", trip_date DATE, distance_km DECIMAL(7, 3), fuel_consumed DECIMAL(7, 3)); " + | 
|  | 29 | +                "INSERT INTO trips (trip_id, driver_id, trip_date, distance_km, fuel_consumed) VALUES" + | 
|  | 30 | +                "(1, 1, '2023-02-15', 120.5, 10.2)," + | 
|  | 31 | +                "(2, 1, '2023-03-20', 200.0, 16.5)," + | 
|  | 32 | +                "(3, 1, '2023-08-10', 150.0, 11.0)," + | 
|  | 33 | +                "(4, 1, '2023-09-25', 180.0, 12.5)," + | 
|  | 34 | +                "(5, 2, '2023-01-10', 100.0, 9.0)," + | 
|  | 35 | +                "(6, 2, '2023-04-15', 250.0, 22.0)," + | 
|  | 36 | +                "(7, 2, '2023-10-05', 200.0, 15.0)," + | 
|  | 37 | +                "(8, 3, '2023-03-12', 80.0, 8.5)," + | 
|  | 38 | +                "(9, 3, '2023-05-18', 90.0, 9.2)," + | 
|  | 39 | +                "(10, 4, '2023-07-22', 160.0, 12.8)," + | 
|  | 40 | +                "(11, 4, '2023-11-30', 140.0, 11.0)," + | 
|  | 41 | +                "(12, 5, '2023-02-28', 110.0, 11.5);" | 
|  | 42 | +            ), | 
|  | 43 | +    ], | 
|  | 44 | +) | 
|  | 45 | +internal class MysqlTest { | 
|  | 46 | +    @Test | 
|  | 47 | +    @Throws(SQLException::class, FileNotFoundException::class) | 
|  | 48 | +    fun testScript(@EmbeddedDatabase dataSource: DataSource) { | 
|  | 49 | +        dataSource.connection.use { connection -> | 
|  | 50 | +            connection.createStatement().use { statement -> | 
|  | 51 | +                statement.executeQuery( | 
|  | 52 | +                    BufferedReader( | 
|  | 53 | +                        FileReader( | 
|  | 54 | +                            ( | 
|  | 55 | +                                "src/main/kotlin/g3601_3700/" + | 
|  | 56 | +                                    "s3601_find_drivers_with_" + | 
|  | 57 | +                                    "improved_fuel_efficiency/" + | 
|  | 58 | +                                    "script.sql" | 
|  | 59 | +                                ), | 
|  | 60 | +                        ), | 
|  | 61 | +                    ) | 
|  | 62 | +                        .lines() | 
|  | 63 | +                        .collect(Collectors.joining("\n")) | 
|  | 64 | +                        .replace("#.*?\\r?\\n".toRegex(), ""), | 
|  | 65 | +                ).use { resultSet -> | 
|  | 66 | +                    MatcherAssert.assertThat<Boolean>(resultSet.next(), CoreMatchers.equalTo<Boolean>(true)) | 
|  | 67 | +                    MatcherAssert.assertThat<String>(resultSet.getNString(1), CoreMatchers.equalTo<String>("2")) | 
|  | 68 | +                    MatcherAssert.assertThat<String>( | 
|  | 69 | +                        resultSet.getNString(2), | 
|  | 70 | +                        CoreMatchers.equalTo<String>("Bob Smith"), | 
|  | 71 | +                    ) | 
|  | 72 | +                    MatcherAssert.assertThat<String>(resultSet.getNString(3), CoreMatchers.equalTo<String>("11.24")) | 
|  | 73 | +                    MatcherAssert.assertThat<String>(resultSet.getNString(4), CoreMatchers.equalTo<String>("13.33")) | 
|  | 74 | +                    MatcherAssert.assertThat<Boolean>(resultSet.next(), CoreMatchers.equalTo<Boolean>(true)) | 
|  | 75 | +                    MatcherAssert.assertThat<String>(resultSet.getNString(1), CoreMatchers.equalTo<String>("1")) | 
|  | 76 | +                    MatcherAssert.assertThat<String>( | 
|  | 77 | +                        resultSet.getNString(2), | 
|  | 78 | +                        CoreMatchers.equalTo<String>("Alice Johnson"), | 
|  | 79 | +                    ) | 
|  | 80 | +                    MatcherAssert.assertThat<String>(resultSet.getNString(3), CoreMatchers.equalTo<String>("11.97")) | 
|  | 81 | +                    MatcherAssert.assertThat<String>(resultSet.getNString(4), CoreMatchers.equalTo<String>("14.02")) | 
|  | 82 | +                    MatcherAssert.assertThat<Boolean>(resultSet.next(), CoreMatchers.equalTo<Boolean>(false)) | 
|  | 83 | +                } | 
|  | 84 | +            } | 
|  | 85 | +        } | 
|  | 86 | +    } | 
|  | 87 | +} | 
0 commit comments