evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Allow templates in rest_command headers (#26099)

Browse source
This commit is contained in:
Pedro Lamas 2019-11-26 03:39:56 +00:00 • committed by Paulus Schoutsen
parent cc255da038
commit db0008e62c
2 changed files with 50 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

32
tests/components/rest_command/test_init.py
View file

@ -236,6 +236,19 @@ class TestRestCommandComponent:
},
"content_type": "text/plain",
},
"headers_template_test": {
"headers": {
"Accept": "application/json",
"User-Agent": "Mozilla/{{ 3 + 2 }}.0",
}
},
"headers_and_content_type_override_template_test": {
"headers": {
"Accept": "application/{{ 1 + 1 }}json",
aiohttp.hdrs.CONTENT_TYPE: "application/pdf",
},
"content_type": "text/json",
},
}
}
@ -245,7 +258,7 @@ class TestRestCommandComponent:
{"url": self.url, "method": "post", "payload": "test data"}
)
with assert_setup_component(5):
with assert_setup_component(7):
setup_component(self.hass, rc.DOMAIN, header_config_variations)
# provide post request data
@ -257,11 +270,13 @@ class TestRestCommandComponent:
"headers_test",
"headers_and_content_type_test",
"headers_and_content_type_override_test",
"headers_template_test",
"headers_and_content_type_override_template_test",
]:
self.hass.services.call(rc.DOMAIN, test_service, {})
self.hass.block_till_done()
assert len(aioclient_mock.mock_calls) == 5
assert len(aioclient_mock.mock_calls) == 7
# no_headers_test
assert aioclient_mock.mock_calls[0][3] is None
@ -293,3 +308,16 @@ class TestRestCommandComponent:
== "text/plain"
)
assert aioclient_mock.mock_calls[4][3].get("Accept") == "application/json"
# headers_template_test
assert len(aioclient_mock.mock_calls[5][3]) == 2
assert aioclient_mock.mock_calls[5][3].get("Accept") == "application/json"
assert aioclient_mock.mock_calls[5][3].get("User-Agent") == "Mozilla/5.0"
# headers_and_content_type_override_template_test
assert len(aioclient_mock.mock_calls[6][3]) == 2
assert (
aioclient_mock.mock_calls[6][3].get(aiohttp.hdrs.CONTENT_TYPE)
== "text/json"
)
assert aioclient_mock.mock_calls[6][3].get("Accept") == "application/2json"