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
|
# Licensed to Elasticsearch B.V. under one or more contributor
# license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright
# ownership. Elasticsearch B.V. licenses this file to you under
# the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License 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.
from collections import defaultdict
from elastic_transport import ApiResponseMeta, HttpHeaders
from elasticsearch import Elasticsearch
class DummyTransport:
def __init__(self, hosts, responses=None, **_):
self.hosts = hosts
self.responses = responses
self.call_count = 0
self.calls = defaultdict(list)
def perform_request(self, method, target, **kwargs):
status, resp = 200, {}
if self.responses:
status, resp = self.responses[self.call_count]
self.call_count += 1
self.calls[(method, target)].append(kwargs)
return (
ApiResponseMeta(
status=status,
http_version="1.1",
headers=HttpHeaders({"X-elastic-product": "Elasticsearch"}),
duration=0.0,
node=None,
),
resp,
)
class DummyAsyncTransport:
def __init__(self, hosts, responses=None, **_):
self.hosts = hosts
self.responses = responses
self.call_count = 0
self.calls = defaultdict(list)
async def perform_request(self, method, target, **kwargs):
status, resp = 200, {}
if self.responses:
status, resp = self.responses[self.call_count]
self.call_count += 1
self.calls[(method, target)].append(kwargs)
return (
ApiResponseMeta(
status=status,
http_version="1.1",
headers=HttpHeaders({"X-elastic-product": "Elasticsearch"}),
duration=0.0,
node=None,
),
resp,
)
class DummyTransportTestCase:
def setup_method(self, _):
self.client = Elasticsearch(
"http://localhost:9200", transport_class=DummyTransport
)
def assert_call_count_equals(self, count):
assert count == self.client.transport.call_count
def assert_url_called(self, method, url, count=1):
assert (method, url) in self.client.transport.calls
calls = self.client.transport.calls[(method, url)]
assert count == len(calls)
return calls
|
