Skip to content

Commit 08aa7b0

Browse files
committed
Update Spring test to match the documentation
1 parent 387187d commit 08aa7b0

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

src/test/java/examples/generated/always/spring/SpringTest.java

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -83,8 +83,8 @@ public void testRender() {
8383

8484
@Test
8585
public void testSelect() {
86-
SelectStatement selectStatement = select(id.as("A_ID"), firstName, lastName, fullName)
87-
.from(generatedAlways, "a")
86+
SelectStatement selectStatement = select(id, firstName, lastName, fullName)
87+
.from(generatedAlways)
8888
.where(id, isGreaterThan(3))
8989
.orderBy(id.descending())
9090
.build()
@@ -99,12 +99,17 @@ public GeneratedAlwaysRecord mapRow(ResultSet rs, int rowNum) throws SQLExceptio
9999
record.setId(rs.getInt(1));
100100
record.setFirstName(rs.getString(2));
101101
record.setLastName(rs.getString(3));
102+
record.setFullName(rs.getString(4));
102103
return record;
103104
}
104105
});
105106

106107
assertThat(records.size()).isEqualTo(3);
107108
assertThat(records.get(0).getId()).isEqualTo(6);
109+
assertThat(records.get(0).getFirstName()).isEqualTo("Bamm Bamm");
110+
assertThat(records.get(0).getLastName()).isEqualTo("Rubble");
111+
assertThat(records.get(0).getFullName()).isEqualTo("Bamm Bamm Rubble");
112+
108113
assertThat(records.get(1).getId()).isEqualTo(5);
109114
assertThat(records.get(2).getId()).isEqualTo(4);
110115
}
@@ -138,13 +143,14 @@ public void testInsert() {
138143
.build()
139144
.render(RenderingStrategy.SPRING_NAMED_PARAMETER);
140145

141-
SqlParameterSource parameterSource = new BeanPropertySqlParameterSource(record);
146+
SqlParameterSource parameterSource = new BeanPropertySqlParameterSource(insertStatement.getRecord());
142147
KeyHolder keyHolder = new GeneratedKeyHolder();
143148

144149
int rows = template.update(insertStatement.getInsertStatement(), parameterSource, keyHolder);
150+
String generatedKey = (String) keyHolder.getKeys().get("FULL_NAME");
145151

146152
assertThat(rows).isEqualTo(1);
147-
assertThat(keyHolder.getKeys().get("FULL_NAME")).isEqualTo("Bob Jones");
153+
assertThat(generatedKey).isEqualTo("Bob Jones");
148154
}
149155

150156
@Test

0 commit comments

Comments
 (0)