|
20 | 20 | FixedSizePool, |
21 | 21 | BatchCreateSessionsRequest, |
22 | 22 | ExecuteSqlRequest, |
23 | | - GetSessionRequest, |
24 | 23 | BeginTransactionRequest, |
25 | 24 | TransactionOptions, |
26 | 25 | ) |
@@ -55,19 +54,16 @@ def test_stale_read_multi_use(self): |
55 | 54 |
|
56 | 55 | # Verify the requests that we got. |
57 | 56 | requests = self.spanner_service.requests |
58 | | - eq_(9, len(requests)) |
| 57 | + eq_(7, len(requests)) |
59 | 58 | is_instance_of(requests[0], BatchCreateSessionsRequest) |
60 | | - # We should get rid of this extra round-trip for GetSession.... |
61 | | - is_instance_of(requests[1], GetSessionRequest) |
62 | | - is_instance_of(requests[2], BeginTransactionRequest) |
| 59 | + is_instance_of(requests[1], BeginTransactionRequest) |
| 60 | + is_instance_of(requests[2], ExecuteSqlRequest) |
63 | 61 | is_instance_of(requests[3], ExecuteSqlRequest) |
64 | | - is_instance_of(requests[4], ExecuteSqlRequest) |
65 | | - is_instance_of(requests[5], GetSessionRequest) |
66 | | - is_instance_of(requests[6], BeginTransactionRequest) |
67 | | - is_instance_of(requests[7], ExecuteSqlRequest) |
68 | | - is_instance_of(requests[8], ExecuteSqlRequest) |
| 62 | + is_instance_of(requests[4], BeginTransactionRequest) |
| 63 | + is_instance_of(requests[5], ExecuteSqlRequest) |
| 64 | + is_instance_of(requests[6], ExecuteSqlRequest) |
69 | 65 | # Verify that the transaction is a read-only transaction. |
70 | | - for index in [2, 6]: |
| 66 | + for index in [1, 4]: |
71 | 67 | begin_request: BeginTransactionRequest = requests[index] |
72 | 68 | eq_( |
73 | 69 | TransactionOptions( |
@@ -105,15 +101,12 @@ def test_stale_read_single_use(self): |
105 | 101 |
|
106 | 102 | # Verify the requests that we got. |
107 | 103 | requests = self.spanner_service.requests |
108 | | - eq_(5, len(requests)) |
| 104 | + eq_(3, len(requests)) |
109 | 105 | is_instance_of(requests[0], BatchCreateSessionsRequest) |
110 | | - # We should get rid of this extra round-trip for GetSession.... |
111 | | - is_instance_of(requests[1], GetSessionRequest) |
| 106 | + is_instance_of(requests[1], ExecuteSqlRequest) |
112 | 107 | is_instance_of(requests[2], ExecuteSqlRequest) |
113 | | - is_instance_of(requests[3], GetSessionRequest) |
114 | | - is_instance_of(requests[4], ExecuteSqlRequest) |
115 | 108 | # Verify that the requests use a stale read. |
116 | | - for index in [2, 4]: |
| 109 | + for index in [1, 2]: |
117 | 110 | execute_request: ExecuteSqlRequest = requests[index] |
118 | 111 | eq_( |
119 | 112 | TransactionOptions( |
|
0 commit comments