From 124343f5079d63184988c2e87b3bd5f524fe8ffc Mon Sep 17 00:00:00 2001
From: kurilova <kurilova@google.com>
Date: Mon, 9 Sep 2024 10:04:33 +0000
Subject: [PATCH] Fix text for High risk profile save window

---
 .../success-dialog/success-dialog.component.html     |  4 ++--
 .../success-dialog/success-dialog.component.spec.ts  | 10 ++++++++++
 .../success-dialog/success-dialog.component.ts       | 12 +++++++++++-
 3 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.html b/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.html
index 7e616388c..956db409a 100644
--- a/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.html
+++ b/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.html
@@ -29,8 +29,8 @@
     </span></span
   >
 
-  <br />This means that your device might be eligible for most networks. Final
-  estimation will be available in the Zip file.
+  <br />{{ getRiskExplanation(data.profile.risk) }} Final estimation will be
+  available in the Zip file.
 </p>
 <mat-dialog-actions align="end" class="delete-form-actions">
   <button
diff --git a/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.spec.ts b/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.spec.ts
index e497af997..79d200852 100644
--- a/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.spec.ts
+++ b/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.spec.ts
@@ -20,6 +20,7 @@ import { TestRunService } from '../../../../services/test-run.service';
 import { MAT_DIALOG_DATA, MatDialogRef } from '@angular/material/dialog';
 import { of } from 'rxjs';
 import { PROFILE_MOCK } from '../../../../mocks/profile.mock';
+import { ProfileRisk } from '../../../../model/profile';
 
 describe('SuccessDialogComponent', () => {
   let component: SuccessDialogComponent;
@@ -67,4 +68,13 @@ describe('SuccessDialogComponent', () => {
 
     closeSpy.calls.reset();
   });
+
+  it('should return proper text for risk', () => {
+    expect(component.getRiskExplanation(ProfileRisk.LIMITED)).toEqual(
+      'This means that your device might be eligible for most networks.'
+    );
+    expect(component.getRiskExplanation(ProfileRisk.HIGH)).toEqual(
+      'This means that your device might be eligible only for some networks.'
+    );
+  });
 });
diff --git a/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.ts b/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.ts
index 6c3fb58d9..244d7431f 100644
--- a/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.ts
+++ b/modules/ui/src/app/pages/risk-assessment/components/success-dialog/success-dialog.component.ts
@@ -21,7 +21,11 @@ import {
 } from '@angular/material/dialog';
 import { MatButtonModule } from '@angular/material/button';
 import { EscapableDialogComponent } from '../../../../components/escapable-dialog/escapable-dialog.component';
-import { Profile, RiskResultClassName } from '../../../../model/profile';
+import {
+  Profile,
+  ProfileRisk,
+  RiskResultClassName,
+} from '../../../../model/profile';
 import { TestRunService } from '../../../../services/test-run.service';
 import { CommonModule } from '@angular/common';
 
@@ -52,4 +56,10 @@ export class SuccessDialogComponent extends EscapableDialogComponent {
   public getRiskClass(riskResult: string): RiskResultClassName {
     return this.testRunService.getRiskClass(riskResult);
   }
+
+  getRiskExplanation(risk: string | undefined) {
+    return risk === ProfileRisk.LIMITED
+      ? 'This means that your device might be eligible for most networks.'
+      : 'This means that your device might be eligible only for some networks.';
+  }
 }