@@ -83,8 +83,8 @@ public void testRender() {
83
83
84
84
@ Test
85
85
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 )
88
88
.where (id , isGreaterThan (3 ))
89
89
.orderBy (id .descending ())
90
90
.build ()
@@ -99,12 +99,17 @@ public GeneratedAlwaysRecord mapRow(ResultSet rs, int rowNum) throws SQLExceptio
99
99
record .setId (rs .getInt (1 ));
100
100
record .setFirstName (rs .getString (2 ));
101
101
record .setLastName (rs .getString (3 ));
102
+ record .setFullName (rs .getString (4 ));
102
103
return record ;
103
104
}
104
105
});
105
106
106
107
assertThat (records .size ()).isEqualTo (3 );
107
108
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
+
108
113
assertThat (records .get (1 ).getId ()).isEqualTo (5 );
109
114
assertThat (records .get (2 ).getId ()).isEqualTo (4 );
110
115
}
@@ -138,13 +143,14 @@ public void testInsert() {
138
143
.build ()
139
144
.render (RenderingStrategy .SPRING_NAMED_PARAMETER );
140
145
141
- SqlParameterSource parameterSource = new BeanPropertySqlParameterSource (record );
146
+ SqlParameterSource parameterSource = new BeanPropertySqlParameterSource (insertStatement . getRecord () );
142
147
KeyHolder keyHolder = new GeneratedKeyHolder ();
143
148
144
149
int rows = template .update (insertStatement .getInsertStatement (), parameterSource , keyHolder );
150
+ String generatedKey = (String ) keyHolder .getKeys ().get ("FULL_NAME" );
145
151
146
152
assertThat (rows ).isEqualTo (1 );
147
- assertThat (keyHolder . getKeys (). get ( "FULL_NAME" ) ).isEqualTo ("Bob Jones" );
153
+ assertThat (generatedKey ).isEqualTo ("Bob Jones" );
148
154
}
149
155
150
156
@ Test
0 commit comments