6 changed files with 241 additions and 3 deletions
@ -0,0 +1,79 @@
|
||||
package com.fanruan.api.decision; |
||||
|
||||
import com.fr.decision.authority.AuthorityContext; |
||||
import com.fr.decision.authority.controller.ConnectionAuthorityController; |
||||
import com.fr.decision.authority.controller.CustomRoleController; |
||||
import com.fr.decision.authority.controller.DefaultAuthorityController; |
||||
import com.fr.decision.authority.controller.DepartmentController; |
||||
import com.fr.decision.authority.controller.PostController; |
||||
import com.fr.decision.authority.controller.UserController; |
||||
import com.fr.decision.authorize.AuthorizeAttr; |
||||
import com.fr.decision.config.FSConfig; |
||||
import org.easymock.EasyMock; |
||||
import org.junit.Assert; |
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.powermock.api.easymock.PowerMock; |
||||
import org.powermock.core.classloader.annotations.PrepareForTest; |
||||
import org.powermock.modules.junit4.PowerMockRunner; |
||||
|
||||
/** |
||||
* @author lidongy |
||||
* @version 10.0 |
||||
* Created by lidongy on 2020/6/29 |
||||
*/ |
||||
@RunWith(PowerMockRunner.class) |
||||
@PrepareForTest({FSConfig.class, AuthorityContext.class}) |
||||
public class AuthorityKitTest { |
||||
|
||||
@Test |
||||
public void testIsAuthorizeAttrOpen() { |
||||
AuthorizeAttr authorizeAttr = EasyMock.createMock(AuthorizeAttr.class); |
||||
EasyMock.expect(authorizeAttr.isGradeAuthority()).andReturn(true).anyTimes(); |
||||
EasyMock.expect(authorizeAttr.isDataConnectionAuthority()).andReturn(true).anyTimes(); |
||||
EasyMock.expect(authorizeAttr.isEditReportAuthority()).andReturn(true).anyTimes(); |
||||
EasyMock.replay(authorizeAttr); |
||||
|
||||
FSConfig fsConfig = EasyMock.createMock(FSConfig.class); |
||||
EasyMock.expect(fsConfig.getAuthorizeAttr()).andReturn(authorizeAttr).anyTimes(); |
||||
EasyMock.replay(fsConfig); |
||||
|
||||
PowerMock.mockStatic(FSConfig.class); |
||||
EasyMock.expect(FSConfig.getInstance()).andReturn(fsConfig).anyTimes(); |
||||
PowerMock.replay(FSConfig.class); |
||||
|
||||
Assert.assertEquals(AuthorityKit.isGradeAuthority(), FSConfig.getInstance().getAuthorizeAttr().isGradeAuthority()); |
||||
Assert.assertEquals(AuthorityKit.isDataConnectionAuthority(), FSConfig.getInstance().getAuthorizeAttr().isDataConnectionAuthority()); |
||||
Assert.assertEquals(AuthorityKit.isEditReportAuthority(), FSConfig.getInstance().getAuthorizeAttr().isEditReportAuthority()); |
||||
} |
||||
|
||||
@Test |
||||
public void testGetController() throws Exception { |
||||
DefaultAuthorityController defaultAuthorityController = EasyMock.createMock(DefaultAuthorityController.class); |
||||
CustomRoleController customRoleController = EasyMock.createMock(CustomRoleController.class); |
||||
DepartmentController departmentController = EasyMock.createMock(DepartmentController.class); |
||||
PostController postController = EasyMock.createMock(PostController.class); |
||||
UserController userController = EasyMock.createMock(UserController.class); |
||||
ConnectionAuthorityController connectionAuthorityController = EasyMock.createMock(ConnectionAuthorityController.class); |
||||
|
||||
AuthorityContext authorityContext = EasyMock.createMock(AuthorityContext.class); |
||||
EasyMock.expect(authorityContext.getAuthorityController()).andReturn(defaultAuthorityController).anyTimes(); |
||||
EasyMock.expect(authorityContext.getCustomRoleController()).andReturn(customRoleController).anyTimes(); |
||||
EasyMock.expect(authorityContext.getDepartmentController()).andReturn(departmentController).anyTimes(); |
||||
EasyMock.expect(authorityContext.getPostController()).andReturn(postController).anyTimes(); |
||||
EasyMock.expect(authorityContext.getUserController()).andReturn(userController).anyTimes(); |
||||
EasyMock.expect(authorityContext.getAuthorityController(ConnectionAuthorityController.class)).andReturn(connectionAuthorityController).anyTimes(); |
||||
EasyMock.replay(authorityContext); |
||||
|
||||
PowerMock.mockStatic(AuthorityContext.class); |
||||
EasyMock.expect(AuthorityContext.getInstance()).andReturn(authorityContext).anyTimes(); |
||||
PowerMock.replay(AuthorityContext.class); |
||||
|
||||
Assert.assertEquals(AuthorityKit.getAuthorityController(), defaultAuthorityController); |
||||
Assert.assertEquals(AuthorityKit.getCustomRoleController(), customRoleController); |
||||
Assert.assertEquals(AuthorityKit.getDepartmentController(), departmentController); |
||||
Assert.assertEquals(AuthorityKit.getPostController(), postController); |
||||
Assert.assertEquals(AuthorityKit.getUserController(), userController); |
||||
Assert.assertEquals(AuthorityKit.getAuthorityController(ConnectionAuthorityController.class), connectionAuthorityController); |
||||
} |
||||
} |
@ -0,0 +1,39 @@
|
||||
package com.fanruan.api.decision.user; |
||||
|
||||
import com.fr.decision.webservice.v10.user.UserService; |
||||
import org.easymock.EasyMock; |
||||
import org.junit.Assert; |
||||
import org.junit.Test; |
||||
import org.junit.runner.RunWith; |
||||
import org.powermock.api.easymock.PowerMock; |
||||
import org.powermock.core.classloader.annotations.PrepareForTest; |
||||
import org.powermock.modules.junit4.PowerMockRunner; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.List; |
||||
|
||||
/** |
||||
* @author lidongy |
||||
* @version 10.0 |
||||
* Created by lidongy on 2020/6/29 |
||||
*/ |
||||
@RunWith(PowerMockRunner.class) |
||||
@PrepareForTest({UserService.class}) |
||||
public class UserKitTest { |
||||
|
||||
@Test |
||||
public void testGetAdminUserIdList() throws Exception { |
||||
List<String> adminUserIdList = new ArrayList<>(); |
||||
adminUserIdList.add("1"); |
||||
|
||||
UserService userService = EasyMock.createMock(UserService.class); |
||||
EasyMock.expect(userService.getAdminUserIdList()).andReturn(adminUserIdList); |
||||
EasyMock.replay(userService); |
||||
|
||||
PowerMock.mockStatic(UserService.class); |
||||
EasyMock.expect(UserService.getInstance()).andReturn(userService); |
||||
PowerMock.replay(UserService.class); |
||||
|
||||
Assert.assertEquals(UserKit.getAdminUserIdList(), adminUserIdList); |
||||
} |
||||
} |
Loading…
Reference in new issue