@@ -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