Java tutorial
/* * Copyright 2013 the original author or authors. * * 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 com.vladmihalcea; import com.vladmihalcea.hibernate.model.store.Company; import com.vladmihalcea.hibernate.model.store.Product; import com.vladmihalcea.hibernate.model.store.WarehouseProductInfo; import com.vladmihalcea.service.WarehouseProductInfoService; import com.vladmihalcea.sql.SQLStatementCountValidator; import com.vladmihalcea.sql.exception.SQLSelectCountMismatchException; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.test.annotation.DirtiesContext; import org.springframework.test.context.ContextConfiguration; import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; import org.springframework.transaction.TransactionStatus; import org.springframework.transaction.support.TransactionCallback; import org.springframework.transaction.support.TransactionTemplate; import javax.persistence.EntityManager; import javax.persistence.PersistenceContext; import javax.sql.DataSource; import java.util.List; import java.util.Map; import static junit.framework.Assert.*; @RunWith(SpringJUnit4ClassRunner.class) @ContextConfiguration(locations = { "classpath:spring/applicationContext-test.xml" }) @DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_CLASS) public class HibernateSQLStatementCountTest { private static final Logger LOG = LoggerFactory.getLogger(HibernateSQLStatementCountTest.class); @PersistenceContext(unitName = "persistenceUnit") private EntityManager entityManager; @Autowired private TransactionTemplate transactionTemplate; @Autowired private WarehouseProductInfoService warehouseProductInfoService; @Autowired private DataSource otherDataSource; @Before public void beforeTest() { CleanDbUtil.cleanStore(transactionTemplate, entityManager); SQLStatementCountValidator.reset(); } @Test public void test() { transactionTemplate.execute(new TransactionCallback<Void>() { @Override public Void doInTransaction(TransactionStatus transactionStatus) { Company company = new Company(); company.setName("TV Company"); entityManager.persist(company); Product product1 = new Product("tvCode"); product1.setName("TV"); product1.setCompany(company); WarehouseProductInfo warehouseProductInfo1 = new WarehouseProductInfo(); warehouseProductInfo1.setQuantity(101); product1.addWarehouse(warehouseProductInfo1); Product product2 = new Product("cdCode"); product2.setName("CD"); product2.setCompany(company); WarehouseProductInfo warehouseProductInfo2 = new WarehouseProductInfo(); warehouseProductInfo2.setQuantity(50); product2.addWarehouse(warehouseProductInfo2); entityManager.persist(product1); entityManager.persist(product2); entityManager.flush(); final JdbcTemplate otherDataSourceJdbcTemplate = new JdbcTemplate(otherDataSource); List<Map<String, Object>> versions = otherDataSourceJdbcTemplate .queryForList(" select * from version "); assertTrue(versions.isEmpty()); return null; } }); try { SQLStatementCountValidator.reset(); warehouseProductInfoService.findAllWithNPlusOne(); SQLStatementCountValidator.assertSelectCount(1); } catch (SQLSelectCountMismatchException e) { assertEquals(3, e.getRecorded()); } SQLStatementCountValidator.reset(); warehouseProductInfoService.findAllWithFetch(); SQLStatementCountValidator.assertSelectCount(1); SQLStatementCountValidator.reset(); warehouseProductInfoService.newWarehouseProductInfo(); SQLStatementCountValidator.assertSelectCount(1); SQLStatementCountValidator.assertInsertCount(2); } }