package cn.flightfeather.supervision.lightshare.repository.impl
|
|
import cn.flightfeather.supervision.domain.entity.BaseInfo
|
import cn.flightfeather.supervision.domain.entity.Userinfo
|
import cn.flightfeather.supervision.domain.enumeration.DistrictType
|
import cn.flightfeather.supervision.domain.mapper.BaseInfoMapper
|
import cn.flightfeather.supervision.domain.mapper.UserinfoMapper
|
import cn.flightfeather.supervision.domain.repository.UserInfoRep
|
import cn.flightfeather.supervision.lightshare.vo.UserSearchCondition
|
import org.junit.Test
|
import org.junit.jupiter.api.extension.ExtendWith
|
import org.junit.runner.RunWith
|
import org.springframework.beans.factory.annotation.Autowired
|
import org.springframework.boot.test.context.SpringBootTest
|
import org.springframework.test.context.junit.jupiter.SpringExtension
|
import org.springframework.test.context.junit4.SpringRunner
|
import tk.mybatis.mapper.entity.Example
|
|
@RunWith(SpringRunner::class)
|
@ExtendWith(SpringExtension::class)
|
@SpringBootTest
|
class UserInfoRepositoryImplTest() {
|
|
@Autowired
|
lateinit var userInfoRep: UserInfoRep
|
|
@Autowired
|
lateinit var userinfoMapper: UserinfoMapper
|
|
@Autowired
|
lateinit var baseInfoMapper: BaseInfoMapper
|
|
@Test
|
fun foo1() {
|
val c = UserSearchCondition().apply {
|
// provinceCode = "310000"
|
// cityCode = "310100"
|
districtName = "静安区"
|
// townCode = "310106019"
|
area
|
mcName
|
userTypeId = 3
|
sceneTypes = listOf("1")
|
}
|
val r = userInfoRep.searchUser(c)
|
println(r)
|
}
|
|
@Test
|
fun foo2() {
|
userinfoMapper.selectByExample(Example(Userinfo::class.java).apply {
|
createCriteria().andEqualTo("usertype", "政府部门")
|
}).forEach {
|
val baseInfo = BaseInfo().apply {
|
biGuid = it?.guid
|
biName = it?.realname
|
biProvinceCode = "310000"
|
biProvinceName = "上海市"
|
biCityCode = "310100"
|
biCityName = "上海市"
|
biDistrictCode = DistrictType.getCode(it?.extension1)
|
biDistrictName = it?.extension1
|
biRemark = it?.usertype
|
}
|
val b = baseInfoMapper.selectOne(BaseInfo().apply { biGuid = it?.guid })
|
if (b == null) {
|
baseInfoMapper.insert(baseInfo)
|
} else {
|
baseInfoMapper.updateByPrimaryKey(baseInfo)
|
}
|
}
|
}
|
}
|
