/*
* @(#)ParameterUtilTests.java 2012-8-24 下午3:34:45
*
* Copyright (c) 2011-2012 Makersoft.org all rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
*
*/
package org.makersoft.shards.unit.utils;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.makersoft.shards.ShardId;
import org.makersoft.shards.domain.User;
import org.makersoft.shards.utils.ParameterUtil;
/**
* unit test.
*
* @version 2012-8-24 下午3:34:45
* @author Feng Kuok
*/
public class ParameterUtilTests {
private ShardId shardId;
private static final String PREFIX = "prefix";
private static final String SUFFIX = "suffix";
private static final String PREFIX_VALUE = "prefix_";
private static final String SUFFIX_VALUE = "_suffix";
@Before
public void setup() throws Exception {
shardId = new ShardId(0);
shardId.setPrefix("prefix");
shardId.setSuffix("suffix");
}
@Test
@SuppressWarnings("unchecked")
public void testPrimitiveTypes() throws Exception {
Object[] primitiveParams = new Object[] { "string", 1L, 10, true, 'c' };
for (Object parameter : primitiveParams) {
Object obj = ParameterUtil.resolve(parameter, shardId);
Assert.assertTrue(obj instanceof Map);
Map<String, Object> map = (Map<String, Object>) obj;
Assert.assertEquals(PREFIX_VALUE, map.get(PREFIX));
Assert.assertEquals(SUFFIX_VALUE, map.get(SUFFIX));
// 对于任何key，调用map的containKey方法均返回true
Assert.assertTrue(map.containsKey("any_key"));
// 对于任何非prefix，suffix 的 key均返回原来传递的参数
Assert.assertEquals(parameter, map.get("any_key"));
}
}
@Test
@SuppressWarnings("unchecked")
public void testMap() throws Exception {
Map<String, Object> parameter = new HashMap<String, Object>();
parameter.put("id", 1);
parameter.put("name", "makersoft");
Object obj = ParameterUtil.resolve(parameter, shardId);
Assert.assertTrue(obj instanceof Map);
Map<String, Object> map = (Map<String, Object>) obj;
Assert.assertEquals(PREFIX_VALUE, map.get(PREFIX));
Assert.assertEquals(SUFFIX_VALUE, map.get(SUFFIX));
Assert.assertEquals(parameter.get("id"), map.get("id"));
Assert.assertEquals(parameter.get("name"), map.get("name"));
}
@Test
@SuppressWarnings("unchecked")
public void testList() throws Exception {
List<String> parameter = new ArrayList<String>();
parameter.add("s1");
parameter.add("s2");
Object obj = ParameterUtil.resolve(parameter, shardId);
Assert.assertTrue(obj instanceof Map);
Map<String, Object> map = (Map<String, Object>) obj;
Assert.assertEquals(PREFIX_VALUE, map.get(PREFIX));
Assert.assertEquals(SUFFIX_VALUE, map.get(SUFFIX));
Assert.assertEquals(parameter, map.get("list"));
}
@Test
@SuppressWarnings("unchecked")
public void testArray() throws Exception {
String[] parameters = new String[] { "a1", "a2" };
Object obj = ParameterUtil.resolve(parameters, shardId);
Assert.assertTrue(obj instanceof Map);
Map<String, Object> map = (Map<String, Object>) obj;
Assert.assertEquals(PREFIX_VALUE, map.get(PREFIX));
Assert.assertEquals(SUFFIX_VALUE, map.get(SUFFIX));
Assert.assertEquals(parameters, map.get("array"));
}
@Test
@SuppressWarnings("unchecked")
public void testObject() throws Exception {
User user = new User();
user.setId("id");
user.setUsername("makersoft");
user.setPassword("password");
user.setAge(20);
Object obj = ParameterUtil.resolve(user, shardId);
Assert.assertTrue(obj instanceof Map);
Map<String, Object> map = (Map<String, Object>) obj;
Assert.assertEquals(PREFIX_VALUE, map.get(PREFIX));
Assert.assertEquals(SUFFIX_VALUE, map.get(SUFFIX));
Assert.assertEquals(user.getId(), map.get("id"));
Assert.assertEquals(user.getUsername(), map.get("username"));
Assert.assertEquals(user.getPassword(), map.get("password"));
Assert.assertEquals(user.getAge(), map.get("age"));
}
@Test
public void testNull() throws Exception {
Object obj = ParameterUtil.resolve(null, shardId);
Assert.assertNull(obj);
}
@Test
public void testExtractPrimaryKey() throws Exception {
final String id = "id";
User user = new User();
user.setId(id);
Serializable privaryKey = ParameterUtil.extractPrimaryKey(user);
Assert.assertEquals(privaryKey, id);
}
@Test
public void testGeneratePrimaryKey() throws Exception {
final String id = "id";
User user = (User) ParameterUtil.generatePrimaryKey(new User(), id);
Assert.assertEquals(id, user.getId());
}
}