@@ -5175,59 +5175,20 @@ public void testRetryOnResourceExhausted() {
51755175 }
51765176
51775177 @ Test
5178- public void testSelect1HasXGoogRequestIdHeader () {
5179- SingerInfo info = SingerInfo .newBuilder ().setSingerId (1 ).build ();
5180- Statement statement = Statement .of ("SELECT * FROM FOO" );
5181- mockSpanner .putStatementResult (
5182- StatementResult .query (
5183- statement ,
5184- com .google .spanner .v1 .ResultSet .newBuilder ()
5185- .setMetadata (
5186- ResultSetMetadata .newBuilder ()
5187- .setRowType (
5188- StructType .newBuilder ()
5189- .addFields (
5190- Field .newBuilder ()
5191- .setName ("a1" )
5192- .setType (
5193- Type .newBuilder ()
5194- .setCodeValue (Integer .MAX_VALUE )
5195- .build ())
5196- .build ())
5197- .addFields (
5198- Field .newBuilder ()
5199- .setName ("b1" )
5200- .setType (
5201- Type .newBuilder ()
5202- .setCodeValue (Integer .MAX_VALUE )
5203- .build ())
5204- .build ())
5205- .build ())
5206- .build ())
5207- .addRows (
5208- ListValue .newBuilder ()
5209- .addValues (
5210- com .google .protobuf .Value .newBuilder ()
5211- .setListValue (
5212- ListValue .newBuilder ()
5213- .addValues (
5214- com .google .protobuf .Value .newBuilder ()
5215- .setNumberValue (6.626 )
5216- .build ())
5217- .addValues (
5218- com .google .protobuf .Value .newBuilder ()
5219- .setNumberValue (-6.626 )
5220- .build ())
5221- .build ())
5222- .build ())
5223- .build ())
5224- .build ()));
5225-
5178+ public void testSelectHasXGoogRequestIdHeader () {
5179+ Statement statement =
5180+ Statement .newBuilder ("select id from test where b=@p1" )
5181+ .bind ("p1" )
5182+ .toBytesArray (
5183+ Arrays .asList (ByteArray .copyFrom ("test1" ), null , ByteArray .copyFrom ("test2" )))
5184+ .build ();
5185+ mockSpanner .putStatementResult (StatementResult .query (statement , SELECT1_RESULTSET ));
52265186 DatabaseClient client =
52275187 spanner .getDatabaseClient (DatabaseId .of (TEST_PROJECT , TEST_INSTANCE , TEST_DATABASE ));
52285188 try (ResultSet resultSet = client .singleUse ().executeQuery (statement )) {
52295189 assertTrue (resultSet .next ());
5230- assertAsString (ImmutableList .of ("6.626" , "-6.626" ), resultSet , 0 );
5190+ assertEquals (1L , resultSet .getLong (0 ));
5191+ assertFalse (resultSet .next ());
52315192 }
52325193 }
52335194
0 commit comments