/*
* @(#)HorizontalShardsTests.java 2013-6-5 下午23:33:33
*
* Copyright (c) 2011-2013 Makersoft.org all rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
*
*/
package org.makersoft.shards.unit.persistence;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Random;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.makersoft.shards.domain.User;
import org.makersoft.shards.mapper.UserMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ActiveProfiles;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.transaction.TransactionConfiguration;
import org.springframework.transaction.annotation.Transactional;
/**
* unit test for horizontal shards.
*/
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = { "classpath:applicationContext.xml" })
@TransactionConfiguration(transactionManager = "transactionManager", defaultRollback = true)
@ActiveProfiles("test")
public class HorizontalShardsTests {
@Autowired(required = true)
private UserMapper userMapper;
private Random random = new Random();
@Test
@Transactional
public void test_insert() throws Exception {
for (int i = 0; i < 1000; i++) {
String username = "makersoft_" + i;
String password = "makersoft_" + i;
int age = random.nextInt(30);
int gender = (i % 2 == 0) ? User.SEX_MALE : User.SEX_FEMALE;
User user = this.insert_new_user(username, password, age, gender);
Assert.assertNotNull(user.getId());
}
}
@Test
@Transactional
public void test_update() throws Exception{
User user = this.insert_new_user("makersoft", "makersoft", 26, User.SEX_MALE);
user.setUsername("www.makersoft.org");
user.setPassword("www.makersoft.org");
//update entity
int rows = userMapper.udpateUser(user);
Assert.assertEquals(1, rows);
}
@Test
@Transactional
public void test_update_by_id() throws Exception {
User user = this.insert_new_user("makersoft", "makersoft", 26, User.SEX_MALE);
user.setUsername("username");
user.setPassword("password");
int rows = userMapper.updateById(user);
Assert.assertEquals(1, rows);
}
@Test
@Transactional
public void test_get_all_count() throws Exception{
int count = userMapper.getAllCount();
Assert.assertEquals(0, count);
for(int i = 0; i < 10; i++){
String username = "makersoft_" + i;
String password = "makersoft_" + i;
int age = random.nextInt(30);
int gender = (i % 2 == 0) ? User.SEX_MALE : User.SEX_FEMALE;
User user = this.insert_new_user(username, password, age, gender);
Assert.assertNotNull(user.getId());
}
count = userMapper.getAllCount();
Assert.assertEquals(10, count);
}
@Test
@Transactional
public void test_find_all() throws Exception{
Map<String, User> caches = new HashMap<String, User>();
for(int i = 0; i < 10; i++){
String username = "makersoft_" + i;
String password = "makersoft_" + i;
int age = random.nextInt(30);
int gender = (i % 2 == 0) ? User.SEX_MALE : User.SEX_FEMALE;
User user = this.insert_new_user(username, password, age, gender);
Assert.assertNotNull(user.getId());
caches.put(user.getId(), user);
}
List<User> users = userMapper.findAll();
Assert.assertEquals(10, users.size());
for(User entity : users) {
User user = caches.get(entity.getId());
Assert.assertNotNull(user);
//has override hashCode and equals method
Assert.assertEquals(entity, user);
}
}
@Test
@Transactional
public void test_find_by_gender() throws Exception{
for(int i = 0; i < 10; i++){
String username = "makersoft_" + i;
String password = "makersoft_" + i;
int age = random.nextInt(30);
int gender = (i % 2 == 0) ? User.SEX_MALE : User.SEX_FEMALE;
User user = this.insert_new_user(username, password, age, gender);
Assert.assertNotNull(user.getId());
}
List<User> users = userMapper.findByGender(User.SEX_MALE);
Assert.assertEquals(5, users.size());
}
@Test
@Transactional
public void test_get_by_id() throws Exception{
User user = this.insert_new_user("makersoft", "makersoft", 26, User.SEX_MALE);
User result = userMapper.getById(user.getId());
Assert.assertNotNull(result);
Assert.assertEquals(user, result);
}
@Test
@Transactional
public void test_delete_by_id() throws Exception{
User user = this.insert_new_user("makersoft", "makersoft", 26, User.SEX_MALE);
int rows = userMapper.deleteById(user.getId());
Assert.assertEquals(1, rows);
user = userMapper.getById(user.getId());
Assert.assertNull(user);
}
@Test
@Transactional
public void test_delete() throws Exception{
for(int i = 0; i < 10; i++){
String username = "makersoft_" + i;
String password = "makersoft_" + i;
int age = random.nextInt(30);
int gender = (i % 2 == 0) ? User.SEX_MALE : User.SEX_FEMALE;
User user = this.insert_new_user(username, password, age, gender);
Assert.assertNotNull(user.getId());
}
int rows = userMapper.deleteAll();
Assert.assertEquals(10, rows);
int count = userMapper.getAllCount();
Assert.assertEquals(0, count);
}
/**
* for insert new user
*/
private User insert_new_user(String username, String password, int age, int gender){
User user = new User(username, password, gender);
user.setAge(age);
userMapper.insertUser(user);
Assert.assertNotNull(user.getId());
return user;
}
}