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.Date; 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.generated.ManagerFactory; import info.archinnov.achilles.generated.ManagerFactoryBuilder; import info.archinnov.achilles.generated.manager.SimpleEntity_Manager; 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; import info.archinnov.achilles.type.TypedMap; @RunWith(MockitoJUnitRunner.class) public class TestNativeQueries { @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).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_native_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 TypedMap actual = manager.query().nativeQuery(statement).getOne(); assertThat(actual).isNotNull(); assertThat(actual.<String>getTyped("value")).contains("0 AM"); } @Test public void should_perform_prepared_native_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 TypedMap actual = manager.query().nativeQuery(preparedStatement, id).getOne(); //Then assertThat(actual).isNotNull(); assertThat(actual.<String>getTyped("value")).contains("0 AM"); } @Test public void should_perform_bound_statement_native_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 TypedMap actual = manager.query().nativeQuery(boundStatement).getOne(); //Then assertThat(actual).isNotNull(); assertThat(actual.<String>getTyped("value")).contains("0 AM"); } @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<TypedMap> iter = manager.query().nativeQuery(statement, id).iterator(); //Then final AtomicBoolean foundEntity = new AtomicBoolean(false); iter.forEachRemaining(instance -> { foundEntity.getAndSet(true); assertThat(instance).isNotNull(); assertThat(instance.<String>getTyped("value")).contains("id - date"); }); assertThat(foundEntity.get()).isTrue(); } @Test public void should_perform_regular_insert_as_native_query() throws Exception { //Given final long id = RandomUtils.nextLong(0L, Long.MAX_VALUE); final SimpleStatement statement = new SimpleStatement( "INSERT INTO simple(id, date, value) VALUES(:id, :date, :value)"); //When manager.query().nativeQuery(statement, id, new Date(), "val").execute(); //Then final Row actual = session.execute("SELECT * FROM simple WHERE id = " + id).one(); assertThat(actual).isNotNull(); assertThat(actual.getString("value")).isEqualTo("val"); } }