Java tutorial
/* * Copyright (C) 2012-2016 DuyHai DOAN * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package info.archinnov.achilles.it; import static info.archinnov.achilles.embedded.CassandraEmbeddedConfigParameters.DEFAULT_CASSANDRA_EMBEDDED_KEYSPACE_NAME; import static org.assertj.core.api.Assertions.assertThat; import java.util.HashMap; import java.util.Iterator; import java.util.Map; import java.util.concurrent.atomic.AtomicBoolean; import org.apache.commons.lang3.RandomUtils; import org.junit.Rule; import org.junit.Test; import org.junit.rules.ExpectedException; import org.junit.runner.RunWith; import org.mockito.runners.MockitoJUnitRunner; import com.datastax.driver.core.*; import com.google.common.collect.ImmutableMap; import info.archinnov.achilles.exception.AchillesException; import info.archinnov.achilles.generated.ManagerFactory; import info.archinnov.achilles.generated.ManagerFactoryBuilder; import info.archinnov.achilles.generated.manager.SimpleEntity_Manager; import info.archinnov.achilles.internals.entities.EntityWithClusteringColumns; import info.archinnov.achilles.internals.entities.SimpleEntity; import info.archinnov.achilles.junit.AchillesTestResource; import info.archinnov.achilles.junit.AchillesTestResourceBuilder; import info.archinnov.achilles.script.ScriptExecutor; @RunWith(MockitoJUnitRunner.class) public class TestTypedQueries { @Rule public ExpectedException exception = ExpectedException.none(); @Rule public AchillesTestResource<ManagerFactory> resource = AchillesTestResourceBuilder.forJunit() .entityClassesToTruncate(SimpleEntity.class).truncateBeforeAndAfterTest() .withScript("functions/createFunctions.cql") .build((cluster, statementsCache) -> ManagerFactoryBuilder.builder(cluster) .withManagedEntityClasses(SimpleEntity.class, EntityWithClusteringColumns.class) .doForceSchemaCreation(true).withStatementsCache(statementsCache) .withDefaultKeyspaceName(DEFAULT_CASSANDRA_EMBEDDED_KEYSPACE_NAME).build()); private Session session = resource.getNativeSession(); private SimpleEntity_Manager manager = resource.getManagerFactory().forSimpleEntity(); private ScriptExecutor scriptExecutor = resource.getScriptExecutor(); @Test public void should_perform_regular_typed_query() throws Exception { //Given final Long id = RandomUtils.nextLong(0L, Long.MAX_VALUE); scriptExecutor.executeScriptTemplate("SimpleEntity/insert_single_row.cql", ImmutableMap.of("id", id, "table", "simple")); final SimpleStatement statement = new SimpleStatement("SELECT * FROM simple WHERE id = " + id); //When final SimpleEntity actual = manager.query().typedQueryForSelect(statement).getOne(); assertThat(actual).isNotNull(); assertThat(actual.getValue()).contains("0 AM"); } @Test public void should_perform_prepared_typed_query() throws Exception { //Given final Long id = RandomUtils.nextLong(0L, Long.MAX_VALUE); scriptExecutor.executeScriptTemplate("SimpleEntity/insert_single_row.cql", ImmutableMap.of("id", id, "table", "simple")); final PreparedStatement preparedStatement = session.prepare("SELECT * FROM simple WHERE id = :id"); //When final SimpleEntity actual = manager.query().typedQueryForSelect(preparedStatement, id).getOne(); //Then assertThat(actual).isNotNull(); assertThat(actual.getValue()).contains("0 AM"); } @Test public void should_perform_bound_statement_typed_query() throws Exception { //Given final Long id = RandomUtils.nextLong(0L, Long.MAX_VALUE); scriptExecutor.executeScriptTemplate("SimpleEntity/insert_single_row.cql", ImmutableMap.of("id", id, "table", "simple")); //When final BoundStatement boundStatement = session.prepare("SELECT * FROM simple WHERE id = :id").bind(id); //Then final SimpleEntity actual = manager.query().typedQueryForSelect(boundStatement).getOne(); //Then assertThat(actual).isNotNull(); assertThat(actual.getValue()).contains("0 AM"); } @Test public void should_fail_regular_typed_query_if_not_SELECT_statement() throws Exception { //Given final SimpleStatement statement = new SimpleStatement("UPDATE simple SET value = 'val' WHERE id = 1"); //When exception.expect(AchillesException.class); exception.expectMessage("Statement provided for typed query should be an SELECT statement"); manager.query().typedQueryForSelect(statement).getList(); } @Test public void should_fail_regular_typed_query_if_table_not_correspond() throws Exception { //Given final RegularStatement statement = new SimpleStatement( "SELECT * FROM entity_with_clusterings WHERE id = :id"); //When exception.expect(AchillesException.class); exception.expectMessage( "The typed query [select * from entity_with_clusterings where id = :id] should contain the table name 'simple' if the entity type is 'info.archinnov.achilles.internals.entities.SimpleEntity'"); manager.query().typedQueryForSelect(statement).getList(); } @Test public void should_iterate_regular_typed_query() throws Exception { //Given final Map<String, Object> values = new HashMap<>(); final long id = RandomUtils.nextLong(0L, Long.MAX_VALUE); values.put("id", id); values.put("date1", "'2015-10-01 00:00:00+0000'"); values.put("date2", "'2015-10-02 00:00:00+0000'"); values.put("date3", "'2015-10-03 00:00:00+0000'"); values.put("date4", "'2015-10-04 00:00:00+0000'"); values.put("date5", "'2015-10-05 00:00:00+0000'"); values.put("date6", "'2015-10-06 00:00:00+0000'"); values.put("date7", "'2015-10-07 00:00:00+0000'"); values.put("date8", "'2015-10-08 00:00:00+0000'"); values.put("date9", "'2015-10-09 00:00:00+0000'"); scriptExecutor.executeScriptTemplate("SimpleEntity/insert_many_rows.cql", values); final SimpleStatement statement = new SimpleStatement("SELECT * FROM simple WHERE id = :id LIMIT 100"); //When final Iterator<SimpleEntity> iter = manager.query().typedQueryForSelect(statement, id).iterator(); //Then final AtomicBoolean foundEntity = new AtomicBoolean(false); iter.forEachRemaining(instance -> { foundEntity.getAndSet(true); assertThat(instance).isNotNull(); assertThat(instance.getValue()).contains("id - date"); }); assertThat(foundEntity.get()).isTrue(); } }