Skip to content

Commit 8925f59

Browse files
committed
Added test to demo setting user agent at client level
1 parent ff117d8 commit 8925f59

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

test_elasticsearch/test_transport.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,18 @@ def test_request_with_custom_user_agent_header(self):
223223
assert calls[0][0] == ("GET", "/")
224224
assert calls[0][1]["headers"]["user-agent"] == "my-custom-value/1.2.3"
225225

226+
def test_request_with_custom_user_agent_header_set_at_client_level(self):
227+
client = Elasticsearch(
228+
"http://localhost:9200", meta_header=False, node_class=DummyNode,
229+
headers={"User-Agent": "my-custom-value/1.2.3"},
230+
)
231+
232+
client.info()
233+
calls = client.transport.node_pool.get().calls
234+
assert 1 == len(calls)
235+
assert calls[0][0] == ("GET", "/")
236+
assert calls[0][1]["headers"]["user-agent"] == "my-custom-value/1.2.3"
237+
226238
def test_client_meta_header(self):
227239
client = Elasticsearch("http://localhost:9200", node_class=DummyNode)
228240
client.info()

0 commit comments

Comments
 (0)