FilesExpand file tree

 main

/

test_main.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

100 lines (83 loc) · 3.35 KB

 main

/

test_main.py

Copy path

Top

File metadata and controls

• Code
• Blame

100 lines (83 loc) · 3.35 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

87

88

89

90

91

92

93

94

95

96

97

98

99

100

"""

SPDX-License-Identifier: Apache-2.0

The OpenSearch Contributors require contributions made to

this file be licensed under the Apache-2.0 license or a

compatible open source license.

Modifications Copyright OpenSearch Contributors. See

GitHub history for details.

"""

"""

Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.

Licensed under the Apache License, Version 2.0 (the "License").

You may not use this file except in compliance with the License.

A copy of the License is located at

http://www.apache.org/licenses/LICENSE-2.0

or in the "license" file accompanying this file. This file is distributed

on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either

express or implied. See the License for the specific language governing

permissions and limitations under the License.

"""

import mock

from textwrap import dedent

from click.testing import CliRunner

from .utils import estest, load_data, TEST_INDEX_NAME

from src.opensearch_sql_cli.main import cli

from src.opensearch_sql_cli.opensearchsql_cli import OpenSearchSqlCli

INVALID_ENDPOINT = "http://invalid:9200"

ENDPOINT = "http://localhost:9200"

QUERY = "select * from %s" % TEST_INDEX_NAME

class TestMain:

@estest

def test_explain(self, connection):

doc = {"a": "aws"}

load_data(connection, doc)

err_message = "Can not connect to endpoint %s" % INVALID_ENDPOINT

expected_output = {

"root": {

"name": "ProjectOperator",

"description": {"fields": "[a]"},

"children": [

{

"name": "OpenSearchIndexScan",

"description": {

"request": 'OpenSearchQueryRequest(indexName=opensearchsql_cli_test, sourceBuilder={"from":0,"size":200,"timeout":"1m","_source":{"includes":["a"],"excludes":[]}}, searchDone=false)'

},

"children": [],

}

],

}

}

expected_tabular_output = dedent(

"""\

fetched rows / total rows = 1/1

+-----+

| a |

|-----|

| aws |

+-----+"""

)

with mock.patch("src.opensearch_sql_cli.main.click.echo") as mock_echo, mock.patch(

"src.opensearch_sql_cli.main.click.secho"

) as mock_secho:

runner = CliRunner()

# test -q -e

result = runner.invoke(cli, [f"-q{QUERY}", "-e"])

mock_echo.assert_called_with(expected_output)

assert result.exit_code == 0

# test -q

result = runner.invoke(cli, [f"-q{QUERY}"])

mock_echo.assert_called_with(expected_tabular_output)

assert result.exit_code == 0

# test invalid endpoint

runner.invoke(cli, [INVALID_ENDPOINT, f"-q{QUERY}", "-e"])

mock_secho.assert_called_with(message=err_message, fg="red")

@estest

def test_cli(self):

with mock.patch.object(OpenSearchSqlCli, "connect") as mock_connect, mock.patch.object(

OpenSearchSqlCli, "run_cli"

) as mock_run_cli:

runner = CliRunner()

result = runner.invoke(cli)

mock_connect.assert_called_with(ENDPOINT, None)

mock_run_cli.asset_called()

assert result.exit_code == 0