Add db functions tests

src/main/scala/palabrot/services/ElasticRepository.scala

@@ -15,16 +15,16 @@ class ElasticRepository(client: ElasticClient, index: String) extends MessageRep
       s""" { "chatId" : "${t.chat.id}", "text" : "${t.text}", "date" : "${t.date}", "user" : "${t.from.getOrElse("unknown")}" } """
   }
 
-  override def addMessage(message: TextMessage): IO[Unit] = {
+  override def addMessage(message: TextMessage): IO[Boolean] = {
     client.execute(indexInto(index).doc(message))
-    IO.unit
+    IO.pure(true)
   }
 
-  override def getMessagesFromChat(num: Int, chat: Chat): IO[String] =
-    IO.pure(client.execute(get("chatId", s"${chat.id}")).mapTo[String].value.get.get  )
+  override def getMessagesFromChat(num: Int, chat: Chat): IO[List[String]] =
+    IO.pure(client.execute(get("chatId", s"${chat.id}")).mapTo[String].value.get.get :: Nil)
 
-  override def deleteMessage(chatId: ChatId): IO[Unit] = {
+  override def deleteMessage(chatId: ChatId): IO[Boolean] = {
     client.execute(deleteById("chatId", s"$chatId"))
-    IO.unit
+    IO.pure(true)
   }
 }

src/main/scala/palabrot/services/MessageRepository.scala

@@ -4,9 +4,9 @@ import canoe.models.messages.TextMessage
 import cats.effect.IO
 
 trait MessageRepository{
-    def addMessage(message: TextMessage): IO[Unit]
-    def getMessagesFromChat(num: Int, chat: Chat): IO[String]
-    def deleteMessage(chatId: ChatId): IO[Unit]
+    def addMessage(message: TextMessage): IO[Boolean]
+    def getMessagesFromChat(num: Int, chat: Chat): IO[List[String]]
+    def deleteMessage(chatId: ChatId): IO[Boolean]
 }
 
 

src/main/scala/palabrot/services/Summarizer.scala

@@ -10,6 +10,6 @@ object Summarizer {
   def summary(command: TextMessage, messagesNumber: Int, db: MessageRepository): IO[String] = {
       Loggers.summary.info(Summary, command)
       //Code for return summary
-      db.getMessagesFromChat(messagesNumber, command.chat)
+      db.getMessagesFromChat(messagesNumber, command.chat).flatMap(l => IO.pure(l.mkString(" ")))
     }
 }

src/test/scala/MainFlowTest.scala

@@ -0,0 +1,19 @@
+import canoe.models.messages.TextMessage
+import canoe.models.{Group, User}
+import munit.CatsEffectSuite
+import palabrot.MainFlow
+
+
+class MainFlowTest extends CatsEffectSuite {
+  val testUser = User(1, false, "Test Name", None, None, None, None, None, None)
+  val testGroup = Group(1, Some("Test Group"))
+  val testMessage = TextMessage(1, testGroup, 1, "Test Message", None, Some(testUser))
+
+  test("When delete messages gets true") {
+    MainFlow.DB.deleteMessage(testMessage.chat.id).assert
+  }
+  test("Summarizer returns summary from n(100) messages from Test Group") {
+    MainFlow.DB.addMessage(testMessage).assert
+  }
+
+}

src/test/scala/palabrot/services/ServicesTest.scala

@@ -11,14 +11,15 @@ class ServicesTest extends CatsEffectSuite {
   val testMessage = TextMessage(1, testGroup, 1, "Test Message", None, Some(testUser))
 
   val testDB: MessageRepository = new MessageRepository {
-    override def addMessage(message: TextMessage): IO[Unit] = IO.unit
-    override def getMessagesFromChat(num: Int, chat: Chat): IO[String] = chat match {
-      case Group(_, Some(title)) => IO.pure (s"Summary of $num messages from chat $title")
-      case _ => IO.pure("Test failed")
+    override def addMessage(message: TextMessage): IO[Boolean] = IO.pure(true)
+    override def getMessagesFromChat(num: Int, chat: Chat): IO[List[String]] = chat match {
+      case Group(_, Some(title)) => IO.pure (s"Summary of $num messages from chat $title" :: Nil)
+      case _ => IO.pure("Test failed" :: Nil)
     }
-    override def deleteMessage(chatId: ChatId): IO[Unit] = IO.unit
+    override def deleteMessage(chatId: ChatId): IO[Boolean] = IO.pure(true)
   }
 
+
   test("Summarizer returns summary from n(100) messages from Test Group") {
     Summarizer.summary(testMessage,100, testDB)
       .assertEquals("Summary of 100 messages from chat Test Group")