GFORMS-3026 - Update data retrieval to correct response from DES

hmrc · Nov 12, 2024 · e1ce12c · e1ce12c
1 parent 2aacd61
commit e1ce12c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/app/uk/gov/hmrc/gform/eval/EvaluationResults.scala b/app/uk/gov/hmrc/gform/eval/EvaluationResults.scala
@@ -637,6 +637,14 @@ case class EvaluationResults(
             dataRetrieveId
           )
         )
+      case d @ DataRetrieveCtx(_, DataRetrieve.Attribute("agencyAddress")) =>
+        evaluationContext.thirdPartyData.dataRetrieve.fold(ExpressionResult.empty) { dr =>
+          DataRetrieveEval.getDataRetrieveAttribute(dr, d) match {
+            case Some(h :: Nil) => AddressResult(List(h))
+            case Some(xs)       => AddressResult(xs)
+            case None           => Empty
+          }
+        }
       case d @ DataRetrieveCtx(_, _) =>
         val exprResult =
           evaluationContext.thirdPartyData.dataRetrieve.fold(ExpressionResult.empty) { dr =>