Update tests.

Author: Adminiuga

tests/test_api.py

@@ -350,3 +350,29 @@ async def test_write_parameter(api):
     assert unk_param not in list(deconz_api.NetworkParameter)
     with pytest.raises(KeyError):
         await api.write_parameter(unk_param, 0x55aa)
+
+
+@pytest.mark.parametrize(
+    "protocol_ver, firmware_version, flags",
+    [
+        (0x010A, 0x123405dd, 0x01),
+        (0x010B, 0x123405dd, 0x04),
+        (0x010A, 0x123407dd, 0x01),
+        (0x010B, 0x123407dd, 0x01),
+    ],
+)
+@pytest.mark.asyncio
+async def test_version(protocol_ver, firmware_version, flags, api):
+    api.read_parameter = mock.MagicMock()
+    api.read_parameter.side_effect = asyncio.coroutine(
+        mock.MagicMock(return_value=protocol_ver))
+    api._command = mock.MagicMock()
+    api._command.side_effect = asyncio.coroutine(
+        mock.MagicMock(return_value=[firmware_version]))
+    r = await api.version()
+    assert r == firmware_version
+    assert api._aps_data_ind_flags == flags
+
+
+def test_handle_version(api):
+    api._handle_version([mock.sentinel.version])

tests/test_application.py

@@ -46,6 +46,15 @@ def addr_nwk(nwk):
     return addr
 
 
+@pytest.fixture
+def addr_nwk_and_ieee(nwk, ieee):
+    addr = t.DeconzAddress()
+    addr.address_mode = t.ADDRESS_MODE.NWK_AND_IEEE
+    addr.address = nwk
+    addr.ieee = ieee
+    return addr
+
+
 def _test_rx(app, addr_ieee, addr_nwk, device, data):
     app.get_device = mock.MagicMock(return_value=device)
     app.devices = (EUI64(addr_ieee.address), )
@@ -96,6 +105,23 @@ def test_rx_ieee(app, addr_ieee, addr_nwk):
     )
 
 
+def test_rx_nwk_ieee(app, addr_ieee, addr_nwk_and_ieee):
+    device = mock.MagicMock()
+    app.handle_message = mock.MagicMock()
+    _test_rx(app, addr_ieee, addr_nwk_and_ieee, device, mock.sentinel.args)
+    assert app.handle_message.call_count == 1
+    assert app.handle_message.call_args == (
+        (
+            device,
+            mock.sentinel.profile_id,
+            mock.sentinel.cluster_id,
+            mock.sentinel.src_ep,
+            mock.sentinel.dst_ep,
+            mock.sentinel.args,
+        ),
+    )
+
+
 def test_rx_wrong_addr_mode(app, addr_ieee, addr_nwk, caplog):
     device = mock.MagicMock()
     app.handle_message = mock.MagicMock()
@@ -160,7 +186,7 @@ async def test_form_network(app):
 async def test_startup(app, monkeypatch, version=0):
 
     async def _version():
-        return [version]
+        return version
 
     app.form_network = mock.MagicMock(
         side_effect=asyncio.coroutine(mock.MagicMock()))