Add filename to Submission persistence

Author: schbetsy

persistence/src/main/scala/hmda/persistence/institutions/SubmissionPersistence.scala

@@ -13,6 +13,7 @@ object SubmissionPersistence {
 
   case object CreateSubmission extends Command
   case class UpdateSubmissionStatus(id: SubmissionId, status: SubmissionStatus) extends Command
+  case class AddSubmissionFileName(id: SubmissionId, fileName: String) extends Command
   case class GetSubmissionById(id: SubmissionId) extends Command
   case object GetLatestSubmission extends Command
 
@@ -40,6 +41,12 @@ object SubmissionPersistence {
           }
 
           SubmissionState(submissions.updated(i, updatedSub))
+
+        case SubmissionFileNameAdded(id, name) =>
+          val sub = submissions.find(_.id == id).getOrElse(Submission())
+          val index = submissions.indexOf(sub)
+          val updated = sub.copy(fileName = name)
+          SubmissionState(submissions.updated(index, updated))
       }
     }
 
@@ -83,6 +90,17 @@ class SubmissionPersistence(institutionId: String, period: String) extends HmdaP
         sender() ! None
       }
 
+    case AddSubmissionFileName(id, name) =>
+      if (state.submissions.map(_.id).contains(id)) {
+        persist(SubmissionFileNameAdded(id, name)) { e =>
+          updateState(e)
+          sender() ! Some(Submission(id, fileName = name))
+        }
+      } else {
+        log.warning(s"Submission does not exist. Could not add filename for submission with id $id")
+        sender() ! None
+      }
+
     case GetSubmissionById(id) =>
       val submission = state.submissions.find(s => s.id == id).getOrElse(Submission(SubmissionId(), Failed("No submission found"), 0L, 0L))
       sender() ! submission

persistence/src/test/scala/hmda/persistence/institutions/SubmissionPersistenceSpec.scala

@@ -54,6 +54,20 @@ class SubmissionPersistenceSpec extends ActorSpec {
       probe.expectMsg(None)
     }
 
+    "add a filename" in {
+      val id = SubmissionId("0", "2017", 2)
+      probe.send(submissionsActor, GetSubmissionById(id))
+      val before = probe.receiveOne(5.seconds)
+      before.asInstanceOf[Submission].fileName mustBe ""
+
+      probe.send(submissionsActor, AddSubmissionFileName(id, "my file name dot txt"))
+      probe.expectMsgType[Some[Submission]]
+
+      probe.send(submissionsActor, GetSubmissionById(id))
+      val after = probe.receiveOne(5.seconds)
+      after.asInstanceOf[Submission].fileName mustBe "my file name dot txt"
+    }
+
     "return list of submissions in order of descending sequence number" in {
       probe.send(submissionsActor, GetState)
       val subs = probe.expectMsgType[List[Submission]]