Remove redundant MQ type on TestData

DFE-Digital · Aug 20, 2024 · 2a7953a · 2a7953a
1 parent ced2453
commit 2a7953a
Showing 1 changed file with 9 additions and 25 deletions.
diff --git a/TeachingRecordSystem/tests/TeachingRecordSystem.TestCommon/TestData.CreatePerson.cs b/TeachingRecordSystem/tests/TeachingRecordSystem.TestCommon/TestData.CreatePerson.cs
@@ -33,7 +33,6 @@ public class CreatePersonBuilder
         private bool? _hasNationalInsuranceNumber;
         private string? _nationalInsuranceNumber;
         private readonly List<Qualification> _qualifications = new();
-        private readonly List<MandatoryQualificationInfo> _mandatoryQualifications = new();
         private readonly List<QtsRegistration> _qtsRegistrations = new();
         private readonly List<Sanction> _sanctions = [];
         private readonly List<CreatePersonMandatoryQualificationBuilder> _mqBuilders = [];
@@ -710,7 +709,7 @@ public CreatePersonMandatoryQualificationBuilder WithImportedByDqtUser(EventMode
             return this;
         }
 
-        internal async Task<MandatoryQualificationInfo> Execute(CreatePersonBuilder createPersonBuilder, TestData testData)
+        internal async Task<MandatoryQualification> Execute(CreatePersonBuilder createPersonBuilder, TestData testData)
         {
             var personId = createPersonBuilder.PersonId;
 
@@ -721,13 +720,13 @@ internal async Task<MandatoryQualificationInfo> Execute(CreatePersonBuilder crea
             var endDate = _endDate.ValueOr(status == MandatoryQualificationStatus.Passed ? testData.GenerateDate(min: (startDate ?? new DateOnly(2000, 1, 1)).AddYears(1)) : null);
             var createdUtc = _createdUtc.ValueOr(testData.Clock.UtcNow);
 
-            await testData.WithDbContext(async dbContext =>
+            return await testData.WithDbContext(async dbContext =>
             {
                 var provider = providerId.HasValue ?
                     await dbContext.MandatoryQualificationProviders.SingleAsync(p => p.MandatoryQualificationProviderId == providerId) :
                     null;
 
-                dbContext.MandatoryQualifications.Add(new MandatoryQualification()
+                var mq = new MandatoryQualification()
                 {
                     QualificationId = QualificationId,
                     CreatedOn = testData.Clock.UtcNow,
@@ -740,7 +739,9 @@ await dbContext.MandatoryQualificationProviders.SingleAsync(p => p.MandatoryQual
                     EndDate = endDate,
                     DqtSpecialismId = _dqtSpecialismId.ValueOr((Guid?)null),
                     DqtMqEstablishmentId = _mqEstablishmentId.ValueOr((Guid?)null)
-                });
+                };
+
+                dbContext.MandatoryQualifications.Add(mq);
 
                 if (_importedByUser.HasValue)
                 {
@@ -799,18 +800,9 @@ await dbContext.MandatoryQualificationProviders.SingleAsync(p => p.MandatoryQual
                 }
 
                 await dbContext.SaveChangesAsync();
-            });
 
-            var mqInfo = new MandatoryQualificationInfo(
-                QualificationId,
-                providerId,
-                specialism,
-                status,
-                startDate,
-                endDate
-            );
-
-            return mqInfo;
+                return mq;
+            });
         }
     }
 
@@ -834,7 +826,7 @@ public record CreatePersonResult
         public required DateOnly? QtsDate { get; init; }
         public required DateOnly? EytsDate { get; init; }
         public required IReadOnlyCollection<Sanction> Sanctions { get; init; }
-        public required IReadOnlyCollection<MandatoryQualificationInfo> MandatoryQualifications { get; init; }
+        public required IReadOnlyCollection<MandatoryQualification> MandatoryQualifications { get; init; }
         public required IReadOnlyCollection<Induction> Inductions { get; init; }
         public required IReadOnlyCollection<InductionPeriod> InductionPeriods { get; init; }
     }
@@ -844,14 +836,6 @@ public record InductionPeriod(Guid InductionId, DateOnly? startDate, DateOnly? e
 
     public record Sanction(Guid SanctionId, string SanctionCode, DateOnly? StartDate, DateOnly? EndDate, DateOnly? ReviewDate, bool Spent, string? Details, string? DetailsLink, bool IsActive);
 
-    public record MandatoryQualificationInfo(
-        Guid QualificationId,
-        Guid? ProviderId,
-        MandatoryQualificationSpecialism? Specialism,
-        MandatoryQualificationStatus? Status,
-        DateOnly? StartDate,
-        DateOnly? EndDate);
-
     public record QtsRegistration(DateOnly? QtsDate, string? TeacherStatusValue, DateTime? CreatedOn, DateOnly? EytsDate, string? EytsStatusValue);
 
     public record Qualification(