SubmissionManager adds Filename to SubmissionPersistence

Author: schbetsy

persistence/src/main/scala/hmda/persistence/processing/SubmissionManager.scala

@@ -7,9 +7,10 @@ import akka.pattern.ask
 import akka.util.Timeout
 import com.typesafe.config.ConfigFactory
 import hmda.model.fi.{ Signed => _, _ }
-import hmda.persistence.institutions.FilingPersistence
+import hmda.persistence.institutions.{ FilingPersistence, SubmissionPersistence }
 import hmda.persistence.institutions.FilingPersistence.{ GetFilingByPeriod, UpdateFilingStatus }
-import hmda.persistence.HmdaSupervisor.{ FindFilings, FindHmdaFiling }
+import hmda.persistence.HmdaSupervisor.{ FindFilings, FindHmdaFiling, FindSubmissions }
+import hmda.persistence.institutions.SubmissionPersistence.AddSubmissionFileName
 import hmda.persistence.messages.CommonMessages.{ Command, GetState, Shutdown }
 import hmda.persistence.model.HmdaActor
 import hmda.persistence.processing.HmdaFileParser.ReadHmdaRawFile
@@ -62,6 +63,7 @@ class SubmissionManager(validationStats: ActorRef, submissionId: SubmissionId) e
     .props(supervisor, validationStats, submissionId)
     .withDispatcher("persistence-dispatcher"))
   val filingPersistence = (supervisor ? FindFilings(FilingPersistence.name, submissionId.institutionId)).mapTo[ActorRef]
+  val submissionPersistence = (supervisor ? FindSubmissions(SubmissionPersistence.name, submissionId.institutionId, submissionId.period)).mapTo[ActorRef]
 
   var uploaded: Int = 0
 
@@ -75,7 +77,7 @@ class SubmissionManager(validationStats: ActorRef, submissionId: SubmissionId) e
   override def receive: Receive = {
 
     case AddFileName(name) =>
-      submissionUpload ! AddFileName(name)
+      submissionPersistence.map(_ ! AddSubmissionFileName(submissionId, name))
 
     case StartUpload =>
       log.info(s"Start upload for submission: ${submissionId.toString}")

persistence/src/test/scala/hmda/persistence/processing/SubmissionManagerSpec.scala

@@ -16,8 +16,8 @@ import hmda.persistence.messages.CommonMessages.GetState
 import hmda.persistence.model.ActorSpec
 import hmda.persistence.processing.ProcessingMessages.{ CompleteUpload, Persisted, StartUpload }
 import hmda.persistence.institutions.FilingPersistence._
-import hmda.persistence.institutions.SubmissionPersistence.CreateSubmission
-import hmda.persistence.processing.HmdaRawFile.AddLine
+import hmda.persistence.institutions.SubmissionPersistence.{ CreateSubmission, GetSubmissionById }
+import hmda.persistence.processing.HmdaRawFile.{ AddFileName, AddLine }
 import hmda.validation.ValidationStats
 
 import scala.concurrent.Await
@@ -97,6 +97,14 @@ class SubmissionManagerSpec extends ActorSpec {
       probe.expectMsg(ValidatedWithErrors)
     }
 
+    "add filename for submission" in {
+      probe.send(submissionManager, AddFileName("my_file_name2.txt"))
+      Thread.sleep(1000)
+      probe.send(submissionPersistence, GetSubmissionById(submissionId))
+      val sub = probe.receiveOne(5.seconds).asInstanceOf[Submission]
+      sub.fileName mustBe "my_file_name2.txt"
+    }
+
     "have Filing status 'completed' after signature" in {
       probe.send(submissionManager, hmda.persistence.processing.ProcessingMessages.Signed)
       probe.expectMsg(Some(Signed))