package cn.iocoder.yudao.module.ums.service.member;
|
|
import cn.iocoder.yudao.framework.test.core.ut.BaseMockitoUnitTest;
|
import cn.iocoder.yudao.module.system.service.auth.WxcpPasswordSetupGrant;
|
import cn.iocoder.yudao.module.system.service.user.AdminUserService;
|
import cn.iocoder.yudao.module.system.dal.dataobject.user.AdminUserDO;
|
import cn.iocoder.yudao.module.ums.dal.dataobject.member.UmsMemberDO;
|
import cn.iocoder.yudao.module.ums.dal.mysql.member.UmsMemberMapper;
|
import com.baomidou.mybatisplus.core.conditions.Wrapper;
|
import com.baomidou.mybatisplus.core.MybatisConfiguration;
|
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
|
import org.apache.ibatis.builder.MapperBuilderAssistant;
|
import org.junit.jupiter.api.Test;
|
import org.mockito.InjectMocks;
|
import org.mockito.Mock;
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.ArgumentMatchers.*;
|
import static org.mockito.Mockito.*;
|
|
class WxcpInitialPasswordServiceTest extends BaseMockitoUnitTest {
|
@InjectMocks private WxcpInitialPasswordService service;
|
@Mock private UmsMemberMapper memberMapper;
|
@Mock private AdminUserService userService;
|
@Mock private WxcpPasswordSetupGrant grant;
|
|
@Test void rejectsSessionWithoutWxcpGrant() {
|
when(grant.valid("token", 1L)).thenReturn(false);
|
assertThrows(RuntimeException.class, () -> service.setPassword(1L, "token", "password1"));
|
verifyNoInteractions(memberMapper, userService);
|
}
|
|
@Test void rejectsAccountWithoutPendingSetup() {
|
when(grant.valid("token", 1L)).thenReturn(true);
|
when(memberMapper.selectByUserId(1L)).thenReturn(new UmsMemberDO().setPasswordSetupRequired(false));
|
assertThrows(RuntimeException.class, () -> service.setPassword(1L, "token", "password1"));
|
verifyNoInteractions(userService);
|
}
|
|
@Test void setsPasswordOnceAndReturnsOwnUsername() {
|
TableInfoHelper.initTableInfo(new MapperBuilderAssistant(new MybatisConfiguration(), ""), UmsMemberDO.class);
|
when(grant.valid("token", 1L)).thenReturn(true);
|
when(memberMapper.selectByUserId(1L)).thenReturn(new UmsMemberDO().setId(2L).setPasswordSetupRequired(true));
|
when(memberMapper.update(isNull(), any(Wrapper.class))).thenReturn(1);
|
when(userService.getUser(1L)).thenReturn(new AdminUserDO().setUsername("wx_example"));
|
assertEquals("wx_example", service.setPassword(1L, "token", "password1"));
|
verify(userService).updateUserPassword(1L, "password1");
|
verify(grant).revoke("token");
|
}
|
}
|
