Fix tests

hyperledger · Jul 13, 2023 · 58623d9 · 58623d9
1 parent 1511ffe
commit 58623d9
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/mediator/src/main/scala/io/iohk/atala/mediator/app/MediatorAgent.scala b/mediator/src/main/scala/io/iohk/atala/mediator/app/MediatorAgent.scala
@@ -109,7 +109,7 @@ case class MediatorAgent(
     Option[EncryptedMessage]
   ] =
     ZIO
-      .logAnnotate("msgHash", msg.hashCode.toString) {
+      .logAnnotate("msgHash", msg.sha1) {
         for {
           _ <- ZIO.log("receivedMessage")
           maybeSyncReplyMsg <-

diff --git a/mediator/src/main/scala/io/iohk/atala/mediator/protocols/PickupExecuter.scala b/mediator/src/main/scala/io/iohk/atala/mediator/protocols/PickupExecuter.scala
@@ -90,7 +90,7 @@ object PickupExecuter
             from = m.to.asFROM,
             to = m.from.asTO,
             recipient_did = m.recipient_did,
-            attachments = messagesToReturn.map(m => (m.hashCode.toString, m.msg)).toMap,
+            attachments = messagesToReturn.map(m => (m._id, m.msg)).toMap,
           )
         } yield SyncReplyOnly(deliveryRequest.toPlaintextMessage)
       case m: MessageDelivery =>

diff --git a/mediator/src/test/scala/io/iohk/atala/mediator/db/MessageItemRepoSpec.scala b/mediator/src/test/scala/io/iohk/atala/mediator/db/MessageItemRepoSpec.scala
@@ -28,7 +28,7 @@ object MessageItemRepoSpec extends ZIOSpecDefault with AccountStubSetup {
       for {
         messageItem <- ZIO.service[MessageItemRepo]
         msg <- ZIO.fromEither(encryptedMessageAlice)
-        result <- messageItem.findById(msg.hashCode())
+        result <- messageItem.findById(msg.sha1)
       } yield {
         assertTrue(result.contains(MessageItem(msg)))
       }
@@ -39,7 +39,7 @@ object MessageItemRepoSpec extends ZIOSpecDefault with AccountStubSetup {
         msg <- ZIO.fromEither(encryptedMessageAlice)
         msg2 <- ZIO.fromEither(encryptedMessageBob)
         msg2Added <- messageItem.insert(MessageItem(msg2))
-        result <- messageItem.findByIds(Seq(msg.hashCode(), msg2.hashCode()))
+        result <- messageItem.findByIds(Seq(msg.sha1, msg2.sha1))
       } yield {
         assertTrue(result.contains(MessageItem(msg)))
         assertTrue(result.contains(MessageItem(msg2)))

diff --git a/mediator/src/test/scala/io/iohk/atala/mediator/db/UserAccountRepoSpec.scala b/mediator/src/test/scala/io/iohk/atala/mediator/db/UserAccountRepoSpec.scala
@@ -131,7 +131,7 @@ object UserAccountRepoSpec extends ZIOSpecDefault with AccountStubSetup {
         val messageMetaData: Seq[MessageMetaData] = didAccount.map(_.messagesRef).getOrElse(Seq.empty)
         assert(messageMetaData)(
           forall(
-            hasField("hash", (m: MessageMetaData) => m.hash, equalTo(msg.hashCode()))
+            hasField("hash", (m: MessageMetaData) => m.hash, equalTo(msg.sha1))
               && hasField("recipient", (m: MessageMetaData) => m.recipient, equalTo(DIDSubject(bob)))
           )
         )
@@ -141,7 +141,7 @@ object UserAccountRepoSpec extends ZIOSpecDefault with AccountStubSetup {
       for {
         userAccount <- ZIO.service[UserAccountRepo]
         msg <- ZIO.fromEither(encryptedMessageAlice)
-        markedDelivered <- userAccount.markAsDelivered(DIDSubject(alice), Seq(msg.hashCode()))
+        markedDelivered <- userAccount.markAsDelivered(DIDSubject(alice), Seq(msg.sha1))
         didAccount <- userAccount.getDidAccount(DIDSubject(alice))
       } yield {
         assertTrue(markedDelivered == 1)