Spring安全测试@WithMockUser不工作

Question

我正在为我的应用程序编写单元测试，我在服务内部执行的步骤之一是从SpringSecurityContext获取当前经过身份验证的用户。

我知道，如果我想模拟Spring Security Authentication，我可以使用@SusMockUser，但它对我不起作用，因为当测试方法到达getAuthentication（）方法调用时，它总是返回null...

我已经搜索了很多问题和博客帖子，但没有一个有解决方案。我用@ExtendWith（MockitoExtension.class）注释我的测试类

我希望避免编写4行代码来模拟单个方法调用

//mock身份验证
//mock主体
//当SpringContextHolder.getContext（）. getAuthentication-

编辑：

测试类

@ExtendWith(MockitoExtension.class)
public class SegmetnServiceTest {
    
    @InjectMocks
    private SegmentService service;


    @Test
    void testWithMockUser() {
            //given
            UpdateSegmentReq request = new UpdateSegmentReq();
            String name = "TEST"
            request.setName(name)

            //when
            SegmentNode segment = service.updateSegment(request);

            //then
            assertEquals(segment.getName(), name)
    }
}

服务等级

public class SegmentService {

    private SegmentRepository repository;

    SegmentNode updateSegment(String code){
        SegmentNode segment = repository.findByCode(code);

        String userId = (String) SecurityContextHolder.getContext().getAuthentication().getPrincipal();

        segment.updatedBy(userId);


        return segment;
    }
}

问题是，即使我用@SusMockUser注释我的Test方法，当它到达服务方法getAuthentication为空并且getMain抛出NPE。

Answer 1

您应该使用@ContextConfiguration将Spring Security测试配置加载到Application Context中（假设您还具有spring-test作为依赖项）请参阅澄清。

您可以使用注释SpringJUnitConfig来组合所需的注释--@ExtendWith（SpringExtension.class）和@ContextConfiguration--在测试中设置spring测试。