dev add test find by searchapo

2026-02-05 12:44:27 +07:00
parent 4395d276ed
commit 6643070999
2 changed files with 9 additions and 2 deletions
@@ -95,7 +95,14 @@ public class UserController {
                            fileStatsService.filesCompanyCount(user.getId()):
                            fileStatsService.filesUserCount(user.getId());

-                    TariffInfo personalTariffInfo = user.getPersonalTariffInfo();
+                    TariffInfo personalTariffInfo;
+
+                    if (user.getCompany() == null) {
+                        personalTariffInfo = user.getPersonalTariffInfo();
+                    } else {
+                        personalTariffInfo = user.getCompany().getTariffInfo();
+                    }
+
                    Tariff tariff = personalTariffInfo.getTariff();

                    TariffInfoDTO infoDTO = TariffInfoDTO.builder().tokens(personalTariffInfo.getTokens())