MET-6144 Add test assertion for task info write counter

Author: Ruufuus

client/src/test/java/eu/europeana/processing/AbstractPerformanceIT.java

@@ -8,6 +8,7 @@
 import eu.europeana.processing.config.db.entity.ExecutionRecordExceptionLog;
 import eu.europeana.processing.config.db.repositories.ExecutionRecordExceptionLogRepository;
 import eu.europeana.processing.config.db.repositories.ExecutionRecordRepository;
+import eu.europeana.processing.config.db.repositories.TaskInfoRepository;
 import jakarta.annotation.Resource;
 import java.io.IOException;
 import java.lang.invoke.MethodHandles;
@@ -40,6 +41,9 @@ public abstract class AbstractPerformanceIT {
   @Resource
   protected ExecutionRecordExceptionLogRepository<ExecutionRecordExceptionLog> executionRecordExceptionLogRepository;
 
+  @Resource
+  protected TaskInfoRepository taskInfoRepository;
+
   private static boolean firstTest = true;
 
   private static boolean cleared;
@@ -87,6 +91,7 @@ protected void validateResult(int stepNumber) throws IOException {
     long expectedErrorCount = stepNumber != 2 ? 0: sourceProperties.getRecordCount() - sourceProperties.getValidRecordCount();
     Assertions.assertThat(executionRecordRepository.countByDatasetIdAndExecutionId(datasetId, taskId)).isEqualTo(expectedRecordCount);
     Assertions.assertThat(executionRecordExceptionLogRepository.countByDatasetIdAndExecutionId(datasetId, taskId)).isEqualTo(expectedErrorCount);
+    Assertions.assertThat(taskInfoRepository.getByTaskId(Long.parseLong(taskId)).getWriteCount()).isEqualTo(expectedRecordCount + expectedErrorCount);
   }
 
 }

client/src/test/java/eu/europeana/processing/config/db/entity/TaskInfo.java

@@ -0,0 +1,25 @@
+package eu.europeana.processing.config.db.entity;
+
+import jakarta.persistence.Column;
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.Table;
+import lombok.Getter;
+import lombok.Setter;
+
+@Entity
+@Getter
+@Setter
+@Table(schema = "batch-framework")
+public class TaskInfo {
+
+    @Id
+    @Column(nullable = false, length = 50)
+    private Long taskId;
+
+    @Column
+    private Long commitCount;
+
+    @Column
+    private Long writeCount;
+}

client/src/test/java/eu/europeana/processing/config/db/repositories/TaskInfoRepository.java

@@ -0,0 +1,10 @@
+package eu.europeana.processing.config.db.repositories;
+
+import eu.europeana.processing.config.db.entity.TaskInfo;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface TaskInfoRepository extends JpaRepository<TaskInfo, Long> {
+    TaskInfo getByTaskId(Long taskId);
+}