|  | 
|  | 1 | +package g3401_3500.s3421_find_students_who_improved | 
|  | 2 | + | 
|  | 3 | +import org.hamcrest.CoreMatchers.equalTo | 
|  | 4 | +import org.hamcrest.MatcherAssert.assertThat | 
|  | 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 Scores (" + | 
|  | 21 | +                "    student_id INT," + | 
|  | 22 | +                "    subject VARCHAR(50)," + | 
|  | 23 | +                "    score INT," + | 
|  | 24 | +                "    exam_date VARCHAR(10)" + | 
|  | 25 | +                ");" + | 
|  | 26 | +                "insert into Scores (student_id, subject, score, exam_date) values " + | 
|  | 27 | +                "('101', 'Math', '70', '15-01-2023');" + | 
|  | 28 | +                "insert into Scores (student_id, subject, score, exam_date) values " + | 
|  | 29 | +                "('101', 'Math', '85', '15-02-2023');" + | 
|  | 30 | +                "insert into Scores (student_id, subject, score, exam_date) values " + | 
|  | 31 | +                "('101', 'Physics', '65', '15-01-2023');" + | 
|  | 32 | +                "insert into Scores (student_id, subject, score, exam_date) values " + | 
|  | 33 | +                "('101', 'Physics', '60', '15-02-2023');" + | 
|  | 34 | +                "insert into Scores (student_id, subject, score, exam_date) values " + | 
|  | 35 | +                "('102', 'Math', '80', '15-01-2023');" + | 
|  | 36 | +                "insert into Scores (student_id, subject, score, exam_date) values " + | 
|  | 37 | +                "('102', 'Math', '85', '15-02-2023');" + | 
|  | 38 | +                "insert into Scores (student_id, subject, score, exam_date) values " + | 
|  | 39 | +                "('103', 'Math', '90', '15-01-2023');" + | 
|  | 40 | +                "insert into Scores (student_id, subject, score, exam_date) values " + | 
|  | 41 | +                "('104', 'Physics', '75', '15-01-2023');" + | 
|  | 42 | +                "insert into Scores (student_id, subject, score, exam_date) values " + | 
|  | 43 | +                "('104', 'Physics', '85', '15-02-2023');" | 
|  | 44 | +            ), | 
|  | 45 | +    ], | 
|  | 46 | +) | 
|  | 47 | +internal class MysqlTest { | 
|  | 48 | +    @Test | 
|  | 49 | +    @Throws(SQLException::class, FileNotFoundException::class) | 
|  | 50 | +    fun testScript(@EmbeddedDatabase dataSource: DataSource) { | 
|  | 51 | +        dataSource.connection.use { connection -> | 
|  | 52 | +            connection.createStatement().use { statement -> | 
|  | 53 | +                statement.executeQuery( | 
|  | 54 | +                    BufferedReader( | 
|  | 55 | +                        FileReader( | 
|  | 56 | +                            "src/main/kotlin/g3401_3500/" + | 
|  | 57 | +                                "s3421_find_students_who_improved/script.sql", | 
|  | 58 | +                        ), | 
|  | 59 | +                    ) | 
|  | 60 | +                        .lines() | 
|  | 61 | +                        .collect(Collectors.joining("\n")) | 
|  | 62 | +                        .replace("#.*?\\r?\\n".toRegex(), ""), | 
|  | 63 | +                ).use { resultSet -> | 
|  | 64 | +                    assertThat<Boolean>(resultSet.next(), equalTo<Boolean>(true)) | 
|  | 65 | +                    assertThat<String>(resultSet.getNString(1), equalTo<String>("101")) | 
|  | 66 | +                    assertThat<String>(resultSet.getNString(2), equalTo<String>("Math")) | 
|  | 67 | +                    assertThat<String>(resultSet.getNString(3), equalTo<String>("70")) | 
|  | 68 | +                    assertThat<String>(resultSet.getNString(4), equalTo<String>("85")) | 
|  | 69 | +                    assertThat<Boolean>(resultSet.next(), equalTo<Boolean>(true)) | 
|  | 70 | +                    assertThat<String>(resultSet.getNString(1), equalTo<String>("102")) | 
|  | 71 | +                    assertThat<String>(resultSet.getNString(2), equalTo<String>("Math")) | 
|  | 72 | +                    assertThat<String>(resultSet.getNString(3), equalTo<String>("80")) | 
|  | 73 | +                    assertThat<String>(resultSet.getNString(4), equalTo<String>("85")) | 
|  | 74 | +                    assertThat<Boolean>(resultSet.next(), equalTo<Boolean>(true)) | 
|  | 75 | +                    assertThat<String>(resultSet.getNString(1), equalTo<String>("104")) | 
|  | 76 | +                    assertThat<String>(resultSet.getNString(2), equalTo<String>("Physics")) | 
|  | 77 | +                    assertThat<String>(resultSet.getNString(3), equalTo<String>("75")) | 
|  | 78 | +                    assertThat<String>(resultSet.getNString(4), equalTo<String>("85")) | 
|  | 79 | +                    assertThat<Boolean>(resultSet.next(), equalTo<Boolean>(false)) | 
|  | 80 | +                } | 
|  | 81 | +            } | 
|  | 82 | +        } | 
|  | 83 | +    } | 
|  | 84 | +} | 
0 commit comments