Add unit test for mapping procedure references to tenders
continuous-integration/drone/push Build is passing

- Introduced a new test function `TestMapProcedureReferencesToTenders` to validate the mapping of procedure references to corresponding tenders.
- Enhanced the `GetByProcedureReferences` method in the tender repository to utilize a more efficient filtering mechanism based on unique contract folder IDs.
- Updated pagination logic to dynamically adjust limits based on the number of unique folders and references, ensuring optimal data retrieval.

This update improves the testing coverage for tender mapping functionality and optimizes the repository's data fetching strategy, enhancing overall performance and reliability.
This commit is contained in:
Mazyar
2026-07-04 13:39:14 +03:30
parent 467090e5d2
commit 4c48e0bb3b
3 changed files with 109 additions and 42 deletions
+43 -20
View File
@@ -17,6 +17,8 @@ import (
"tm/pkg/logger"
orm "tm/pkg/mongo"
"tm/pkg/response"
"golang.org/x/sync/errgroup"
)
// AISummarizerClient defines the interface for on-demand AI operations.
@@ -1754,29 +1756,50 @@ func (s *tenderService) resolveRecommendedTenders(ctx context.Context, recommend
})
}
var (
byProcedure = make(map[string]Tender)
byMongoID = make(map[string]Tender)
)
group, groupCtx := errgroup.WithContext(ctx)
if len(procRefs) > 0 {
byProcedure, err := s.repository.GetByProcedureReferences(ctx, procRefs)
if err != nil {
return nil, fmt.Errorf("failed to load recommended tenders by procedure reference: %w", err)
}
for ref, tender := range byProcedure {
out[ref] = tender
}
}
if len(mongoIDs) == 0 {
return out, nil
}
tenders, err := s.repository.GetByIDs(ctx, mongoIDs)
if err != nil {
s.logger.Error("Failed to load recommended tenders by Mongo ID", map[string]interface{}{
"error": err.Error(),
group.Go(func() error {
byProcedureResult, err := s.repository.GetByProcedureReferences(groupCtx, procRefs)
if err != nil {
return fmt.Errorf("failed to load recommended tenders by procedure reference: %w", err)
}
for ref, tender := range byProcedureResult {
byProcedure[ref] = tender
}
return nil
})
return nil, fmt.Errorf("failed to load recommended tenders: %w", err)
}
for i := range tenders {
out[tenders[i].GetID()] = tenders[i]
if len(mongoIDs) > 0 {
group.Go(func() error {
tenders, err := s.repository.GetByIDs(groupCtx, mongoIDs)
if err != nil {
s.logger.Error("Failed to load recommended tenders by Mongo ID", map[string]interface{}{
"error": err.Error(),
})
return fmt.Errorf("failed to load recommended tenders: %w", err)
}
for i := range tenders {
byMongoID[tenders[i].GetID()] = tenders[i]
}
return nil
})
}
if err := group.Wait(); err != nil {
return nil, err
}
for ref, tender := range byProcedure {
out[ref] = tender
}
for ref, tender := range byMongoID {
out[ref] = tender
}
return out, nil