From cf7512326e7c77abb63dadfc899633ea9b646d35 Mon Sep 17 00:00:00 2001
From: Randy Woods <randy1@radiorandy.com>
Date: Fri, 14 Feb 2025 14:06:52 -0700
Subject: [PATCH] Made POAM export a bit more flexible for other CMMC

---
 .../CSETWebCore.Business/Maturity/CmmcBusiness.cs         | 6 ++++--
 .../CSETWeb_ApiCore/Controllers/CmmcController.cs         | 8 +-------
 .../results/reports/report-list/report-list.component.ts  | 4 ++--
 3 files changed, 7 insertions(+), 11 deletions(-)

diff --git a/CSETWebApi/CSETWeb_Api/CSETWebCore.Business/Maturity/CmmcBusiness.cs b/CSETWebApi/CSETWeb_Api/CSETWebCore.Business/Maturity/CmmcBusiness.cs
index 1a74bbbd2f..3821222c53 100644
--- a/CSETWebApi/CSETWeb_Api/CSETWebCore.Business/Maturity/CmmcBusiness.cs
+++ b/CSETWebApi/CSETWeb_Api/CSETWebCore.Business/Maturity/CmmcBusiness.cs
@@ -12,7 +12,6 @@
 using System.Collections.Generic;
 using System.Xml.Linq;
 using System.Linq;
-using CSETWebCore.Business.Aggregation;
 
 
 namespace CSETWebCore.Business.Maturity
@@ -122,8 +121,11 @@ public int GetScoreForLevel(int assessmentId, int level)
         {
             _context.FillEmptyMaturityQuestionsForAnalysis(assessmentId);
 
+            var modelId = _context.AVAILABLE_MATURITY_MODELS.Where(x => x.Assessment_Id == assessmentId).FirstOrDefault()?.model_id;
+
+
             var levelId = _context.MATURITY_LEVELS
-                .Where(x => x.Level == level && x.Maturity_Model_Id == modelIdCmmc2)
+                .Where(x => x.Level == level && x.Maturity_Model_Id == modelId)
                 .Select(x => x.Maturity_Level_Id)
                 .FirstOrDefault();
 
diff --git a/CSETWebApi/CSETWeb_Api/CSETWeb_ApiCore/Controllers/CmmcController.cs b/CSETWebApi/CSETWeb_Api/CSETWeb_ApiCore/Controllers/CmmcController.cs
index 502dbabcef..95d8e56e3d 100644
--- a/CSETWebApi/CSETWeb_Api/CSETWeb_ApiCore/Controllers/CmmcController.cs
+++ b/CSETWebApi/CSETWeb_Api/CSETWeb_ApiCore/Controllers/CmmcController.cs
@@ -1,17 +1,11 @@
 using CSETWebCore.Business.Authorization;
-using CSETWebCore.Business.Document;
 using CSETWebCore.Business.Maturity;
 using CSETWebCore.DataLayer.Model;
 using CSETWebCore.Interfaces.AdminTab;
-using CSETWebCore.Interfaces.Document;
 using CSETWebCore.Interfaces.Helpers;
 using CSETWebCore.Interfaces.Reports;
-using CSETWebCore.Model.Document;
-using CSETWebCore.Model.Question;
-using Microsoft.AspNetCore.Http;
 using Microsoft.AspNetCore.Mvc;
-using Microsoft.CodeAnalysis.Elfie.Model.Map;
-using System.Collections.Generic;
+
 
 namespace CSETWebCore.Api.Controllers
 {
diff --git a/CSETWebNg/src/app/assessment/results/reports/report-list/report-list.component.ts b/CSETWebNg/src/app/assessment/results/reports/report-list/report-list.component.ts
index a77674b629..ecd720d00a 100644
--- a/CSETWebNg/src/app/assessment/results/reports/report-list/report-list.component.ts
+++ b/CSETWebNg/src/app/assessment/results/reports/report-list/report-list.component.ts
@@ -87,12 +87,12 @@ export class ReportListComponent implements OnInit {
    */
   setCmmcLevelAchievement() {
     const a = this.assessSvc.assessment;
-
     const cmmcModels = ['CMMC', 'CMMC2', 'CMMC2F'];
+
     if (a.maturityModel?.maturityTargetLevel > 1 &&
       cmmcModels.indexOf(a.maturityModel?.modelName) >= 0) {
       this.maturitySvc.getCmmcScores().subscribe((scores: any) => {
-        this.cmmcLevel1Achieved = scores.level1Score == scores.level1MaxScore;
+        this.cmmcLevel1Achieved = scores.level1Score >= scores.level1MaxScore;
       });
     }
   }