在Spring Junit测试中使用不同的Auth模型

Question

我使用Spring和Junit4来测试我的服务器应用程序。我的应用程序使用Spring Security性，一些查询利用SpEL语法检查记录的用户。

因为我正在测试一些需要加载数据的场景，所以我遇到了一些问题，因为我不适合这样做。

例如：

@Test
    @WithUserDetails(userDetailsServiceBeanName = "mockUserDetailService", value = "customer")
    public void saveEcommerceTicket() {
        ZonedDateTime entryDate = ZonedDateTime.now().plusDays(1);
        createStandardFare();
        createDailyCode(entryDate.toInstant());
        createPaymentWithWallet();
        Customer customer = createCustomerWithWallet(new BigDecimal(100));

        Ticket ticket = new Ticket();
        ticket.setLicensePlate1("AA123BB");
        ticket.setArea(areaCentroStorico);
        ticket.setDailyCode("CC");
        ticket.setEntryDate(entryDate.toInstant());
        ticket.setExitDate(entryDate.plusDays(1).toInstant());
        ticket.setPassengers(50);
        ticket.setPassengersCountry(italy);
        ticket.setCustomer(customer);

        ticket = ticketService.saveFromEcommerceWithWalletPayment(ticket, new ArrayList<Media>());

        assertEquals(true, ticket.isStandard());
        assertEquals(TicketStatus.VALID, ticket.getStatus());
        assertTrue(new BigDecimal(100).compareTo(ticketRepository.sumPayments(ticket.getId())) == 0);
        assertTrue(BigDecimal.ZERO.compareTo(customer.getWallet().getBalance()) == 0);
        assertEquals(true, ticket.isPaid());
    }

这是一个使用自定义UserDetailService的测试示例。不幸的是，有些操作，如:createStandardFare()、createDailyCode()等，需要不同的角色。我试图手动设置身份验证，但最终总是让整个html" target="_blank">测试由一个角色完成。相反，我需要一些部分与特定角色一起执行（比方说ADMIN），而另一些部分与另一个角色一起执行（比方说CUSTOMER）。

有什么方法可以用Spring和JUnit来实现这一点吗？

Answer 1

如果您在单个测试方法中调用不同的方法，而这些方法需要不同的角色，那么实际上只有两个选项。

• 配置一个模拟用户（例如，通过@WithMockUser或类似的方式），为其分配测试方法中使用的所有角色。
• 在测试方法本身内以编程方式切换当前模拟用户。
  • 例如，使用以下代码段。

  @Test
  public void test() {
      login("customer", "password", "ROLE_CUSTOMER");
      // ...
      login("admin", "password", "ROLE_ADMIN");
      // ...
  }
  
  static void login(String username, String password, String... roles) {
      SecurityContext context = SecurityContextHolder.createEmptyContext();
  
      User user = new User(username, password, roles(roles));
  
      Authentication auth = new UsernamePasswordAuthenticationToken(user, password, user.getAuthorities());
      context.setAuthentication(auth);
      SecurityContextHolder.setContext(context);
  }
  
  static List<SimpleGrantedAuthority> roles(String... roles) {
      return Arrays.stream(roles).map(SimpleGrantedAuthority::new).collect(toList());
  }
  

  注意:user在上面的示例中是org.springframework.security.core.userdetails.user。