@Override public Iterator<X> iterate() { return ((com.impetus.kundera.query.Query)query).iterate(); }
/** * Assert book scrolling. * * @param queryClause * the query clause * @param expected * the expected * @return the list */ private List<Book> assertBookScrolling(String queryClause, int expected) { Query query = (Query) em.createQuery(queryClause, Book.class); int count = 0; Iterator<Book> bookItr = query.iterate(); List<Book> books = new ArrayList<Book>(); while (bookItr.hasNext()) { Book book = bookItr.next(); Assert.assertNotNull(book); count++; books.add(book); } Assert.assertTrue(count > 0); Assert.assertTrue(count == expected); return books; }
@Test public void chunkScrollTest() throws Exception { //without where clause int chunkSize = 50; Query query = (Query) em.createQuery("Select p from PersonHBase p", PersonHBase.class); query.setFetchSize(1000); Iterator<PersonHBase> iter = query.iterate(); IResultIterator<PersonHBase> iIter = (IResultIterator<PersonHBase>) iter; assertScrolling(chunkSize, iIter); //with where clause String name = "vivek"; query = (Query) em.createQuery("Select p from PersonHBase p where p.personName = '"+name+"'", PersonHBase.class); query.setFetchSize(1000); iter = query.iterate(); iIter = (IResultIterator<PersonHBase>) iter; assertScrolling(chunkSize, iIter); }
private void assertOnTokenScroll(String queryClause, int expected) { Query query = (Query) em.createQuery(queryClause, HbaseToken.class); int count = 0; Iterator<HbaseToken> tokens = query.iterate(); while (tokens.hasNext()) { HbaseToken token = tokens.next(); Assert.assertNotNull(token); Assert.assertNotNull(token.getClient()); Assert.assertEquals(2, token.getClient().getTokens().size()); count++; } Assert.assertTrue(count > 0); Assert.assertTrue(count == expected); }
private void assertOnTokenScroll(String queryClause, int expected) { Query query = (Query) em.createQuery(queryClause, MongoToken.class); int count = 0; Iterator<MongoToken> tokens = query.iterate(); while (tokens.hasNext()) { MongoToken token = tokens.next(); Assert.assertNotNull(token); Assert.assertNotNull(token.getClient()); Assert.assertEquals(2, token.getClient().getTokens().size()); count++; } Assert.assertTrue(count > 0); Assert.assertTrue(count == expected); }
private void assertOnTokenScroll(String queryClause, int expected) { Query query = (Query) em.createQuery(queryClause, CouchDBToken.class); int count = 0; Iterator<CouchDBToken> tokens = query.iterate(); while (tokens.hasNext()) { CouchDBToken token = tokens.next(); Assert.assertNotNull(token); Assert.assertNotNull(token.getClient()); Assert.assertEquals(2, token.getClient().getTokens().size()); count++; } Assert.assertTrue(count > 0); Assert.assertTrue(count == expected); }
private void assertOnFetch(Query query, Integer fetchSize, int available) { query.setFetchSize(fetchSize); int counter=0; Iterator<PrimeUser> iter = query.iterate(); while (iter.hasNext()) { Assert.assertNotNull(iter.next()); counter++; } Assert.assertEquals(counter, fetchSize != null && fetchSize == 0 ? 0:available); try { iter.next(); Assert.fail(); } catch (NoSuchElementException nsex) { Assert.assertNotNull(nsex.getMessage()); } }
private void assertOnFetch(Query query, Integer fetchSize, int expectedCount) { query.setFetchSize(fetchSize); int counter = 0; Iterator<PersonCouchDB> iter = query.iterate(); while (iter.hasNext()) { Assert.assertNotNull(iter.next()); counter++; } Assert.assertEquals(counter, fetchSize == null || expectedCount < fetchSize ? expectedCount : fetchSize); try { iter.next(); Assert.fail(); } catch (NoSuchElementException nsex) { Assert.assertNotNull(nsex.getMessage()); } }
private void assertOnFetch(Query query, Integer fetchSize, int available) { query.setFetchSize(fetchSize); int counter = 0; Iterator<PersonMongo> iter = query.iterate(); while (iter.hasNext()) { Assert.assertNotNull(iter.next()); counter++; } Assert.assertEquals(counter, fetchSize == null || available < fetchSize ? available : fetchSize); try { iter.next(); Assert.fail(); } catch (NoSuchElementException nsex) { Assert.assertNotNull(nsex.getMessage()); } }
private void assertOnTokenScroll(String queryClause, int expected) { Query query = (Query) em.createQuery(queryClause, Token.class); int count=0; Iterator<Token> tokens = query.iterate(); while(tokens.hasNext()) { Token token = tokens.next(); Assert.assertNotNull(token); Assert.assertNotNull(token.getClient()); Assert.assertEquals(2, token.getClient().getTokens().size()); count++; } Assert.assertTrue(count > 0); Assert.assertTrue(count == expected); }
private void assertOnTokenScroll(String queryClause, int expected) { com.impetus.kundera.query.Query query = (com.impetus.kundera.query.Query) em.createQuery(queryClause, Token.class); int count=0; Iterator<Token> tokens = query.iterate(); while(tokens.hasNext()) { Token token = tokens.next(); Assert.assertNotNull(token); Assert.assertNotNull(token.getClient()); Assert.assertEquals(2, token.getClient().getTokens().size()); count++; } Assert.assertTrue(count > 0); Assert.assertTrue(count == expected); }
private void assertOnFetch(Query query, Integer fetchSize, int available) { query.setFetchSize(fetchSize); int counter=0; Iterator<CassandraPrimeUser> iter = query.iterate(); while (iter.hasNext()) { Assert.assertNotNull(iter.next()); counter++; } Assert.assertEquals(counter, fetchSize == null || available < fetchSize?available:fetchSize); try { iter.next(); Assert.fail(); } catch (NoSuchElementException nsex) { Assert.assertNotNull(nsex.getMessage()); } }
private void assertOnFetch(Query query, Integer fetchSize, int available) { query.setFetchSize(fetchSize); int counter = 0; Iterator<PersonHBase> iter = query.iterate(); while (iter.hasNext()) { Assert.assertNotNull(iter.next()); counter++; } Assert.assertEquals(counter, fetchSize == null || available < fetchSize ? available : fetchSize); try { iter.next(); Assert.fail(); } catch (NoSuchElementException nsex) { Assert.assertNotNull(nsex.getMessage()); } }
private void assertOnFetch(Query query, Integer fetchSize, int available) { query.setFetchSize(fetchSize); int counter = 0; Iterator<PersonRDBMS> iter = query.iterate(); while (iter.hasNext()) { Assert.assertNotNull(iter.next()); counter++; } Assert.assertEquals(counter, fetchSize == null || available < fetchSize ? available : fetchSize); try { iter.next(); Assert.fail(); } catch (NoSuchElementException nsex) { Assert.assertNotNull(nsex.getMessage()); } }
private void assertOnTokenScroll(String queryClause, int expected) { Query query = (Query) em.createQuery(queryClause, RDBMSToken.class); int count = 0; Iterator<RDBMSToken> tokens = query.iterate(); while (tokens.hasNext()) { RDBMSToken token = tokens.next(); Assert.assertNotNull(token); RDBMSTokenClient client = token.getClient(); Assert.assertNotNull(client); Assert.assertEquals("tokenClient1", client.getClientName()); // Assert.assertEquals(2, client.getTokens().size()); count++; } Assert.assertTrue(count > 0); Assert.assertTrue(count == expected); }
private void assertOnFetch(Query query, Integer fetchSize, int available) { query.setFetchSize(fetchSize); int counter=0; try { Iterator<PersonCassandra> iter = query.iterate(); while (iter.hasNext()) { Assert.assertNotNull(iter.next()); counter++; } Assert.assertEquals(counter, fetchSize == null || available < fetchSize?available:fetchSize); try { iter.next(); Assert.fail(); } catch (NoSuchElementException nsex) { Assert.assertNotNull(nsex.getMessage()); } }catch(UnsupportedOperationException e) { Assert.assertEquals("Iteration not supported over native queries",e.getMessage()); } }
private void assertOnFetch(Query query, Integer fetchSize, int available) { query.setFetchSize(fetchSize); int counter=0; try { Iterator<PersonCassandra> iter = query.iterate(); while (iter.hasNext()) { Assert.assertNotNull(iter.next()); counter++; } Assert.assertEquals(counter, fetchSize != null && fetchSize == 0 ? 0:available); try { iter.next(); Assert.fail(); } catch (NoSuchElementException nsex) { Assert.assertNotNull(nsex.getMessage()); } }catch(UnsupportedOperationException e) { Assert.assertEquals("Iteration not supported over native queries",e.getMessage()); } }
Iterator<Book> iter = query.iterate();
@Test public void testIteratorWithOneFilter() { Map<String, Client> clients = (Map<String, Client>) em.getDelegate(); HBaseClient client = (HBaseClient) clients.get("hbaseTest"); Filter filter = new PrefixFilter(Bytes.toBytes("100" + "_")); client.setFilter(new KeyOnlyFilter()); client.addFilter("city_similarity", filter); String query2 = "select u from CitySimilarity u"; com.impetus.kundera.query.Query queryObject = (com.impetus.kundera.query.Query) em .createQuery(query2); queryObject.setFetchSize(10); Iterator<CitySimilarity> resultIterator = queryObject.iterate(); CitySimilarity cityS = null; int counter = 0; while (resultIterator.hasNext()) { counter++; cityS = resultIterator.next(); Assert.assertNotNull(cityS); Assert.assertNotNull(cityS.getId()); Assert.assertNull(cityS.getBookInfo()); } Assert.assertEquals(2, counter); }
@Test public void testIterator() { String query2 = "select u from CitySimilarity u"; com.impetus.kundera.query.Query queryObject = (com.impetus.kundera.query.Query) em .createQuery(query2); queryObject.setFetchSize(10); Iterator<CitySimilarity> resultIterator = queryObject.iterate(); CitySimilarity cityS = null; int counter = 0; while (resultIterator.hasNext()) { counter++; cityS = resultIterator.next(); Assert.assertNotNull(cityS); Assert.assertNotNull(cityS.getId()); Assert.assertNotNull(cityS.getBookInfo()); Assert.assertNotNull(cityS.getBookInfo().getBookId()); } Assert.assertEquals(2, counter); }