|  | 
|  | 1 | +package g3701_3800.s3716_find_churn_risk_customers; | 
|  | 2 | + | 
|  | 3 | +import static org.hamcrest.CoreMatchers.equalTo; | 
|  | 4 | +import static org.hamcrest.MatcherAssert.assertThat; | 
|  | 5 | + | 
|  | 6 | +import java.io.BufferedReader; | 
|  | 7 | +import java.io.FileNotFoundException; | 
|  | 8 | +import java.io.FileReader; | 
|  | 9 | +import java.sql.Connection; | 
|  | 10 | +import java.sql.ResultSet; | 
|  | 11 | +import java.sql.SQLException; | 
|  | 12 | +import java.sql.Statement; | 
|  | 13 | +import java.util.stream.Collectors; | 
|  | 14 | +import javax.sql.DataSource; | 
|  | 15 | +import org.junit.jupiter.api.Test; | 
|  | 16 | +import org.zapodot.junit.db.annotations.EmbeddedDatabase; | 
|  | 17 | +import org.zapodot.junit.db.annotations.EmbeddedDatabaseTest; | 
|  | 18 | +import org.zapodot.junit.db.common.CompatibilityMode; | 
|  | 19 | + | 
|  | 20 | +@EmbeddedDatabaseTest( | 
|  | 21 | +        compatibilityMode = CompatibilityMode.MySQL, | 
|  | 22 | +        initialSqls = | 
|  | 23 | +                "CREATE TABLE subscription_events (" | 
|  | 24 | +                        + "    event_id INTEGER PRIMARY KEY," | 
|  | 25 | +                        + "    user_id INTEGER NOT NULL," | 
|  | 26 | +                        + "    event_date DATE NOT NULL," | 
|  | 27 | +                        + "    event_type VARCHAR(20) NOT NULL," | 
|  | 28 | +                        + "    plan_name VARCHAR(20)," | 
|  | 29 | +                        + "    monthly_amount DECIMAL(10,2) NOT NULL" | 
|  | 30 | +                        + ");" | 
|  | 31 | +                        + "INSERT INTO subscription_events (event_id, user_id, event_date, " | 
|  | 32 | +                        + "event_type, plan_name, monthly_amount) VALUES" | 
|  | 33 | +                        + "(1, 501, '2024-01-01', 'start',     'premium', 29.99)," | 
|  | 34 | +                        + "(2, 501, '2024-02-15', 'downgrade', 'standard', 19.99)," | 
|  | 35 | +                        + "(3, 501, '2024-03-20', 'downgrade', 'basic', 9.99)," | 
|  | 36 | +                        + "(4, 502, '2024-01-05', 'start',     'standard', 19.99)," | 
|  | 37 | +                        + "(5, 502, '2024-02-10', 'upgrade',   'premium', 29.99)," | 
|  | 38 | +                        + "(6, 502, '2024-03-15', 'downgrade', 'basic', 9.99)," | 
|  | 39 | +                        + "(7, 503, '2024-01-10', 'start',     'basic', 9.99)," | 
|  | 40 | +                        + "(8, 503, '2024-02-20', 'upgrade',   'standard', 19.99)," | 
|  | 41 | +                        + "(9, 503, '2024-03-25', 'upgrade',   'premium', 29.99)," | 
|  | 42 | +                        + "(10, 504, '2024-01-15', 'start',    'premium', 29.99)," | 
|  | 43 | +                        + "(11, 504, '2024-03-01', 'downgrade','standard', 19.99)," | 
|  | 44 | +                        + "(12, 504, '2024-03-30', 'cancel',   NULL, 0.00)," | 
|  | 45 | +                        + "(13, 505, '2024-02-01', 'start',    'basic', 9.99)," | 
|  | 46 | +                        + "(14, 505, '2024-02-28', 'upgrade',  'standard', 19.99)," | 
|  | 47 | +                        + "(15, 506, '2024-01-20', 'start',    'premium', 29.99)," | 
|  | 48 | +                        + "(16, 506, '2024-03-10', 'downgrade','basic', 9.99);" | 
|  | 49 | +                        + "") | 
|  | 50 | +class MysqlTest { | 
|  | 51 | +    @Test | 
|  | 52 | +    void testScript(@EmbeddedDatabase DataSource dataSource) | 
|  | 53 | +            throws SQLException, FileNotFoundException { | 
|  | 54 | +        try (final Connection connection = dataSource.getConnection()) { | 
|  | 55 | +            try (final Statement statement = connection.createStatement(); | 
|  | 56 | +                    final ResultSet resultSet = | 
|  | 57 | +                            statement.executeQuery( | 
|  | 58 | +                                    new BufferedReader( | 
|  | 59 | +                                                    new FileReader( | 
|  | 60 | +                                                            "src/main/java/g3701_3800/" | 
|  | 61 | +                                                                    + "s3716_find_churn_risk_customers/" | 
|  | 62 | +                                                                    + "script.sql")) | 
|  | 63 | +                                            .lines() | 
|  | 64 | +                                            .collect(Collectors.joining("\n")) | 
|  | 65 | +                                            .replaceAll("#.*?\\r?\\n", ""))) { | 
|  | 66 | +                assertThat(resultSet.next(), equalTo(true)); | 
|  | 67 | +                assertThat(resultSet.getString(1), equalTo("501")); | 
|  | 68 | +                assertThat(resultSet.getString(2), equalTo("basic")); | 
|  | 69 | +                assertThat(resultSet.getString(3), equalTo("9.99")); | 
|  | 70 | +                assertThat(resultSet.getString(4), equalTo("29.99")); | 
|  | 71 | +                assertThat(resultSet.getString(5), equalTo("79")); | 
|  | 72 | +                assertThat(resultSet.next(), equalTo(true)); | 
|  | 73 | +                assertThat(resultSet.getString(1), equalTo("502")); | 
|  | 74 | +                assertThat(resultSet.getString(2), equalTo("basic")); | 
|  | 75 | +                assertThat(resultSet.getString(3), equalTo("9.99")); | 
|  | 76 | +                assertThat(resultSet.getString(4), equalTo("29.99")); | 
|  | 77 | +                assertThat(resultSet.getString(5), equalTo("70")); | 
|  | 78 | +                assertThat(resultSet.next(), equalTo(false)); | 
|  | 79 | +            } | 
|  | 80 | +        } | 
|  | 81 | +    } | 
|  | 82 | +} | 
0 commit comments