Move components to folders (#20774)
* Move all components into folders * Move component tests into folders * Fix init moving * Move tests * Lint * Update coverage * Fix service descriptions * Update CODEOWNERS
This commit is contained in:
Paulus Schoutsen
GitHub
parent
d13b2ca6ef
commit
b8cc547fa3
275 changed files with 512 additions and 684 deletions
303
tests/components/rest_command/test_init.py
Normal file
303
tests/components/rest_command/test_init.py
Normal file
|
|
@ -0,0 +1,303 @@
|
|||
"""The tests for the rest command platform."""
|
||||
import asyncio
|
||||
|
||||
import aiohttp
|
||||
|
||||
import homeassistant.components.rest_command as rc
|
||||
from homeassistant.setup import setup_component
|
||||
|
||||
from tests.common import (
|
||||
get_test_home_assistant, assert_setup_component)
|
||||
|
||||
|
||||
class TestRestCommandSetup:
|
||||
"""Test the rest command component."""
|
||||
|
||||
def setup_method(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.hass = get_test_home_assistant()
|
||||
|
||||
self.config = {
|
||||
rc.DOMAIN: {'test_get': {
|
||||
'url': 'http://example.com/'
|
||||
}}
|
||||
}
|
||||
|
||||
def teardown_method(self):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
|
||||
def test_setup_component(self):
|
||||
"""Test setup component."""
|
||||
with assert_setup_component(1):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
def test_setup_component_timeout(self):
|
||||
"""Test setup component timeout."""
|
||||
self.config[rc.DOMAIN]['test_get']['timeout'] = 10
|
||||
|
||||
with assert_setup_component(1):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
def test_setup_component_test_service(self):
|
||||
"""Test setup component and check if service exits."""
|
||||
with assert_setup_component(1):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
assert self.hass.services.has_service(rc.DOMAIN, 'test_get')
|
||||
|
||||
|
||||
class TestRestCommandComponent:
|
||||
"""Test the rest command component."""
|
||||
|
||||
def setup_method(self):
|
||||
"""Set up things to be run when tests are started."""
|
||||
self.url = "https://example.com/"
|
||||
self.config = {
|
||||
rc.DOMAIN: {
|
||||
'get_test': {
|
||||
'url': self.url,
|
||||
'method': 'get',
|
||||
},
|
||||
'post_test': {
|
||||
'url': self.url,
|
||||
'method': 'post',
|
||||
},
|
||||
'put_test': {
|
||||
'url': self.url,
|
||||
'method': 'put',
|
||||
},
|
||||
'delete_test': {
|
||||
'url': self.url,
|
||||
'method': 'delete',
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
self.hass = get_test_home_assistant()
|
||||
|
||||
def teardown_method(self):
|
||||
"""Stop everything that was started."""
|
||||
self.hass.stop()
|
||||
|
||||
def test_setup_tests(self):
|
||||
"""Set up test config and test it."""
|
||||
with assert_setup_component(4):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
assert self.hass.services.has_service(rc.DOMAIN, 'get_test')
|
||||
assert self.hass.services.has_service(rc.DOMAIN, 'post_test')
|
||||
assert self.hass.services.has_service(rc.DOMAIN, 'put_test')
|
||||
assert self.hass.services.has_service(rc.DOMAIN, 'delete_test')
|
||||
|
||||
def test_rest_command_timeout(self, aioclient_mock):
|
||||
"""Call a rest command with timeout."""
|
||||
with assert_setup_component(4):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
aioclient_mock.get(self.url, exc=asyncio.TimeoutError())
|
||||
|
||||
self.hass.services.call(rc.DOMAIN, 'get_test', {})
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
def test_rest_command_aiohttp_error(self, aioclient_mock):
|
||||
"""Call a rest command with aiohttp exception."""
|
||||
with assert_setup_component(4):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
aioclient_mock.get(self.url, exc=aiohttp.ClientError())
|
||||
|
||||
self.hass.services.call(rc.DOMAIN, 'get_test', {})
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
def test_rest_command_http_error(self, aioclient_mock):
|
||||
"""Call a rest command with status code 400."""
|
||||
with assert_setup_component(4):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
aioclient_mock.get(self.url, status=400)
|
||||
|
||||
self.hass.services.call(rc.DOMAIN, 'get_test', {})
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
def test_rest_command_auth(self, aioclient_mock):
|
||||
"""Call a rest command with auth credential."""
|
||||
data = {
|
||||
'username': 'test',
|
||||
'password': '123456',
|
||||
}
|
||||
self.config[rc.DOMAIN]['get_test'].update(data)
|
||||
|
||||
with assert_setup_component(4):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
aioclient_mock.get(self.url, content=b'success')
|
||||
|
||||
self.hass.services.call(rc.DOMAIN, 'get_test', {})
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
def test_rest_command_form_data(self, aioclient_mock):
|
||||
"""Call a rest command with post form data."""
|
||||
data = {
|
||||
'payload': 'test'
|
||||
}
|
||||
self.config[rc.DOMAIN]['post_test'].update(data)
|
||||
|
||||
with assert_setup_component(4):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
aioclient_mock.post(self.url, content=b'success')
|
||||
|
||||
self.hass.services.call(rc.DOMAIN, 'post_test', {})
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
assert aioclient_mock.mock_calls[0][2] == b'test'
|
||||
|
||||
def test_rest_command_get(self, aioclient_mock):
|
||||
"""Call a rest command with get."""
|
||||
with assert_setup_component(4):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
aioclient_mock.get(self.url, content=b'success')
|
||||
|
||||
self.hass.services.call(rc.DOMAIN, 'get_test', {})
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
def test_rest_command_delete(self, aioclient_mock):
|
||||
"""Call a rest command with delete."""
|
||||
with assert_setup_component(4):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
aioclient_mock.delete(self.url, content=b'success')
|
||||
|
||||
self.hass.services.call(rc.DOMAIN, 'delete_test', {})
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
|
||||
def test_rest_command_post(self, aioclient_mock):
|
||||
"""Call a rest command with post."""
|
||||
data = {
|
||||
'payload': 'data',
|
||||
}
|
||||
self.config[rc.DOMAIN]['post_test'].update(data)
|
||||
|
||||
with assert_setup_component(4):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
aioclient_mock.post(self.url, content=b'success')
|
||||
|
||||
self.hass.services.call(rc.DOMAIN, 'post_test', {})
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
assert aioclient_mock.mock_calls[0][2] == b'data'
|
||||
|
||||
def test_rest_command_put(self, aioclient_mock):
|
||||
"""Call a rest command with put."""
|
||||
data = {
|
||||
'payload': 'data',
|
||||
}
|
||||
self.config[rc.DOMAIN]['put_test'].update(data)
|
||||
|
||||
with assert_setup_component(4):
|
||||
setup_component(self.hass, rc.DOMAIN, self.config)
|
||||
|
||||
aioclient_mock.put(self.url, content=b'success')
|
||||
|
||||
self.hass.services.call(rc.DOMAIN, 'put_test', {})
|
||||
self.hass.block_till_done()
|
||||
|
||||
assert len(aioclient_mock.mock_calls) == 1
|
||||
assert aioclient_mock.mock_calls[0][2] == b'data'
|
||||
|
||||
def test_rest_command_headers(self, aioclient_mock):
|
||||
"""Call a rest command with custom headers and content types."""
|
||||
header_config_variations = {
|
||||
rc.DOMAIN: {
|
||||
'no_headers_test': {},
|
||||
'content_type_test': {
|
||||
'content_type': 'text/plain'
|
||||
},
|
||||
'headers_test': {
|
||||
'headers': {
|
||||
'Accept': 'application/json',
|
||||
'User-Agent': 'Mozilla/5.0'
|
||||
}
|
||||
},
|
||||
'headers_and_content_type_test': {
|
||||
'headers': {
|
||||
'Accept': 'application/json'
|
||||
},
|
||||
'content_type': 'text/plain'
|
||||
},
|
||||
'headers_and_content_type_override_test': {
|
||||
'headers': {
|
||||
'Accept': 'application/json',
|
||||
aiohttp.hdrs.CONTENT_TYPE: 'application/pdf'
|
||||
},
|
||||
'content_type': 'text/plain'
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
# add common parameters
|
||||
for variation in header_config_variations[rc.DOMAIN].values():
|
||||
variation.update({'url': self.url, 'method': 'post',
|
||||
'payload': 'test data'})
|
||||
|
||||
with assert_setup_component(5):
|
||||
setup_component(self.hass, rc.DOMAIN, header_config_variations)
|
||||
|
||||
# provide post request data
|
||||
aioclient_mock.post(self.url, content=b'success')
|
||||
|
||||
for test_service in ['no_headers_test',
|
||||
'content_type_test',
|
||||
'headers_test',
|
||||
'headers_and_content_type_test',
|
||||
'headers_and_content_type_override_test']:
|
||||
self.hass.services.call(rc.DOMAIN, test_service, {})
|
||||
|
||||
self.hass.block_till_done()
|
||||
assert len(aioclient_mock.mock_calls) == 5
|
||||
|
||||
# no_headers_test
|
||||
assert aioclient_mock.mock_calls[0][3] is None
|
||||
|
||||
# content_type_test
|
||||
assert len(aioclient_mock.mock_calls[1][3]) == 1
|
||||
assert aioclient_mock.mock_calls[1][3].get(
|
||||
aiohttp.hdrs.CONTENT_TYPE) == 'text/plain'
|
||||
|
||||
# headers_test
|
||||
assert len(aioclient_mock.mock_calls[2][3]) == 2
|
||||
assert aioclient_mock.mock_calls[2][3].get(
|
||||
'Accept') == 'application/json'
|
||||
assert aioclient_mock.mock_calls[2][3].get(
|
||||
'User-Agent') == 'Mozilla/5.0'
|
||||
|
||||
# headers_and_content_type_test
|
||||
assert len(aioclient_mock.mock_calls[3][3]) == 2
|
||||
assert aioclient_mock.mock_calls[3][3].get(
|
||||
aiohttp.hdrs.CONTENT_TYPE) == 'text/plain'
|
||||
assert aioclient_mock.mock_calls[3][3].get(
|
||||
'Accept') == 'application/json'
|
||||
|
||||
# headers_and_content_type_override_test
|
||||
assert len(aioclient_mock.mock_calls[4][3]) == 2
|
||||
assert aioclient_mock.mock_calls[4][3].get(
|
||||
aiohttp.hdrs.CONTENT_TYPE) == 'text/plain'
|
||||
assert aioclient_mock.mock_calls[4][3].get(
|
||||
'Accept') == 'application/json'
|
||||
Loading…
Add table
Add a link
Reference in a new issue