|  | 
|  | 1 | +package g3601_3700.s3642_find_books_with_polarized_opinions | 
|  | 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 books (" + | 
|  | 21 | +                "    book_id INT PRIMARY KEY," + | 
|  | 22 | +                "    title VARCHAR(255)," + | 
|  | 23 | +                "    author VARCHAR(255)," + | 
|  | 24 | +                "    genre VARCHAR(50)," + | 
|  | 25 | +                "    pages INT" + | 
|  | 26 | +                ");" + | 
|  | 27 | +                "INSERT INTO books (book_id, title, author, genre, pages) VALUES" + | 
|  | 28 | +                "(1, 'The Great Gatsby', 'F. Scott', 'Fiction', 180)," + | 
|  | 29 | +                "(2, 'To Kill a Mockingbird', 'Harper Lee', 'Fiction', 281)," + | 
|  | 30 | +                "(3, '1984', 'George Orwell', 'Dystopian', 328)," + | 
|  | 31 | +                "(4, 'Pride and Prejudice', 'Jane Austen', 'Romance', 432)," + | 
|  | 32 | +                "(5, 'The Catcher in the Rye', 'J.D. Salinger', 'Fiction', 277);" + | 
|  | 33 | +                "CREATE TABLE reading_sessions (" + | 
|  | 34 | +                "    session_id INT PRIMARY KEY," + | 
|  | 35 | +                "    book_id INT," + | 
|  | 36 | +                "    reader_name VARCHAR(100)," + | 
|  | 37 | +                "    pages_read INT," + | 
|  | 38 | +                "    session_rating INT," + | 
|  | 39 | +                "    FOREIGN KEY (book_id) REFERENCES books(book_id)" + | 
|  | 40 | +                ");" + | 
|  | 41 | +                "INSERT INTO reading_sessions (session_id, book_id, " + | 
|  | 42 | +                "reader_name, pages_read, session_rating) VALUES" + | 
|  | 43 | +                "(1, 1, 'Alice', 50, 5)," + | 
|  | 44 | +                "(2, 1, 'Bob', 60, 1)," + | 
|  | 45 | +                "(3, 1, 'Carol', 40, 4)," + | 
|  | 46 | +                "(4, 1, 'David', 30, 2)," + | 
|  | 47 | +                "(5, 1, 'Emma', 45, 5)," + | 
|  | 48 | +                "(6, 2, 'Frank', 80, 4)," + | 
|  | 49 | +                "(7, 2, 'Grace', 70, 4)," + | 
|  | 50 | +                "(8, 2, 'Henry', 90, 5)," + | 
|  | 51 | +                "(9, 2, 'Ivy', 60, 4)," + | 
|  | 52 | +                "(10, 2, 'Jack', 75, 4)," + | 
|  | 53 | +                "(11, 3, 'Kate', 100, 2)," + | 
|  | 54 | +                "(12, 3, 'Liam', 120, 1)," + | 
|  | 55 | +                "(13, 3, 'Mia', 80, 2)," + | 
|  | 56 | +                "(14, 3, 'Noah', 90, 1)," + | 
|  | 57 | +                "(15, 3, 'Olivia', 110, 4)," + | 
|  | 58 | +                "(16, 3, 'Paul', 95, 5)," + | 
|  | 59 | +                "(17, 4, 'Quinn', 150, 3)," + | 
|  | 60 | +                "(18, 4, 'Ruby', 140, 3)," + | 
|  | 61 | +                "(19, 5, 'Sam', 80, 1)," + | 
|  | 62 | +                "(20, 5, 'Tara', 70, 2);" | 
|  | 63 | +            ), | 
|  | 64 | +    ], | 
|  | 65 | +) | 
|  | 66 | +internal class MysqlTest { | 
|  | 67 | +    @Test | 
|  | 68 | +    @Throws(SQLException::class, FileNotFoundException::class) | 
|  | 69 | +    fun testScript(@EmbeddedDatabase dataSource: DataSource) { | 
|  | 70 | +        dataSource.connection.use { connection -> | 
|  | 71 | +            connection.createStatement().use { statement -> | 
|  | 72 | +                statement.executeQuery( | 
|  | 73 | +                    BufferedReader( | 
|  | 74 | +                        FileReader( | 
|  | 75 | +                            ( | 
|  | 76 | +                                "src/main/kotlin/g3601_3700/" + | 
|  | 77 | +                                    "s3642_find_books_with_" + | 
|  | 78 | +                                    "polarized_opinions/" + | 
|  | 79 | +                                    "script.sql" | 
|  | 80 | +                                ), | 
|  | 81 | +                        ), | 
|  | 82 | +                    ) | 
|  | 83 | +                        .lines() | 
|  | 84 | +                        .collect(Collectors.joining("\n")) | 
|  | 85 | +                        .replace("#.*?\\r?\\n".toRegex(), ""), | 
|  | 86 | +                ).use { resultSet -> | 
|  | 87 | +                    assertThat<Boolean>(resultSet.next(), equalTo<Boolean>(true)) | 
|  | 88 | +                    assertThat<String>(resultSet.getNString(1), equalTo<String>("1")) | 
|  | 89 | +                    assertThat<String>( | 
|  | 90 | +                        resultSet.getNString(2), | 
|  | 91 | +                        equalTo<String>("The Great Gatsby"), | 
|  | 92 | +                    ) | 
|  | 93 | +                    assertThat<String>( | 
|  | 94 | +                        resultSet.getNString(3), | 
|  | 95 | +                        equalTo<String>("F. Scott"), | 
|  | 96 | +                    ) | 
|  | 97 | +                    assertThat<String>(resultSet.getNString(4), equalTo<String>("Fiction")) | 
|  | 98 | +                    assertThat<String>(resultSet.getNString(5), equalTo<String>("180")) | 
|  | 99 | +                    assertThat<String>(resultSet.getNString(6), equalTo<String>("4")) | 
|  | 100 | +                    assertThat<String>(resultSet.getNString(7), equalTo<String>("1.00")) | 
|  | 101 | +                    assertThat<Boolean>(resultSet.next(), equalTo<Boolean>(true)) | 
|  | 102 | +                    assertThat<String>(resultSet.getNString(1), equalTo<String>("3")) | 
|  | 103 | +                    assertThat<String>(resultSet.getNString(2), equalTo<String>("1984")) | 
|  | 104 | +                    assertThat<String>( | 
|  | 105 | +                        resultSet.getNString(3), | 
|  | 106 | +                        equalTo<String>("George Orwell"), | 
|  | 107 | +                    ) | 
|  | 108 | +                    assertThat<String>( | 
|  | 109 | +                        resultSet.getNString(4), | 
|  | 110 | +                        equalTo<String>("Dystopian"), | 
|  | 111 | +                    ) | 
|  | 112 | +                    assertThat<String>(resultSet.getNString(5), equalTo<String>("328")) | 
|  | 113 | +                    assertThat<String>(resultSet.getNString(6), equalTo<String>("4")) | 
|  | 114 | +                    assertThat<String>(resultSet.getNString(7), equalTo<String>("1.00")) | 
|  | 115 | +                    assertThat<Boolean>(resultSet.next(), equalTo<Boolean>(false)) | 
|  | 116 | +                } | 
|  | 117 | +            } | 
|  | 118 | +        } | 
|  | 119 | +    } | 
|  | 120 | +} | 
0 commit comments