# Copyright 2017-2020 Palantir Technologies, Inc.
# Copyright 2021- Python Language Server Contributors.

import os

import pytest

from pylsp import uris
from pylsp.plugins.references import pylsp_references
from pylsp.workspace import Document

DOC1_NAME = "test1.py"
DOC2_NAME = "test2.py"

DOC1 = """class Test1():
    pass
"""

DOC2 = """from test1 import Test1

try:
    Test1()
except UnicodeError:
    pass
"""


@pytest.fixture
def tmp_workspace(temp_workspace_factory):
    return temp_workspace_factory(
        {
            DOC1_NAME: DOC1,
            DOC2_NAME: DOC2,
        }
    )


def test_references(tmp_workspace) -> None:
    # Over 'Test1' in class Test1():
    position = {"line": 0, "character": 8}
    DOC1_URI = uris.from_fs_path(os.path.join(tmp_workspace.root_path, DOC1_NAME))
    doc1 = Document(DOC1_URI, tmp_workspace)

    refs = pylsp_references(doc1, position, exclude_declaration=False)

    # Definition, the import and the instantiation
    assert len(refs) == 3

    # Briefly check excluding the definitions (also excludes imports, only counts uses)
    no_def_refs = pylsp_references(doc1, position, exclude_declaration=True)
    assert len(no_def_refs) == 1

    # Make sure our definition is correctly located
    doc1_ref = [u for u in refs if u["uri"] == DOC1_URI][0]
    assert doc1_ref["range"]["start"] == {"line": 0, "character": 6}
    assert doc1_ref["range"]["end"] == {"line": 0, "character": 11}

    # Make sure our import is correctly located
    doc2_import_ref = [u for u in refs if u["uri"] != DOC1_URI][0]
    assert doc2_import_ref["range"]["start"] == {"line": 0, "character": 18}
    assert doc2_import_ref["range"]["end"] == {"line": 0, "character": 23}

    doc2_usage_ref = [u for u in refs if u["uri"] != DOC1_URI][1]
    assert doc2_usage_ref["range"]["start"] == {"line": 3, "character": 4}
    assert doc2_usage_ref["range"]["end"] == {"line": 3, "character": 9}


def test_references_builtin(tmp_workspace) -> None:
    # Over 'UnicodeError':
    position = {"line": 4, "character": 7}
    doc2_uri = uris.from_fs_path(os.path.join(str(tmp_workspace.root_path), DOC2_NAME))
    doc2 = Document(doc2_uri, tmp_workspace)

    refs = pylsp_references(doc2, position, exclude_declaration=False)
    assert len(refs) >= 1

    expected = {
        "start": {"line": 4, "character": 7},
        "end": {"line": 4, "character": 19},
    }
    ranges = [r["range"] for r in refs]
    assert expected in ranges