File tree Expand file tree Collapse file tree 2 files changed +8
-0
lines changed
google-cloud-spanner/src/test/java/com/google/cloud/spanner Expand file tree Collapse file tree 2 files changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -1481,6 +1481,8 @@ public void testSessionNotFoundReadWriteTransaction() {
14811481 when (closedSession .getName ())
14821482 .thenReturn ("projects/dummy/instances/dummy/database/dummy/sessions/session-closed" );
14831483 when (closedSession .getErrorHandler ()).thenReturn (DefaultErrorHandler .INSTANCE );
1484+ when (closedSession .getRequestIdCreator ())
1485+ .thenReturn (new XGoogSpannerRequestId .NoopRequestIdCreator ());
14841486
14851487 Span oTspan = mock (Span .class );
14861488 ISpan span = new OpenTelemetrySpan (oTspan );
@@ -1521,6 +1523,8 @@ public void testSessionNotFoundReadWriteTransaction() {
15211523 TransactionRunnerImpl openTransactionRunner = new TransactionRunnerImpl (openSession );
15221524 openTransactionRunner .setSpan (span );
15231525 when (openSession .readWriteTransaction ()).thenReturn (openTransactionRunner );
1526+ when (openSession .getRequestIdCreator ())
1527+ .thenReturn (new XGoogSpannerRequestId .NoopRequestIdCreator ());
15241528
15251529 ResultSet openResultSet = mock (ResultSet .class );
15261530 when (openResultSet .next ()).thenReturn (true , false );
Original file line number Diff line number Diff line change @@ -67,6 +67,8 @@ public void setup() {
6767 .setCommitTimestamp (Timestamp .newBuilder ().setSeconds (99L ).setNanos (10 ).build ())
6868 .build ()));
6969 when (session .getName ()).thenReturn ("test" );
70+ when (session .getRequestIdCreator ())
71+ .thenReturn (new XGoogSpannerRequestId .NoopRequestIdCreator ());
7072 doNothing ().when (span ).setStatus (any (Throwable .class ));
7173 doNothing ().when (span ).end ();
7274 doNothing ().when (span ).addAnnotation ("Starting Commit" );
@@ -210,6 +212,8 @@ public void testReturnCommitStats() {
210212 private void batchDml (int status ) {
211213 SessionImpl session = mock (SessionImpl .class );
212214 when (session .getName ()).thenReturn ("test" );
215+ when (session .getRequestIdCreator ())
216+ .thenReturn (new XGoogSpannerRequestId .NoopRequestIdCreator ());
213217 SpannerRpc rpc = mock (SpannerRpc .class );
214218 ExecuteBatchDmlResponse response =
215219 ExecuteBatchDmlResponse .newBuilder ()
You can’t perform that action at this time.
0 commit comments