Skip to content

Commit 612fea8

Browse files
committed
Update tests with session.getRequestIdCreator
1 parent f880d1c commit 612fea8

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

google-cloud-spanner/src/test/java/com/google/cloud/spanner/TransactionContextImplTest.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff 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()

0 commit comments

Comments
 (0)