Today, while writing\u00a0unit test case for one of the Java method which looks like below:<\/p>\n
public ApplicationUser getApplicationUser() {\r\n\tApplicationUser applicationUser = (ApplicationUser) SecurityContextHolder.getContext().getAuthentication().getPrincipal();\r\n\treturn applicationUser;\r\n}<\/pre>\nI want to mock Spring Security Context to get the Principal, to achieve the same I mocked each level of method calls as follows:<\/p>\n
import org.junit.Before;\r\nimport org.junit.Rule;\r\nimport org.junit.Test;\r\nimport static org.mockito.Mockito.when;\r\nimport static org.mockito.Mockito.mock;\r\nimport org.mockito.MockitoAnnotations;\r\nimport org.springframework.security.core.Authentication;\r\nimport org.springframework.security.core.context.SecurityContext;\r\nimport org.springframework.security.core.context.SecurityContextHolder;\r\nimport com.arpit.security.user.ApplicationUser;\r\n\r\npublic class BaseTest {\r\n\r\n\t@Before\r\n\tpublic void setupMock() {\r\n\t\tMockitoAnnotations.initMocks(this);\r\n\t}\r\n\r\n\t@Test\r\n\tpublic void mockApplicationUser() {\r\n\t\tApplicationUser applicationUser = mock(ApplicationUser.class);\r\n\t\tAuthentication authentication = mock(Authentication.class);\r\n\t\tSecurityContext securityContext = mock(SecurityContext.class);\r\n\t\twhen(securityContext.getAuthentication()).thenReturn(authentication);\r\n\t\tSecurityContextHolder.setContext(securityContext);\r\n\t\twhen(SecurityContextHolder.getContext().getAuthentication().getPrincipal()).thenReturn(applicationUser);\r\n\t}\r\n\r\n}<\/pre>\n\n