package com.squareup.leakcanary.deobfuscation

import org.assertj.core.api.Assertions.assertThat
import org.assertj.core.api.Assertions.assertThatExceptionOfType
import org.gradle.api.GradleException
import org.gradle.testfixtures.ProjectBuilder
import org.junit.Rule
import org.junit.Test
import org.junit.rules.TemporaryFolder
import java.io.File

class CopyObfuscationMappingFileTaskTest {

  @get:Rule
  val tempFolder = TemporaryFolder()

  private val task =
    ProjectBuilder
      .builder()
      .build()
      .tasks
      .create(
        "testCopyObfuscationMappingFileTask",
        CopyObfuscationMappingFileTask::class.java
      )

  @Test
  fun `should throw if mapping file and merge assets dir are not specified`() {
    assertThatExceptionOfType(GradleException::class.java)
      .isThrownBy { task.copyObfuscationMappingFile() }
  }

  @Test
  fun `should throw if mapping file is not specified`() {
    task.mergeAssetsDirectory.set(tempFolder.newFolder("mergeAssetsDir"))

    assertThatExceptionOfType(GradleException::class.java)
      .isThrownBy { task.copyObfuscationMappingFile() }
  }

  @Test
  fun `should throw if merge assets dir is not specified`() {
    task.mappingFile.set(tempFolder.newFile("mapping.txt"))

    assertThatExceptionOfType(GradleException::class.java)
      .isThrownBy { task.copyObfuscationMappingFile() }
  }

  @Test
  fun `existing mapping copied and merge assets dir generated if not exists`() {
    task.mappingFile.set(tempFolder.newFile("mapping.txt"))
    task.mergeAssetsDirectory.set(File(tempFolder.root, "mergeAssetsDir"))

    assertThat(task.mergeAssetsDirectory.get().exists()).isFalse()

    task.copyObfuscationMappingFile()

    assertThat(task.mergeAssetsDirectory.get().exists()).isTrue()
    assertThat(task.leakCanaryAssetsOutputFile.exists()).isTrue()
  }

  @Test
  fun `previous mapping overwritten`() {
    task.mergeAssetsDirectory.set(tempFolder.newFolder("mergeAssetsDir"))

    // create first mapping file
    task.mappingFile.set(tempFolder.newFile("firstMappingFile.txt")
      .apply {
        writeText("firstMappingFile")
      })
    task.copyObfuscationMappingFile()

    // create second mapping file
    task.mappingFile.set(tempFolder.newFile("secondMappingFile.txt")
      .apply {
        writeText("secondMappingFile")
      })
    task.copyObfuscationMappingFile()

    assertThat(task.leakCanaryAssetsOutputFile.exists()).isTrue()
    assertThat(task.leakCanaryAssetsOutputFile.readText()).isEqualTo("secondMappingFile")
  }
}