|
4 | 4 |
|
5 | 5 | import com.datastax.oss.driver.api.core.CqlIdentifier; |
6 | 6 | import com.datastax.oss.driver.api.core.CqlSession; |
| 7 | +import com.datastax.oss.driver.api.core.DefaultConsistencyLevel; |
7 | 8 | import com.datastax.oss.driver.api.core.config.DefaultDriverOption; |
8 | 9 | import com.datastax.oss.driver.api.core.cql.PreparedStatement; |
9 | 10 | import com.datastax.oss.driver.api.core.cql.ResultSet; |
@@ -120,6 +121,10 @@ public class DefaultMetadataTabletMapIT { |
120 | 121 | private static final SimpleStatement STMT_SELECT_CK_CONCRETE = |
121 | 122 | buildStatement("SELECT pk, ck FROM %s.%s WHERE pk = ? AND ck = 1"); |
122 | 123 |
|
| 124 | + private static final SimpleStatement STMT_SELECT_LOCAL_SERIAL = |
| 125 | + buildStatement("SELECT pk,ck FROM %s.%s WHERE pk = ? AND ck = ?") |
| 126 | + .setConsistencyLevel(DefaultConsistencyLevel.LOCAL_SERIAL); |
| 127 | + |
123 | 128 | private static final SimpleStatement STMT_UPDATE = |
124 | 129 | buildStatement("UPDATE %s.%s SET val = 1 WHERE pk = ? AND ck = ?"); |
125 | 130 |
|
@@ -195,6 +200,8 @@ public void every_statement_should_deliver_tablet_info() { |
195 | 200 | statements.put("SELECT_CONCRETE_PREPARED", s -> s.prepare(STMT_SELECT_CONCRETE).bind()); |
196 | 201 | statements.put("SELECT_PK_CONCRETE_PREPARED", s -> s.prepare(STMT_SELECT_PK_CONCRETE).bind(2)); |
197 | 202 | statements.put("SELECT_CK_CONCRETE_PREPARED", s -> s.prepare(STMT_SELECT_CK_CONCRETE).bind(2)); |
| 203 | + statements.put( |
| 204 | + "SELECT_LOCAL_SERIAL_PREPARED", s -> s.prepare(STMT_SELECT_LOCAL_SERIAL).bind(2, 2)); |
198 | 205 | statements.put("INSERT_CONCRETE", s -> STMT_INSERT_CONCRETE); |
199 | 206 | statements.put("INSERT_PREPARED", s -> s.prepare(STMT_INSERT).bind(2, 2)); |
200 | 207 | statements.put("INSERT_NO_KS_PREPARED", s -> s.prepare(STMT_INSERT_NO_KS).bind(2, 2)); |
@@ -256,6 +263,13 @@ public void every_statement_should_deliver_tablet_info() { |
256 | 263 | ex.addSuppressed(e); |
257 | 264 | throw ex; |
258 | 265 | } |
| 266 | + if (stmtEntry.getKey().contains("LOCAL_SERIAL") && !stmt.isLWT()) { |
| 267 | + testErrors.add( |
| 268 | + String.format( |
| 269 | + "Statement %s on session %s was not recognized as LWT", |
| 270 | + stmtEntry.getKey(), sessionEntry.getKey())); |
| 271 | + continue; |
| 272 | + } |
259 | 273 | try { |
260 | 274 | if (!executeOnAllHostsAndReturnIfResultHasTabletsInfo(session, stmt)) { |
261 | 275 | testErrors.add( |
|
0 commit comments