/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * license agreements; and to You under the Apache License, version 2.0:
 *
 *   https://www.apache.org/licenses/LICENSE-2.0
 *
 * This file is part of the Apache Pekko project, which was derived from Akka.
 */

/*
 * Copyright (C) 2021-2022 Lightbend Inc. <https://www.lightbend.com>
 */

package org.apache.pekko.persistence.testkit.state.scaladsl

import org.apache.pekko
import pekko.actor.ExtendedActorSystem
import pekko.actor.testkit.typed.scaladsl.LogCapturing
import pekko.actor.testkit.typed.scaladsl.ScalaTestWithActorTestKit
import pekko.persistence.query.NoOffset
import pekko.persistence.query.Sequence
import pekko.persistence.query.UpdatedDurableState
import pekko.persistence.testkit.PersistenceTestKitDurableStateStorePlugin
import pekko.stream.scaladsl.Sink
import pekko.stream.testkit.scaladsl.TestSink

import com.typesafe.config.ConfigFactory
import org.scalatest.wordspec.AnyWordSpecLike

object PersistenceTestKitDurableStateStoreSpec {
  val config =
    PersistenceTestKitDurableStateStorePlugin.config.withFallback(ConfigFactory.parseString("""
    pekko.loglevel = DEBUG
      """))
  case class Record(id: Int, name: String)
}

