# 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