From 27a5fb951045e83990f39f63375162608f2982a6 Mon Sep 17 00:00:00 2001
From: Thanh Le <lenguyenthanh@hotmail.com>
Date: Sat, 25 Nov 2023 01:44:01 +0100
Subject: [PATCH] Write integration test for a almost real game

---
 app/src/main/scala/model.scala           |  2 +-
 app/src/test/scala/IntegrationTest.scala | 89 ++++++++++++++++++++++--
 2 files changed, 84 insertions(+), 7 deletions(-)

diff --git a/app/src/main/scala/model.scala b/app/src/main/scala/model.scala
index d76a68b..e2be36e 100644
--- a/app/src/main/scala/model.scala
+++ b/app/src/main/scala/model.scala
@@ -63,7 +63,7 @@ object Lila:
 
   case class Move(gameId: GameId, moves: String, uci: Uci):
     def sign  = moves.takeRight(20).replace(" ", "")
-    def write = s"${gameId} $sign ${uci.uci}"
+    def write = s"$gameId $sign ${uci.uci}"
 
   case class Clock(wtime: Int, btime: Int, inc: Int) derives Codec.AsObject
 
diff --git a/app/src/test/scala/IntegrationTest.scala b/app/src/test/scala/IntegrationTest.scala
index 129101e..f80b3e3 100644
--- a/app/src/test/scala/IntegrationTest.scala
+++ b/app/src/test/scala/IntegrationTest.scala
@@ -5,35 +5,112 @@ import cats.effect.kernel.Resource
 import cats.syntax.all.*
 import com.comcast.ip4s.{ Host, Port }
 import com.dimafeng.testcontainers.GenericContainer
+import io.chrisdavenport.rediculous.RedisPubSub
+import io.circe.Json
+import lila.fishnet.Fishnet.*
+import lila.fishnet.http.HealthCheck.AppStatus
+import org.http4s.*
+import org.http4s.circe.CirceEntityDecoder.*
+import org.http4s.circe.CirceEntityEncoder.*
 import org.http4s.ember.client.EmberClientBuilder
+import org.http4s.implicits.*
 import org.testcontainers.containers.wait.strategy.Wait
 import org.typelevel.log4cats.Logger
 import org.typelevel.log4cats.noop.NoOpLogger
-import org.http4s.circe.CirceEntityDecoder.*
 import weaver.*
-import lila.fishnet.http.HealthCheck.AppStatus
+import org.http4s.client.Client
+import scala.concurrent.duration.*
+import cats.effect.kernel.Ref
 
-object IntegrationTest extends SimpleIOSuite:
+object IntegrationTest extends IOSuite:
 
   given Logger[IO] = NoOpLogger[IO]
-  private def resource: Resource[IO, Unit] =
+
+  override type Res = AppResources
+  // start our server
+  override def sharedResource: Resource[IO, Res] =
     for
       redis         <- RedisContainer.startRedis
       defaultConfig <- Resource.eval(AppConfig.load)
       config = defaultConfig.copy(redis = redis)
       res <- AppResources.instance(config.redis)
       _   <- FishnetApp(res, config).run()
-    yield ()
+    yield res
 
   test("health check should return healthy"):
-    (resource >> client)
+    client
       .use(
         _.expect[AppStatus]("http://localhost:9665/health")
           .map(expect.same(_, AppStatus(true)))
       )
 
+  test("let's play a game"): res =>
+    val fishnet               = Fishnet("2.7.2", ClientKey("secret-key"))
+    val fishnetAcquireRequest = Acquire(fishnet)
+    val bestMoves             = List("e7e6", "d7d5", "d8d6")
+    val postMoves             = bestMoves.map(m => PostMove(fishnet, Move(BestMove(m))))
+
+    val gameId = "CPzkP0tq"
+    val lilaRequests =
+      List(
+        "CPzkP0tq;1;;;;d2d4",
+        "CPzkP0tq;1;;;;d2d4 e7e6 h2h4",
+        "CPzkP0tq;1;;;;d2d4 e7e6 h2h4 d7d5 e2e3",
+        "CPzkP0tq;1;;;;d2d4 e7e6 h2h4 d7d5 e2e3 d8d6 f1d3"
+      )
+
+    val expectedMoves = List(
+      s"$gameId d2d4 e7e6",
+      s"$gameId d2d4e7e6h2h4 d7d5",
+      s"$gameId e7e6h2h4d7d5e2e3 d8d6"
+    )
+
+    def simulateFishnetClient(client: Client[IO]) =
+      client
+        .expect[Json](acquireRequest(fishnetAcquireRequest))
+        .map(toWorkId)
+        .flatMap: workId =>
+          postMoves.foldM[IO, WorkId](workId): (workId, move) =>
+            client.expect[Json](bestMoveRequest(workId, move)).map(toWorkId)
+
+    def toWorkId(json: Json) =
+      WorkId(json.hcursor.downField("work").downField("id").as[String].toOption.get)
+
+    // sleep to make sure that moves are in order
+    def scenario(client: Client[IO]) =
+      lilaRequests.traverse_(sendWorkRequest(res, _) >> IO.sleep(100.millis)) >> simulateFishnetClient(client)
+
+    val x = for
+      client <- client
+      ref    <- Resource.eval(Ref.of[IO, List[String]](Nil))
+      _      <- RedisFishnetInListener(res.redisPubsub, ref).background
+      _      <- Resource.eval(scenario(client))
+      x      <- Resource.eval(ref.get)
+    yield x
+    x.use(x => IO.pure(expect.same(x, expectedMoves)))
+
+  def acquireRequest(acquire: Acquire) = Request[IO](
+    method = Method.POST,
+    uri = uri"http://localhost:9665/fishnet/acquire"
+  ).withEntity(acquire)
+
+  def bestMoveRequest(workId: WorkId, move: PostMove) = Request[IO](
+    method = Method.POST,
+    uri = uri"http://localhost:9665/fishnet/move" / workId.value
+  ).withEntity(move)
+
+  private def sendWorkRequest(res: AppResources, work: String): IO[Unit] =
+    res.redisPubsub.publish("fishnet-out", work).void
+
   private def client = EmberClientBuilder.default[IO].build
 
+object RedisFishnetInListener:
+  def apply(pubsub: RedisPubSub[IO], ref: Ref[IO, List[String]]): IO[Unit] =
+    pubsub.subscribe(
+      "fishnet-in",
+      msg => ref.update(_ :+ msg.message) >> IO.println(s"recieved $msg")
+    ) *> IO.println("sub") *> pubsub.runMessages *> IO.println("run")
+
 object RedisContainer:
 
   private val REDIS_PORT = 6379