Skip to content

Commit ed3cced

Browse files
authored
Added test 584.
1 parent 21e1305 commit ed3cced

File tree

1 file changed

+57
-0
lines changed

1 file changed

+57
-0
lines changed
Lines changed: 57 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,57 @@
1+
package g0501_0600.s0584_find_customer_referee;
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 Customer(id INTEGER PRIMARY KEY, name VARCHAR, referee_id INTEGER); "
24+
+ "INSERT INTO Customer(id, name, referee_id) VALUES (1, 'Will', NULL); "
25+
+ "INSERT INTO Customer(id, name, referee_id) VALUES (2, 'Jane', NULL); "
26+
+ "INSERT INTO Customer(id, name, referee_id) VALUES (3, 'Alex', 2); "
27+
+ "INSERT INTO Customer(id, name, referee_id) VALUES (4, 'Bill', NULL); "
28+
+ "INSERT INTO Customer(id, name, referee_id) VALUES (5, 'Zack', 1); "
29+
+ "INSERT INTO Customer(id, name, referee_id) VALUES (6, 'Mark', 2); ")
30+
class MysqlTest {
31+
@Test
32+
void testScript(@EmbeddedDatabase DataSource dataSource)
33+
throws SQLException, FileNotFoundException {
34+
try (final Connection connection = dataSource.getConnection()) {
35+
try (final Statement statement = connection.createStatement();
36+
final ResultSet resultSet =
37+
statement.executeQuery(
38+
new BufferedReader(
39+
new FileReader(
40+
"src/main/java/g0501_0600/"
41+
+ "s0584_find_customer_referee/script.sql"))
42+
.lines()
43+
.collect(Collectors.joining("\n"))
44+
.replaceAll("#.*?\\r?\\n", ""))) {
45+
assertThat(resultSet.next(), equalTo(true));
46+
assertThat(resultSet.getString(1), equalTo("Will"));
47+
assertThat(resultSet.next(), equalTo(true));
48+
assertThat(resultSet.getString(1), equalTo("Jane"));
49+
assertThat(resultSet.next(), equalTo(true));
50+
assertThat(resultSet.getString(1), equalTo("Bill"));
51+
assertThat(resultSet.next(), equalTo(true));
52+
assertThat(resultSet.getString(1), equalTo("Zack"));
53+
assertThat(resultSet.next(), equalTo(false));
54+
}
55+
}
56+
}
57+
}

0 commit comments

Comments
 (0)