package com.hxh.spring.test;
|
|
import com.yeshi.buwan.dao.user.LoginUserDao;
|
import com.yeshi.buwan.dao.user.LoginUserExtraDao;
|
import com.yeshi.buwan.domain.user.LoginUser;
|
import com.yeshi.buwan.domain.user.LoginUserExtra;
|
import com.yeshi.buwan.util.StringUtil;
|
import org.junit.Test;
|
import org.junit.runner.RunWith;
|
import org.springframework.data.mongodb.core.query.Query;
|
import org.springframework.test.context.ContextConfiguration;
|
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
|
import org.springframework.test.context.web.WebAppConfiguration;
|
|
import javax.annotation.Resource;
|
import java.util.List;
|
|
@RunWith(SpringJUnit4ClassRunner.class)
|
@ContextConfiguration(locations = {"classpath:spring.xml"})
|
@WebAppConfiguration
|
public class UserTest {
|
|
@Resource
|
private LoginUserExtraDao loginUserExtraDao;
|
|
@Resource
|
private LoginUserDao loginUserDao;
|
|
@Test
|
public void test() {
|
Query query = new Query();
|
List<LoginUserExtra> extraList = loginUserExtraDao.findList(query);
|
for (LoginUserExtra extra : extraList) {
|
LoginUser user = loginUserDao.find(LoginUser.class, extra.getId());
|
if (user != null && StringUtil.isNullOrEmpty(user.getPortrait())) {
|
if (!StringUtil.isNullOrEmpty(extra.getWxPortrait())) {
|
user.setPortrait(extra.getWxPortrait());
|
loginUserDao.update(user);
|
} else if (!StringUtil.isNullOrEmpty(extra.getQqPortrait())) {
|
user.setPortrait(extra.getQqPortrait());
|
loginUserDao.update(user);
|
}
|
}
|
}
|
}
|
|
|
}
|
