@Override
public CompanyResponse getCompanyByUserId(String userId) {
    Optional<User> optionalUser = userRepository.findUserByUserId(userId);
    if (optionalUser.isEmpty()) {
        throw new CompanyManagerException(USER_NOT_FOUND);
    }
    User user = optionalUser.get();
    Optional<Company> optionalCompany = companyRepository.findCompanyByCompanyName(user.getCompanyName());
    if (optionalCompany.isEmpty()) {
        throw new CompanyManagerException(COMPANY_NOT_FOUND);
    }
}

@Test
public void testGetCompany() {
    String companyId = "companyId";
    when(companyService.getCompanyByUserId(companyId)).thenReturn(new CompanyResponse());
    ResponseEntity<CompanyResponse> response = companyController.getCompany(companyId);
    assertThat(response.getStatusCode()).isEqualTo(HttpStatus.OK);
    assertThat(response.getBody()).isInstanceOf(CompanyResponse.class);
}

when(userRepository.findUserByUserId(userId)).thenReturn(Optional.empty());
```"
<details>
<summary>英文:</summary>
you need to mock your UserRepository and return empty when findUserByUserId is called if you don&#39;t want to manually throw it with previous method
    when(userRepository.findUserByUserId(userId)).thenReturn(Optional.empty());
</details>
# 答案2
**得分**: 1
你需要在你的测试类中模拟`companyRepository`和`userRepository`。
我假设你的`getCompanyByUserId`方法在`CompanyService`类中。
```java
@Mock
private CompanyRepository companyRepository;
@Mock
private UserRepository userRepository;
@InjectMocks
private CompanyService service;
@Test
public void testGetCompanyByUserIdException() {
    Mockito.when(userRepository.findUserByUserId(Mockito.anyString())).thenReturn("");
    Mockito.when(companyRepository.findCompanyByCompanyName(Mockito.anyString())).thenReturn(null);
    try {
        service.getCompanyByUserId("");
    } catch(CompanyManagerException ex) {
    // 在这里断言 COMPANY_NOT_FOUND
    }
}

@Test(expected = CompanyManagerException.class)
public void testGetCompanyShoukdThrowException(){
String userId = "UserId";
when(companyService.getCompanyByUserId(userId)).thenThrow(new CompanyManagerException("USER_NOT_FOUND"));
//实际调用和断言语句将在这里进行
}