test_that("Document highlight works", {
    skip_on_cran()
    client <- language_client()

    temp_file <- withr::local_tempfile(fileext = ".R")
    writeLines(
        c(
            "my_fn <- 3",
            "my_fn",
            "ggplot2::ggplot",
            "ggplot2::geom_abline"
        ),
        temp_file)

    client %>% did_open(temp_file)

    # query at the beginning of token
    result <- client %>% respond_document_highlight(temp_file, c(1, 0))

    expect_length(result, 2)
    expect_equal(result[[1]]$range$start, list(line = 0, character = 0))
    expect_equal(result[[1]]$range$end, list(line = 0, character = 5))
    expect_equal(result[[2]]$range$start, list(line = 1, character = 0))
    expect_equal(result[[2]]$range$end, list(line = 1, character = 5))

    # query in the middle of token
    result <- client %>% respond_document_highlight(temp_file, c(1, 3))

    expect_length(result, 2)
    expect_equal(result[[1]]$range$start, list(line = 0, character = 0))
    expect_equal(result[[1]]$range$end, list(line = 0, character = 5))
    expect_equal(result[[2]]$range$start, list(line = 1, character = 0))
    expect_equal(result[[2]]$range$end, list(line = 1, character = 5))

     # query at the end of token
    result <- client %>% respond_document_highlight(temp_file, c(1, 5))

    expect_length(result, 2)
    expect_equal(result[[1]]$range$start, list(line = 0, character = 0))
    expect_equal(result[[1]]$range$end, list(line = 0, character = 5))
    expect_equal(result[[2]]$range$start, list(line = 1, character = 0))
    expect_equal(result[[2]]$range$end, list(line = 1, character = 5))


    result <- client %>% respond_document_highlight(temp_file, c(3, 0))

    expect_length(result, 2)
    expect_equal(result[[1]]$range$start, list(line = 2, character = 0))
    expect_equal(result[[1]]$range$end, list(line = 2, character = 7))
    expect_equal(result[[2]]$range$start, list(line = 3, character = 0))
    expect_equal(result[[2]]$range$end, list(line = 3, character = 7))

})

test_that("Document highlight works in Rmarkdown", {
    skip_on_cran()
    client <- language_client()

    temp_file <- withr::local_tempfile(fileext = ".Rmd")
    writeLines(
        c(
            "---",
            "title: r markdown",
            "---",
            "my_fn and ggplot2 are defined",
            "```{r}",
            "my_fn <- 3",
            "my_fn",
            "ggplot2::ggplot",
            "ggplot2::geom_abline",
            "```"
        ),
        temp_file
    )

    client %>% did_open(temp_file)

    result <- client %>% respond_document_highlight(temp_file, c(6, 0))

    expect_length(result, 2)
    expect_equal(result[[1]]$range$start, list(line = 5, character = 0))
    expect_equal(result[[1]]$range$end, list(line = 5, character = 5))
    expect_equal(result[[2]]$range$start, list(line = 6, character = 0))
    expect_equal(result[[2]]$range$end, list(line = 6, character = 5))


    result <- client %>% respond_document_highlight(temp_file, c(8, 0))

    expect_length(result, 2)
    expect_equal(result[[1]]$range$start, list(line = 7, character = 0))
    expect_equal(result[[1]]$range$end, list(line = 7, character = 7))
    expect_equal(result[[2]]$range$start, list(line = 8, character = 0))
    expect_equal(result[[2]]$range$end, list(line = 8, character = 7))
})