|  | 
|  | 1 | +package g3601_3700.s3611_find_overbooked_employees | 
|  | 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 employees (" + | 
|  | 21 | +                "    employee_id INTEGER," + | 
|  | 22 | +                "    employee_name VARCHAR(50)," + | 
|  | 23 | +                "    department VARCHAR(50)" + | 
|  | 24 | +                ");" + | 
|  | 25 | +                "INSERT INTO employees (employee_id, employee_name, department) VALUES" + | 
|  | 26 | +                "    (1, 'Alice Johnson', 'Engineering')," + | 
|  | 27 | +                "    (2, 'Bob Smith', 'Marketing')," + | 
|  | 28 | +                "    (3, 'Carol Davis', 'Sales')," + | 
|  | 29 | +                "    (4, 'David Wilson', 'Engineering')," + | 
|  | 30 | +                "    (5, 'Emma Brown', 'HR');" + | 
|  | 31 | +                "CREATE TABLE meetings (" + | 
|  | 32 | +                "    meeting_id INTEGER," + | 
|  | 33 | +                "    employee_id INTEGER," + | 
|  | 34 | +                "    meeting_date DATE," + | 
|  | 35 | +                "    meeting_type VARCHAR(20)," + | 
|  | 36 | +                "    duration_hours DECIMAL(4,1)" + | 
|  | 37 | +                ");" + | 
|  | 38 | +                "INSERT INTO meetings (meeting_id, employee_id, " + | 
|  | 39 | +                "meeting_date, meeting_type, duration_hours) VALUES" + | 
|  | 40 | +                "    (1, 1, '2023-06-05', 'Team', 8.0)," + | 
|  | 41 | +                "    (2, 1, '2023-06-06', 'Client', 6.0)," + | 
|  | 42 | +                "    (3, 1, '2023-06-07', 'Training', 7.0)," + | 
|  | 43 | +                "    (4, 1, '2023-06-12', 'Team', 12.0)," + | 
|  | 44 | +                "    (5, 1, '2023-06-13', 'Client', 9.0)," + | 
|  | 45 | +                "    (6, 2, '2023-06-05', 'Team', 15.0)," + | 
|  | 46 | +                "    (7, 2, '2023-06-06', 'Client', 8.0)," + | 
|  | 47 | +                "    (8, 2, '2023-06-12', 'Training', 10.0)," + | 
|  | 48 | +                "    (9, 3, '2023-06-05', 'Team', 4.0)," + | 
|  | 49 | +                "    (10, 3, '2023-06-06', 'Client', 3.0)," + | 
|  | 50 | +                "    (11, 4, '2023-06-05', 'Team', 25.0)," + | 
|  | 51 | +                "    (12, 4, '2023-06-19', 'Client', 22.0)," + | 
|  | 52 | +                "    (13, 5, '2023-06-05', 'Training', 2.0);" | 
|  | 53 | +            ), | 
|  | 54 | +    ], | 
|  | 55 | +) | 
|  | 56 | +internal class MysqlTest { | 
|  | 57 | +    @Test | 
|  | 58 | +    @Throws(SQLException::class, FileNotFoundException::class) | 
|  | 59 | +    fun testScript(@EmbeddedDatabase dataSource: DataSource) { | 
|  | 60 | +        dataSource.connection.use { connection -> | 
|  | 61 | +            connection.createStatement().use { statement -> | 
|  | 62 | +                statement.executeQuery( | 
|  | 63 | +                    BufferedReader( | 
|  | 64 | +                        FileReader( | 
|  | 65 | +                            ( | 
|  | 66 | +                                "src/main/kotlin/g3601_3700/" + | 
|  | 67 | +                                    "s3611_find_overbooked_employees/" + | 
|  | 68 | +                                    "script.sql" | 
|  | 69 | +                                ), | 
|  | 70 | +                        ), | 
|  | 71 | +                    ) | 
|  | 72 | +                        .lines() | 
|  | 73 | +                        .collect(Collectors.joining("\n")) | 
|  | 74 | +                        .replace("#.*?\\r?\\n".toRegex(), "") | 
|  | 75 | +                        .replace("WEEKOFYEAR", "ISO_WEEK"), | 
|  | 76 | +                ).use { resultSet -> | 
|  | 77 | +                    assertThat<Boolean>(resultSet.next(), equalTo<Boolean>(true)) | 
|  | 78 | +                    assertThat<String>(resultSet.getNString(1), equalTo<String>("1")) | 
|  | 79 | +                    assertThat<String>( | 
|  | 80 | +                        resultSet.getNString(2), | 
|  | 81 | +                        equalTo<String>("Alice Johnson"), | 
|  | 82 | +                    ) | 
|  | 83 | +                    assertThat<String>( | 
|  | 84 | +                        resultSet.getNString(3), | 
|  | 85 | +                        equalTo<String>("Engineering"), | 
|  | 86 | +                    ) | 
|  | 87 | +                    assertThat<String>(resultSet.getNString(4), equalTo<String>("2")) | 
|  | 88 | +                    assertThat<Boolean>(resultSet.next(), equalTo<Boolean>(true)) | 
|  | 89 | +                    assertThat<String>(resultSet.getNString(1), equalTo<String>("4")) | 
|  | 90 | +                    assertThat<String>( | 
|  | 91 | +                        resultSet.getNString(2), | 
|  | 92 | +                        equalTo<String>("David Wilson"), | 
|  | 93 | +                    ) | 
|  | 94 | +                    assertThat<String>( | 
|  | 95 | +                        resultSet.getNString(3), | 
|  | 96 | +                        equalTo<String>("Engineering"), | 
|  | 97 | +                    ) | 
|  | 98 | +                    assertThat<String>(resultSet.getNString(4), equalTo<String>("2")) | 
|  | 99 | +                    assertThat<Boolean>(resultSet.next(), equalTo<Boolean>(false)) | 
|  | 100 | +                } | 
|  | 101 | +            } | 
|  | 102 | +        } | 
|  | 103 | +    } | 
|  | 104 | +} | 
0 commit comments