package Ġcc . data fabric . sc yll ard f Ċ Ċ import Ġcom . data st ax . driver . core . ResultSet Ċ import Ġcom . data st ax . driver . core . ResultSet Future Ċ import Ġcom . data st ax . driver . core . Row Ċ import Ġorg . mockito . Argument Matchers Ċ import Ġorg . mockito . Mockito Ċ Ċ object ĠTestUtils Ġ{ ĊĠĠĠ Ġfun Ġcreate ResultSet Future ( elements : ĠArray < String >): ĠResultSet Future Ġ{ ĊĠĠĠĠĠĠĠ Ġval Ġlist Ġ= Ġmutable ListOf < Row >() ĊĠĠĠĠĠĠĠ Ġelements . forEach Ġ{ ĊĠĠĠĠĠĠĠĠĠĠĠ Ġval Ġrow Ġ= ĠMockito . mock ( Row :: class . java ) ĊĠĠĠĠĠĠĠĠĠĠĠ ĠMockito .` when `( row . getString ( Argument Matchers . any Int ())). thenReturn ( it ) ĊĠĠĠĠĠĠĠĠĠĠĠ Ġlist . add ( row ) ĊĠĠĠĠĠĠĠ Ġ} ĊĊĠĠĠĠĠĠĠ Ġval Ġiterator : ĠMutable Iterator < Row > Ġ= Ġlist . iterator () Ġas ĠMutable Iterator < Row > ĊĊĠĠĠĠĠĠĠ Ġval Ġfuture Ġ= ĠMockito . mock ( ResultSet Future :: class . java ) ĊĠĠĠĠĠĠĠ Ġval ĠresultSet Ġ= ĠMockito . mock ( ResultSet :: class . java ) ĊĠĠĠĠĠĠĠ ĠMockito .` when `( resultSet . iterator ()). thenReturn ( iterator ) ĊĊĠĠĠĠĠĠĠ ĠMockito .` when `( future . get ()). thenReturn ( resultSet ) ĊĠĠĠĠĠĠĠ ĠMockito .` when `( future . un interrupt ibly ). thenReturn ( resultSet ) ĊĠĠĠĠĠĠĠ ĠMockito .` when `( future . cancel ( true )). thenReturn ( true ) ĊĊĠĠĠĠĠĠĠ Ġreturn Ġfuture ĊĠĠĠ Ġ} Ċ }