// Copyright 2000-2024 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
package com.intellij.cce.actions

import com.intellij.cce.core.*
import org.junit.jupiter.api.Assertions.assertEquals
import org.junit.jupiter.api.Assertions.assertTrue
import org.junit.jupiter.api.Test

class ActionSerializerTest {
  @Test
  fun `empty actions`() {
    doTest(withActions(0) {})
  }

  @Test
  fun `with properties`() {
    val properties = SimpleTokenProperties.create(TypeProperty.METHOD_CALL, SymbolLocation.UNKNOWN) {}
    doTest(withActions(1) { session { callFeature("foo", 0, properties) } })
  }

  @Test
  fun `with java properties`() {
    val properties = JvmProperties.create(TypeProperty.METHOD_CALL, SymbolLocation.PROJECT) {
      isStatic = true
      packageName = "java.util"
    }
    val after = doTest(withActions(1) { session { callFeature("foo", 0, properties) } })
    val javaProperties = PropertyAdapters.Jvm.adapt((after.actions[0] as CallFeature).nodeProperties)
    assertTrue(javaProperties?.isStatic!!)
    assertEquals("java.util", javaProperties.packageName)
  }

  @Test
  fun `with all actions`() {
    val props = SimpleTokenProperties.create(TypeProperty.METHOD_CALL, SymbolLocation.UNKNOWN) {}
    doTest(withActions(1) { session {
      moveCaret(10)
      deleteRange(10, 20)
      printText("Hello")
      callCompletion(props) }
    })
  }

  @Test
  fun `properties with custom value`() {
    val props = SimpleTokenProperties.create(TypeProperty.METHOD_CALL, SymbolLocation.UNKNOWN) {
      put("custom", "42")
    }
    val action = doTest(withActions(1) { session { callCompletion(props) } }).actions[0]
    assert((action as CallFeature).nodeProperties.additionalProperty("custom") == "42")
  }

  private fun ActionsBuilder.SessionBuilder.callCompletion(properties: TokenProperties) {
    callFeature("foo", 0, properties)
  }

  private fun withActions(sessionsCount: Int, init: ActionsBuilder.() -> Unit): FileActions {
    return FileActions("foo/bar", "42", sessionsCount, ActionsBuilder().apply(init).build())
  }

  private fun doTest(before: FileActions): FileActions {
    val after = ActionSerializer.deserialize(ActionSerializer.serialize(before))
    assertEquals(before.path, after.path)
    assertEquals(before.sessionsCount, after.sessionsCount)
    assertEquals(before.checksum, after.checksum)
    assertActionsEquals(before.actions, after.actions)
    return after
  }

  private fun assertActionsEquals(before: List<Action>, after: List<Action>) {
    assertEquals(before.size, after.size)
    for ((actionBefore, actionAfter) in before.zip(after)) {
      assertEquals(actionBefore.type, actionAfter.type)
      if (actionBefore is CallFeature) {
        require(actionAfter is CallFeature)
        assertEquals(actionBefore.expectedText, actionAfter.expectedText)
        assertEquals(actionBefore.offset, actionAfter.offset)
        assertEquals(actionBefore.nodeProperties.describe(), actionAfter.nodeProperties.describe())
      }
      else {
        assertEquals(actionBefore, actionAfter)
      }
    }
  }
}