@Before public void before() { CqlResult result = new CqlResult(); result.setRows(ImmutableList.of()); when(queryExecutor.execute(any(), any())).thenAnswer(invocation -> { Uninterruptibles.sleepUninterruptibly(queryDelayMillis, TimeUnit.MILLISECONDS); return result; }); CqlPreparedResult preparedResult = new CqlPreparedResult(); preparedResult.setItemId(1); when(queryExecutor.prepare(any(), any(), any())).thenReturn(preparedResult); when(queryExecutor.executePrepared(eq(1), any())).thenReturn(result); }
@Test public void getSizeForCqlResultWithRows() { assertThat(ThriftObjectSizeUtils.getCqlResultSize( new CqlResult(CqlResultType.ROWS).setRows(ImmutableList.of(new CqlRow())))) .isEqualTo(NULL_SIZE * 5); }
@Test public void passesCqlQueriesAndResultsToAndFromDelegate() throws TException { ByteBuffer byteBuffer = ByteBuffer.allocate(1); CqlResult standardCqlResult = new CqlResult(CqlResultType.ROWS) .setRows(ImmutableList.of(new CqlRow(byteBuffer, ImmutableList.of(new Column(byteBuffer))))); setDelegateResponseToCqlQuery(standardCqlResult); assertThat(executeCqlQuery()).isEqualTo(standardCqlResult); verifyCqlQueryWasExecuted(); }
private CqlResult makeCqlResult(Entry[][] rowsOfcolsOfKvps, int position) { CqlResult result = new CqlResult(CqlResultType.ROWS); CqlMetadata meta = null; CqlRow row = null; Column column = null; List<Column> columnlist = new LinkedList<Column>(); List<CqlRow> rowlist = new LinkedList<CqlRow>(); List<String> colNamesList = new ArrayList<String>(); for (int rowcnt = 0; rowcnt < rowsOfcolsOfKvps.length; rowcnt++ ) { colNamesList = new ArrayList<String>(); columnlist = new LinkedList<Column>(); for (int colcnt = 0; colcnt < rowsOfcolsOfKvps[0].length; colcnt++ ) { column = makeColumn(rowsOfcolsOfKvps[rowcnt][colcnt].name,rowsOfcolsOfKvps[rowcnt][colcnt].value); columnlist.add(column); colNamesList.add(rowsOfcolsOfKvps[rowcnt][colcnt].name); } row = makeRow(rowsOfcolsOfKvps[rowcnt][position-1].name,columnlist); rowlist.add(row); } meta = makeMetadataAllString(colNamesList); result.setSchema(meta).setRows(rowlist); return result; }
result.setSchema(meta).setRows(rowlist); return result;
res.setRows(cqlRows).setSchema(schema); return res;
res.setRows(cqlRows).setSchema(schema); return res;
res.setRows(cqlRows).setSchema(schema); return res;
unsetRows(); } else { setRows((List<CqlRow>)value);
unsetRows(); } else { setRows((List<CqlRow>)value);
unsetRows(); } else { setRows((List<CqlRow>)value);