Spec for calculateYear in ReportUtil

Author: schbetsy

publication/src/test/scala/hmda/publication/reports/util/ReportUtilSpec.scala

@@ -1,10 +1,19 @@
 package hmda.publication.reports.util
 
+import akka.actor.ActorSystem
+import akka.stream.ActorMaterializer
+import akka.stream.scaladsl.Source
+import hmda.model.fi.lar.LarGenerators
 import hmda.model.publication.reports.MSAReport
 import hmda.publication.reports.util.ReportUtil._
+import hmda.query.repository.filing.LarConverter._
 import org.scalatest.{ AsyncWordSpec, MustMatchers }
 
-class ReportUtilSpec extends AsyncWordSpec with MustMatchers {
+class ReportUtilSpec extends AsyncWordSpec with MustMatchers with LarGenerators {
+
+  implicit val system = ActorSystem()
+  implicit val ec = system.dispatcher
+  implicit val materializer = ActorMaterializer()
 
   "msaReport" must {
     "get MSA info for given fips code and return MSAReport object" in {
@@ -31,4 +40,15 @@ class ReportUtilSpec extends AsyncWordSpec with MustMatchers {
     }
   }
 
+  "calculateYear" must {
+    "given a larSource, return the Activity Year for the LARs in it" in {
+      // Activity Year is the same year as Action Taken Date. This is enforced by edit S270
+      val lars = larListGen.sample.get.map(_.copy(actionTakenDate = 20090822))
+      val src = Source.fromIterator(() => lars.toIterator)
+        .map(lar => toLoanApplicationRegisterQuery(lar))
+
+      calculateYear(src).map(_ mustBe 2009)
+    }
+  }
+
 }