重构社交登录的时候，增加独立的社交绑定表

2025-10-31 18:28:43 +08:00 · 2022-04-25 01:36:29 +08:00
parent 6e40469735
commit 705a5ff645
12 changed files with 322 additions and 366 deletions
--- a/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/social/SocialUserServiceTest.java
+++ b/yudao-module-system/yudao-module-system-biz/src/test/java/cn/iocoder/yudao/module/system/service/social/SocialUserServiceTest.java
@@ -1,32 +1,29 @@
 package cn.iocoder.yudao.module.system.service.social;

-import cn.iocoder.yudao.module.system.dal.dataobject.social.SocialUserDO;
 import cn.iocoder.yudao.module.system.dal.mysql.social.SocialUserMapper;
-import cn.iocoder.yudao.module.system.dal.redis.social.SocialAuthUserRedisDAO;
-import cn.iocoder.yudao.module.system.enums.social.SocialTypeEnum;
-import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
 import cn.iocoder.yudao.framework.test.core.ut.BaseDbAndRedisUnitTest;
+import com.baomidou.mybatisplus.core.toolkit.PluginUtils;
 import com.xkcoding.justauth.AuthRequestFactory;
-import me.zhyd.oauth.model.AuthUser;
+import me.zhyd.oauth.request.AuthRequest;
+import me.zhyd.oauth.utils.AuthStateUtils;
 import org.junit.jupiter.api.Test;
+import org.mockito.MockedStatic;
 import org.springframework.boot.test.mock.mockito.MockBean;
 import org.springframework.context.annotation.Import;

 import javax.annotation.Resource;
-import java.util.List;

 import static cn.hutool.core.util.RandomUtil.randomEle;
 import static cn.hutool.core.util.RandomUtil.randomString;
-import static cn.iocoder.yudao.framework.common.util.json.JsonUtils.toJsonString;
-import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomLongId;
 import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.randomPojo;
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.mockito.Mockito.*;

