First unittest for view

homeassistant/components/hassio.py

@@ -80,9 +80,13 @@ def async_setup(hass, config):
         _LOGGER.error("No HassIO supervisor detect!")
         return False
 
-    # register api views
-    for base in ('host', 'supervisor', 'network', 'homeassistant'):
+    # register base api views
+    for base in ('host', 'homeassistant'):
         hass.http.register_view(HassIOBaseView(hassio, base))
+    for base in ('supervisor', 'network'):
+        hass.http.register_view(HassIOBaseEditView(hassio, base))
+
+    # register view for addons
     hass.http.register_view(HassIOAddonsView(hassio))
 
     @asyncio.coroutine
@@ -193,7 +197,6 @@ def __init__(self, hassio, base):
         """Initialize a hassio base view."""
         self.hassio = hassio
         self._url_info = "/{}/info".format(base)
-        self._url_options = "/{}/options".format(base)
 
         self.url = "/api/hassio/{}".format(base)
         self.name = "api:hassio:{}".format(base)
@@ -206,6 +209,15 @@ def get(self, request):
             raise HTTPBadGateway()
         return web.json_response(data)
 
+
+class HassIOBaseEditView(HassIOBaseView):
+    """HassIO view to handle base with options support."""
+
+    def __init__(self, hassio, base):
+        """Initialize a hassio base edit view."""
+        super().__init__(hassio, base)
+        self._url_options = "/{}/options".format(base)
+
     @asyncio.coroutine
     def post(self, request):
         """Set options on host."""

tests/components/test_hassio.py

@@ -5,7 +5,7 @@
 import aiohttp
 
 import homeassistant.components.hassio as ho
-from homeassistant.setup import setup_component
+from homeassistant.setup import setup_component, async_setup_component
 
 from tests.common import (
     get_test_home_assistant, assert_setup_component)
@@ -292,3 +292,35 @@ def test_rest_command_http_addon_stop(self, aioclient_mock):
         self.hass.block_till_done()
 
         assert len(aioclient_mock.mock_calls) == 1
+
+
+@asyncio.coroutine
+def test_async_hassio_host_view(aioclient_mock, hass, test_client):
+    """Test that it fetches the given url."""
+    os.environ['HASSIO'] = "127.0.0.1"
+
+    result = yield from async_setup_component(hass, ho.DOMAIN, {ho.DOMAIN: {}})
+    assert result, 'Failed to setup hasio'
+
+    client = yield from test_client(hass.http.app)
+
+    aioclient_mock.get('http://127.0.0.1/host/info', json={
+        'result': 'ok',
+        'data': {
+            'os': 'resinos',
+            'version': '0.3',
+            'current': '0.4',
+            'level': 16,
+            'hostname': 'test',
+        }
+    })
+
+    resp = yield from client.get('/api/hassio/host')
+    data = yield from resp.json()
+
+    assert resp.status == 200
+    assert data['os'] == 'resinos'
+    assert data['version'] == '0.3'
+    assert data['current'] == '0.4'
+    assert data['level'] == 16
+    assert data['hostname'] == 'test'