FilesExpand file tree

develop

/

test_language_server.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

86 lines (68 loc) · 2.62 KB

develop

/

test_language_server.py

Copy path

Top

File metadata and controls

• Code
• Blame

86 lines (68 loc) · 2.62 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

# Copyright 2017-2020 Palantir Technologies, Inc.

# Copyright 2021- Python Language Server Contributors.

import os

import sys

import time

import pytest

from flaky import flaky

from pylsp_jsonrpc.exceptions import JsonRpcMethodNotFound

from test.test_utils import ClientServerPair, send_initialize_request

RUNNING_IN_CI = bool(os.environ.get("CI"))

CALL_TIMEOUT_IN_SECONDS = 10

@pytest.fixture

def client_exited_server() -> None:

"""A fixture that sets up a client/server pair that support checking parent process aliveness

and assert the server has already exited

"""

client_server_pair_obj = ClientServerPair(True, True)

yield client_server_pair_obj

assert client_server_pair_obj.server_process.is_alive() is False

@flaky(max_runs=10, min_passes=1)

@pytest.mark.skipif(sys.platform == "darwin", reason="Too flaky on Mac")

def test_initialize(client_server_pair) -> None:

client, _ = client_server_pair

response = client._endpoint.request(

"initialize",

{"rootPath": os.path.dirname(__file__), "initializationOptions": {}},

).result(timeout=CALL_TIMEOUT_IN_SECONDS)

assert "capabilities" in response

@flaky(max_runs=10, min_passes=1)

@pytest.mark.skipif(

not sys.platform.startswith("Linux"), reason="Skipped on win and flaky on mac"

)

def test_exit_with_parent_process_died(

client_exited_server,

) -> None:

# language server should have already exited before responding

lsp_server, mock_process = (

client_exited_server.client,

client_exited_server.server_process,

)

# with pytest.raises(Exception):

lsp_server._endpoint.request(

"initialize",

{

"processId": mock_process.pid,

"rootPath": os.path.dirname(__file__),

"initializationOptions": {},

},

).result(timeout=CALL_TIMEOUT_IN_SECONDS)

mock_process.terminate()

time.sleep(CALL_TIMEOUT_IN_SECONDS)

assert not client_exited_server.client_thread.is_alive()

@flaky(max_runs=10, min_passes=1)

@pytest.mark.skipif(sys.platform.startswith("linux"), reason="Fails on linux")

def test_not_exit_without_check_parent_process_flag(

client_server_pair,

) -> None:

client, _ = client_server_pair

response = send_initialize_request(client)

assert "capabilities" in response

@flaky(max_runs=10, min_passes=1)

@pytest.mark.skipif(RUNNING_IN_CI, reason="This test is hanging on CI")

def test_missing_message(client_server_pair) -> None:

client, _ = client_server_pair

with pytest.raises(JsonRpcMethodNotFound):

client._endpoint.request("unknown_method").result(

timeout=CALL_TIMEOUT_IN_SECONDS

)