-@Import({SocialUserServiceImpl.class, SocialAuthUserRedisDAO.class})
+@Import(SocialUserServiceImpl.class)
 public class SocialUserServiceTest extends BaseDbAndRedisUnitTest {

    @Resource
-    private SocialUserServiceImpl socialService;
+    private SocialUserServiceImpl socialUserService;

    @Resource
    private SocialUserMapper socialUserMapper;
@@ -34,132 +31,152 @@ public class SocialUserServiceTest extends BaseDbAndRedisUnitTest {
    @MockBean
    private AuthRequestFactory authRequestFactory;

-    /**
-     * 情况一，创建 SocialUserDO 的情况
-     */
    @Test
-    public void testBindSocialUser_create() {
-        // mock 数据
-        // 准备参数
-        Long userId = randomLongId();
-        Integer type = randomEle(SocialTypeEnum.values()).getType();
-        AuthUser authUser = randomPojo(AuthUser.class);
-        // mock 方法
+    public void testGetAuthorizeUrl() {
+        try (MockedStatic<AuthStateUtils> authStateUtilsMock = mockStatic(AuthStateUtils.class)) {
+            // 准备参数
+            Integer type = 31;
+            String redirectUri = "sss";
+            // mock 获得对应的 AuthRequest 实现
+            AuthRequest authRequest = mock(AuthRequest.class);
+            when(authRequestFactory.get(eq("WECHAT_MP"))).thenReturn(authRequest);
+            // mock 方法
+            authStateUtilsMock.when(AuthStateUtils::createState).thenReturn("aoteman");
+            when(authRequest.authorize(eq("aoteman"))).thenReturn("https://www.iocoder.cn?redirect_uri=yyy");

-        // 调用
-        socialService.bindSocialUser(userId, UserTypeEnum.ADMIN.getValue(), type, authUser);
-        // 断言
-        List<SocialUserDO> socialUsers = socialUserMapper.selectList("user_id", userId);
-        assertEquals(1, socialUsers.size());
-        assertBindSocialUser(socialUsers.get(0), authUser, userId, type);
+            // 调用
+            String url = socialUserService.getAuthorizeUrl(type, redirectUri);
+            // 断言
+            assertEquals("https://www.iocoder.cn/?redirect_uri=sss", url);
+        }
    }

-    /**
-     * 情况二，更新 SocialUserDO 的情况
-     */
-    @Test
-    public void testBindSocialUser_update() {
-        // mock 数据
-        SocialUserDO dbSocialUser = randomPojo(SocialUserDO.class, socialUserDO -> {
-            socialUserDO.setUserType(UserTypeEnum.ADMIN.getValue());
-            socialUserDO.setType(randomEle(SocialTypeEnum.values()).getType());
-        });
-        socialUserMapper.insert(dbSocialUser);
-        // 准备参数
-        Long userId = dbSocialUser.getUserId();
-        Integer type = dbSocialUser.getType();
-        AuthUser authUser = randomPojo(AuthUser.class);
-        // mock 方法
-
-        // 调用
-        socialService.bindSocialUser(userId, UserTypeEnum.ADMIN.getValue(), type, authUser);
-        // 断言
-        List<SocialUserDO> socialUsers = socialUserMapper.selectList("user_id", userId);
-        assertEquals(1, socialUsers.size());
-        assertBindSocialUser(socialUsers.get(0), authUser, userId, type);
-    }
-
-    /**
-     * 情况一和二都存在的，逻辑二的场景
-     */
-    @Test
-    public void testBindSocialUser_userId() {
-        // mock 数据
-        SocialUserDO dbSocialUser = randomPojo(SocialUserDO.class, socialUserDO -> {
-            socialUserDO.setUserType(UserTypeEnum.ADMIN.getValue());
-            socialUserDO.setType(randomEle(SocialTypeEnum.values()).getType());
-        });
-        socialUserMapper.insert(dbSocialUser);
-        // 准备参数
-        Long userId = randomLongId();
-        Integer type = dbSocialUser.getType();
-        AuthUser authUser = randomPojo(AuthUser.class);
-        // mock 方法
-
-        // 调用
-        socialService.bindSocialUser(userId, UserTypeEnum.ADMIN.getValue(), type, authUser);
-        // 断言
-        List<SocialUserDO> socialUsers = socialUserMapper.selectList("user_id", userId);
-        assertEquals(1, socialUsers.size());
-    }
-
-    private void assertBindSocialUser(SocialUserDO socialUser, AuthUser authUser, Long userId,
-                                      Integer type) {
-        assertEquals(authUser.getToken().getAccessToken(), socialUser.getToken());
-        assertEquals(toJsonString(authUser.getToken()), socialUser.getRawTokenInfo());
-        assertEquals(authUser.getNickname(), socialUser.getNickname());
-        assertEquals(authUser.getAvatar(), socialUser.getAvatar());
-        assertEquals(toJsonString(authUser.getRawUserInfo()), socialUser.getRawUserInfo());
-        assertEquals(userId, socialUser.getUserId());
-        assertEquals(UserTypeEnum.ADMIN.getValue(), socialUser.getUserType());
-        assertEquals(type, socialUser.getType());
-        assertEquals(authUser.getUuid(), socialUser.getOpenid());
-        assertEquals(socialService.getAuthUserUnionId(authUser), socialUser.getUnionId());
-    }
-
-    /**
-     * 情况一，如果新老的 unionId 是一致的，无需解绑
-     */
-    @Test
-    public void testUnbindOldSocialUser_no() {
-        // mock 数据
-        SocialUserDO oldSocialUser = randomPojo(SocialUserDO.class, socialUserDO -> {
-            socialUserDO.setUserType(UserTypeEnum.ADMIN.getValue());
-            socialUserDO.setType(randomEle(SocialTypeEnum.values()).getType());
-        });
-        socialUserMapper.insert(oldSocialUser);
-        // 准备参数
-        Long userId = oldSocialUser.getUserId();
-        Integer type = oldSocialUser.getType();
-        String newUnionId = oldSocialUser.getUnionId();
-
-        // 调用
-        socialService.unbindOldSocialUser(userId, UserTypeEnum.ADMIN.getValue(), type, newUnionId);
-        // 断言
-        assertEquals(1L, socialUserMapper.selectCount(null).longValue());
-    }
-
-
-    /**
-     * 情况二，如果新老的 unionId 不一致的，需解绑
-     */
-    @Test
-    public void testUnbindOldSocialUser_yes() {
-        // mock 数据
-        SocialUserDO oldSocialUser = randomPojo(SocialUserDO.class, socialUserDO -> {
-            socialUserDO.setUserType(UserTypeEnum.ADMIN.getValue());
-            socialUserDO.setType(randomEle(SocialTypeEnum.values()).getType());
-        });
-        socialUserMapper.insert(oldSocialUser);
-        // 准备参数
-        Long userId = oldSocialUser.getUserId();
-        Integer type = oldSocialUser.getType();
-        String newUnionId = randomString(10);
-
-        // 调用
-        socialService.unbindOldSocialUser(userId, UserTypeEnum.ADMIN.getValue(), type, newUnionId);
-        // 断言
-        assertEquals(0L, socialUserMapper.selectCount(null).longValue());
-    }
+//    /**
+//     * 情况一，创建 SocialUserDO 的情况
+//     */
+//    @Test
+//    public void testBindSocialUser_create() {
+//        // mock 数据
+//        // 准备参数
+//        Long userId = randomLongId();
+//        Integer type = randomEle(SocialTypeEnum.values()).getType();
+//        AuthUser authUser = randomPojo(AuthUser.class);
+//        // mock 方法
+//
+//        // 调用
+//        socialService.bindSocialUser(userId, UserTypeEnum.ADMIN.getValue(), type, authUser);
+//        // 断言
+//        List<SocialUserDO> socialUsers = socialUserMapper.selectList("user_id", userId);
+//        assertEquals(1, socialUsers.size());
+//        assertBindSocialUser(socialUsers.get(0), authUser, userId, type);
+//    }
+//
+//    /**
+//     * 情况二，更新 SocialUserDO 的情况
+//     */
+//    @Test
+//    public void testBindSocialUser_update() {
+//        // mock 数据
+//        SocialUserDO dbSocialUser = randomPojo(SocialUserDO.class, socialUserDO -> {
+//            socialUserDO.setUserType(UserTypeEnum.ADMIN.getValue());
+//            socialUserDO.setType(randomEle(SocialTypeEnum.values()).getType());
+//        });
+//        socialUserMapper.insert(dbSocialUser);
+//        // 准备参数
+//        Long userId = dbSocialUser.getUserId();
+//        Integer type = dbSocialUser.getType();
+//        AuthUser authUser = randomPojo(AuthUser.class);
+//        // mock 方法
+//
+//        // 调用
+//        socialService.bindSocialUser(userId, UserTypeEnum.ADMIN.getValue(), type, authUser);
+//        // 断言
+//        List<SocialUserDO> socialUsers = socialUserMapper.selectList("user_id", userId);
+//        assertEquals(1, socialUsers.size());
+//        assertBindSocialUser(socialUsers.get(0), authUser, userId, type);
+//    }
+//
+//    /**
+//     * 情况一和二都存在的，逻辑二的场景
+//     */
+//    @Test
+//    public void testBindSocialUser_userId() {
+//        // mock 数据
+//        SocialUserDO dbSocialUser = randomPojo(SocialUserDO.class, socialUserDO -> {
+//            socialUserDO.setUserType(UserTypeEnum.ADMIN.getValue());
+//            socialUserDO.setType(randomEle(SocialTypeEnum.values()).getType());
+//        });
+//        socialUserMapper.insert(dbSocialUser);
+//        // 准备参数
+//        Long userId = randomLongId();
+//        Integer type = dbSocialUser.getType();
+//        AuthUser authUser = randomPojo(AuthUser.class);
+//        // mock 方法
+//
+//        // 调用
+//        socialService.bindSocialUser(userId, UserTypeEnum.ADMIN.getValue(), type, authUser);
+//        // 断言
+//        List<SocialUserDO> socialUsers = socialUserMapper.selectList("user_id", userId);
+//        assertEquals(1, socialUsers.size());
+//    }
+//
+//    private void assertBindSocialUser(SocialUserDO socialUser, AuthUser authUser, Long userId,
+//                                      Integer type) {
+//        assertEquals(authUser.getToken().getAccessToken(), socialUser.getToken());
+//        assertEquals(toJsonString(authUser.getToken()), socialUser.getRawTokenInfo());
+//        assertEquals(authUser.getNickname(), socialUser.getNickname());
+//        assertEquals(authUser.getAvatar(), socialUser.getAvatar());
+//        assertEquals(toJsonString(authUser.getRawUserInfo()), socialUser.getRawUserInfo());
+//        assertEquals(userId, socialUser.getUserId());
+//        assertEquals(UserTypeEnum.ADMIN.getValue(), socialUser.getUserType());
+//        assertEquals(type, socialUser.getType());
+//        assertEquals(authUser.getUuid(), socialUser.getOpenid());
+//        assertEquals(socialService.getAuthUserUnionId(authUser), socialUser.getUnionId());
+//    }
+//
+//    /**
+//     * 情况一，如果新老的 unionId 是一致的，无需解绑
+//     */
+//    @Test
+//    public void testUnbindOldSocialUser_no() {
+//        // mock 数据
+//        SocialUserDO oldSocialUser = randomPojo(SocialUserDO.class, socialUserDO -> {
+//            socialUserDO.setUserType(UserTypeEnum.ADMIN.getValue());
+//            socialUserDO.setType(randomEle(SocialTypeEnum.values()).getType());
+//        });
+//        socialUserMapper.insert(oldSocialUser);
+//        // 准备参数
+//        Long userId = oldSocialUser.getUserId();
+//        Integer type = oldSocialUser.getType();
+//        String newUnionId = oldSocialUser.getUnionId();
+//
+//        // 调用
+//        socialService.unbindOldSocialUser(userId, UserTypeEnum.ADMIN.getValue(), type, newUnionId);
+//        // 断言
+//        assertEquals(1L, socialUserMapper.selectCount(null).longValue());
+//    }
+//
+//
+//    /**
+//     * 情况二，如果新老的 unionId 不一致的，需解绑
+//     */
+//    @Test
+//    public void testUnbindOldSocialUser_yes() {
+//        // mock 数据
+//        SocialUserDO oldSocialUser = randomPojo(SocialUserDO.class, socialUserDO -> {
+//            socialUserDO.setUserType(UserTypeEnum.ADMIN.getValue());
+//            socialUserDO.setType(randomEle(SocialTypeEnum.values()).getType());
+//        });
+//        socialUserMapper.insert(oldSocialUser);
+//        // 准备参数
+//        Long userId = oldSocialUser.getUserId();
+//        Integer type = oldSocialUser.getType();
+//        String newUnionId = randomString(10);
+//
+//        // 调用
+//        socialService.unbindOldSocialUser(userId, UserTypeEnum.ADMIN.getValue(), type, newUnionId);
+//        // 断言
+//        assertEquals(0L, socialUserMapper.selectCount(null).longValue());
+//    }

 }