class PersistenceTestKitDurableStateStoreSpec
    extends ScalaTestWithActorTestKit(PersistenceTestKitDurableStateStoreSpec.config)
    with LogCapturing
    with AnyWordSpecLike {

  import PersistenceTestKitDurableStateStoreSpec._

  implicit val classic: pekko.actor.ActorSystem = system.classicSystem

  "Persistent test kit state store" must {
    "find individual objects" in {
      val stateStore = new PersistenceTestKitDurableStateStore[Record](classic.asInstanceOf[ExtendedActorSystem])
      val record = Record(1, "name-1")
      val tag = "tag-1"
      val persistenceId = "record-1"
      stateStore.upsertObject(persistenceId, 1L, record, tag).futureValue
      val updated = stateStore.getObject(persistenceId).futureValue
      updated.value should be(Some(record))
      updated.revision should be(1L)
    }

    "find tagged state changes ordered by upsert" in {
      val stateStore = new PersistenceTestKitDurableStateStore[Record](classic.asInstanceOf[ExtendedActorSystem])
      val record = Record(1, "name-1")
      val recordChange = Record(1, "my-name-1")
      val tag = "tag-1"
      stateStore.upsertObject("record-1", 1L, record, tag)
      val testSink = stateStore
        .changes(tag, NoOffset)
        .collect { case u: UpdatedDurableState[Record] => u }
        .runWith(TestSink[UpdatedDurableState[Record]]())

      val firstStateChange = testSink.request(1).expectNext()
      firstStateChange.value should be(record)
      firstStateChange.revision should be(1L)

      stateStore.upsertObject("record-1", 2L, recordChange, tag)
      val secondStateChange = testSink.request(1).expectNext()
      secondStateChange.value should be(recordChange)
      secondStateChange.revision should be(2L)
      secondStateChange.offset
        .asInstanceOf[Sequence]
        .value should be >= (firstStateChange.offset.asInstanceOf[Sequence].value)
    }

    "find tagged current state changes ordered by upsert" in {
      val stateStore = new PersistenceTestKitDurableStateStore[Record](classic.asInstanceOf[ExtendedActorSystem])
      val record = Record(1, "name-1")
      val tag = "tag-1"

      stateStore.upsertObject("record-1", 1L, record, tag)

      val testSinkCurrentChanges =
        stateStore
          .currentChanges(tag, NoOffset)
          .collect { case u: UpdatedDurableState[Record] => u }
          .runWith(TestSink[UpdatedDurableState[Record]]())

      stateStore.upsertObject("record-1", 2L, record.copy(name = "my-name-1-2"), tag)
      stateStore.upsertObject("record-1", 3L, record.copy(name = "my-name-1-3"), tag)

      val currentStateChange = testSinkCurrentChanges.request(1).expectNext()

      currentStateChange.value should be(record)
      currentStateChange.revision should be(1L)
      testSinkCurrentChanges.request(1).expectComplete()

      val testSinkIllegalOffset =
        stateStore
          .currentChanges(tag, Sequence(100L))
          .collect { case u: UpdatedDurableState[Record] => u }
          .runWith(TestSink[UpdatedDurableState[Record]]())
      testSinkIllegalOffset.request(1).expectNoMessage()
    }

    "return current changes when there are no further changes" in {
      val stateStore = new PersistenceTestKitDurableStateStore[Record](classic.asInstanceOf[ExtendedActorSystem])
      val record = Record(1, "name-1")
      val tag = "tag-1"

      stateStore.upsertObject("record-1", 1L, record, tag)

      val testSinkCurrentChanges =
        stateStore
          .currentChanges(tag, NoOffset)
          .collect { case u: UpdatedDurableState[Record] => u }
          .runWith(TestSink[UpdatedDurableState[Record]]())

      val currentStateChange = testSinkCurrentChanges.request(1).expectNext()

      currentStateChange.value should be(record)
      currentStateChange.revision should be(1L)
      testSinkCurrentChanges.request(1).expectComplete()
    }

    "find state changes by slice ordered by upsert" in {
      val stateStore = new PersistenceTestKitDurableStateStore[Record](classic.asInstanceOf[ExtendedActorSystem])
      val record = Record(1, "name-1")
      val recordChange = Record(1, "my-name-1")
      stateStore.upsertObject("Test|record-1", 1L, record, "")
      val maxSlice = stateStore.sliceRanges(1).head.max
      val testSink = stateStore
        .changesBySlices("Test", 0, maxSlice, NoOffset)
        .collect { case u: UpdatedDurableState[Record] => u }
        .runWith(TestSink[UpdatedDurableState[Record]]())

      val firstStateChange = testSink.request(1).expectNext()
      firstStateChange.value should be(record)
      firstStateChange.revision should be(1L)

      stateStore.upsertObject("Test|record-1", 2L, recordChange, "")
      val secondStateChange = testSink.request(1).expectNext()
      secondStateChange.value should be(recordChange)
      secondStateChange.revision should be(2L)
      secondStateChange.offset
        .asInstanceOf[Sequence]
        .value should be >= (firstStateChange.offset.asInstanceOf[Sequence].value)
    }

    "find current state changes by slice ordered by upsert" in {
      val stateStore = new PersistenceTestKitDurableStateStore[Record](classic.asInstanceOf[ExtendedActorSystem])
      val record = Record(1, "name-1")

      stateStore.upsertObject("Test|record-1", 1L, record, "")

      val maxSlice = stateStore.sliceRanges(1).head.max
      val testSinkCurrentChanges =
        stateStore
          .currentChangesBySlices("Test", 0, maxSlice, NoOffset)
          .collect { case u: UpdatedDurableState[Record] => u }
          .runWith(TestSink[UpdatedDurableState[Record]]())

      stateStore.upsertObject("record-1", 2L, record.copy(name = "my-name-1-2"), "")
      stateStore.upsertObject("record-1", 3L, record.copy(name = "my-name-1-3"), "")

      val currentStateChange = testSinkCurrentChanges.request(1).expectNext()

      currentStateChange.value should be(record)
      currentStateChange.revision should be(1L)
      testSinkCurrentChanges.request(1).expectComplete()

      val testSinkIllegalOffset =
        stateStore
          .currentChangesBySlices("Test", 0, maxSlice, Sequence(100L))
          .collect { case u: UpdatedDurableState[Record] => u }
          .runWith(TestSink[UpdatedDurableState[Record]]())
      testSinkIllegalOffset.request(1).expectNoMessage()
    }

    "return current changes by slice when there are no further changes" in {
      val stateStore = new PersistenceTestKitDurableStateStore[Record](classic.asInstanceOf[ExtendedActorSystem])
      val record = Record(1, "name-1")
      val tag = "tag-1"

      stateStore.upsertObject("Test|record-1", 1L, record, tag)

      val maxSlice = stateStore.sliceRanges(1).head.max
      val testSinkCurrentChanges =
        stateStore
          .currentChangesBySlices("Test", 0, maxSlice, NoOffset)
          .collect { case u: UpdatedDurableState[Record] => u }
          .runWith(TestSink[UpdatedDurableState[Record]]())

      val currentStateChange = testSinkCurrentChanges.request(1).expectNext()

      currentStateChange.value should be(record)
      currentStateChange.revision should be(1L)
      testSinkCurrentChanges.request(1).expectComplete()
    }

    "return all current persistence ids" in {
      val stateStore = new PersistenceTestKitDurableStateStore[Record](classic.asInstanceOf[ExtendedActorSystem])

      val record = Record(1, "name-1")
      val tag = "tag-1"
      val ids = (1 to 20).map(i => s"id-$i")
      for (id <- ids) {
        stateStore.upsertObject(id, 1, record, tag)
      }

      val resultIds = stateStore.currentPersistenceIds(None, Long.MaxValue).runWith(Sink.seq).futureValue
      resultIds should have size 20
      resultIds should contain allElementsOf ids
    }

    "return paged current persistence ids" in {
      val stateStore = new PersistenceTestKitDurableStateStore[Record](classic.asInstanceOf[ExtendedActorSystem])

      val record = Record(1, "name-1")
      val tag = "tag-1"
      val ids = (1 to 20).map(i => s"id-$i")
      for (id <- ids) {
        stateStore.upsertObject(id, 1, record, tag)
      }

      val page1 = stateStore.currentPersistenceIds(None, 9).runWith(Sink.seq).futureValue
      page1 should have size 9
      val page2 = stateStore.currentPersistenceIds(page1.lastOption, 9).runWith(Sink.seq).futureValue
      page2 should have size 9
      val page3 = stateStore.currentPersistenceIds(page2.lastOption, 9).runWith(Sink.seq).futureValue
      page3 should have size 2

      (page1 ++ page2 ++ page3) should contain allElementsOf ids
    }
  }
}
