LANDGRIF-162 - Add tmp logging to task for testing

Vizzuality · Feb 23, 2025 · da1ff18 · da1ff18
1 parent 32e2626
commit da1ff18
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 0 deletions.
diff --git a/api/src/modules/indicator-records/indicator-records.module.ts b/api/src/modules/indicator-records/indicator-records.module.ts
@@ -11,6 +11,7 @@ import { CachedDataModule } from 'modules/cached-data/cached-data.module';
 import { ImpactCalculator } from 'modules/indicator-records/services/impact-calculator.service';
 import { ImpactQueryBuilder } from 'modules/indicator-records/services/indicator-dependency-manager.service';
 import { IndicatorRecordRepository } from 'modules/indicator-records/indicator-record.repository';
+import { TasksModule } from '../tasks/tasks.module';
 
 @Module({
   imports: [
@@ -20,6 +21,7 @@ import { IndicatorRecordRepository } from 'modules/indicator-records/indicator-r
     MaterialsModule,
     SourcingRecordsModule,
     CachedDataModule,
+    TasksModule,
   ],
   controllers: [IndicatorRecordsController],
   providers: [

diff --git a/api/src/modules/indicator-records/services/impact-calculator.service.ts b/api/src/modules/indicator-records/services/impact-calculator.service.ts
@@ -33,6 +33,7 @@ import { ImpactCalculationProgressTracker } from 'modules/impact/progress-tracke
 import { ImportProgressTrackerFactory } from 'modules/events/import-data-progress/import-progress.tracker.factory';
 import { SourcingLocation } from 'modules/sourcing-locations/sourcing-location.entity';
 import { AppConfig } from 'utils/app.config';
+import { TasksService } from '../../tasks/tasks.service';
 
 /**
  * @description: This is PoC (Proof of Concept) for the updated LG methodology v0.1
@@ -56,6 +57,7 @@ export class ImpactCalculator {
     private readonly cachedDataService: CachedDataService,
     private readonly dataSource: DataSource,
     private readonly importProgressTrackerFactory: ImportProgressTrackerFactory,
+    private readonly taskService: TasksService,
   ) {}
 
   async calculateImpactForAllSourcingRecords(
@@ -519,6 +521,13 @@ export class ImpactCalculator {
       SourcingRecordsWithIndicatorRawData[]
     > = new Map<string, SourcingRecordsWithIndicatorRawData[]>();
     const repository = this.dataSource.getRepository(SourcingLocation);
+    // For testing purposes, track locations with no production in task, will remove this later
+    const locationIdsWithNoProduction: string[] = [];
+    const { id } = await this.taskService.taskRepository.findOneOrFail({
+      order: {
+        updatedAt: 'DESC',
+      },
+    });
 
     // Group records by location where production is 0 or null
     // TODO: We must apply this when harvesting is 0 as well, but given the use of this approach is not straightforward, and how to apply the new values
@@ -527,10 +536,17 @@ export class ImpactCalculator {
       if (!record.production || record.production === 0) {
         if (!recordsPorLocation.has(record.sourcingLocationId)) {
           recordsPorLocation.set(record.sourcingLocationId, []);
+          locationIdsWithNoProduction.push(record.sourcingLocationId);
         }
         recordsPorLocation.get(record.sourcingLocationId)!.push(record);
       }
     }
+    if (locationIdsWithNoProduction.length) {
+      await this.taskService.updateImportTask({
+        taskId: id,
+        newData: { locationsWithNoProduction: locationIdsWithNoProduction },
+      });
+    }
     const dataArray = Array.from(recordsPorLocation.entries());
     const promises: Promise<any>[] = dataArray.map(async (elem) => {
       // For each location that has no production value, get its geo region

diff --git a/...test/integration/indicator-record/calculations/distributed-impact-over-geo-region.spec.ts b/...test/integration/indicator-record/calculations/distributed-impact-over-geo-region.spec.ts
@@ -12,6 +12,7 @@ import { IndicatorsService } from '../../../../src/modules/indicators/indicators
 import { ImpactQueryBuilder } from '../../../../src/modules/indicator-records/services/indicator-dependency-manager.service';
 import { CachedDataService } from '../../../../src/modules/cached-data/cached-data.service';
 import { ImportProgressTrackerFactory } from '../../../../src/modules/events/import-data-progress/import-progress.tracker.factory';
+import { TasksService } from '../../../../src/modules/tasks/tasks.service';
 
 const dummyProvider = { useValue: {} };
 
@@ -34,6 +35,17 @@ describe('ImpactCalculator.updateDistributedImpactOverGeoRegion', () => {
       providers: [
         ImpactCalculator,
         { provide: DataSource, useValue: dataSourceMock },
+        {
+          provide: TasksService,
+          useValue: {
+            taskRepository: {
+              findOneOrFail: () => ({
+                id: 1,
+              }),
+            },
+            updateImportTask: jest.fn(),
+          },
+        },
         { provide: IndicatorRecordRepository, ...dummyProvider },
         { provide: MaterialsToH3sService, ...dummyProvider },
         { provide: IndicatorsService, ...dummyProvider },