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