From 365e8a74ee639381d8625daa2f501052c4b2a2e9 Mon Sep 17 00:00:00 2001
From: Tobias Sauerwein <cgtobi@users.noreply.github.com>
Date: Sat, 20 Mar 2021 08:26:11 +0100
Subject: [PATCH] Add tests for Netatmo webhook handler (#46396)

* Add tests for Netatmo webhook handler

* Add async prefix

* Remove freezegun dependency

* Clean up
---
 .coveragerc                              |   1 -
 tests/components/netatmo/test_webhook.py | 126 +++++++++++++++++++++++
 2 files changed, 126 insertions(+), 1 deletion(-)
 create mode 100644 tests/components/netatmo/test_webhook.py

diff --git a/.coveragerc b/.coveragerc
index 48fb4ce756f..33f4071c8f4 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -648,7 +648,6 @@ omit =
     homeassistant/components/netatmo/helper.py
     homeassistant/components/netatmo/netatmo_entity_base.py
     homeassistant/components/netatmo/sensor.py
-    homeassistant/components/netatmo/webhook.py
     homeassistant/components/netdata/sensor.py
     homeassistant/components/netgear/device_tracker.py
     homeassistant/components/netgear_lte/*
diff --git a/tests/components/netatmo/test_webhook.py b/tests/components/netatmo/test_webhook.py
new file mode 100644
index 00000000000..a56bd2f0fde
--- /dev/null
+++ b/tests/components/netatmo/test_webhook.py
@@ -0,0 +1,126 @@
+"""The tests for Netatmo webhook events."""
+from homeassistant.components.netatmo.const import DATA_DEVICE_IDS, DATA_PERSONS
+from homeassistant.components.netatmo.webhook import async_handle_webhook
+from homeassistant.helpers.dispatcher import async_dispatcher_connect
+from homeassistant.util.aiohttp import MockRequest
+
+
+async def test_webhook(hass):
+    """Test that webhook events are processed."""
+    webhook_called = False
+
+    async def handle_event(_):
+        nonlocal webhook_called
+        webhook_called = True
+
+    response = (
+        b'{"user_id": "123", "user": {"id": "123", "email": "foo@bar.com"},'
+        b'"push_type": "webhook_activation"}'
+    )
+    request = MockRequest(content=response, mock_source="test")
+
+    async_dispatcher_connect(
+        hass,
+        "signal-netatmo-webhook-None",
+        handle_event,
+    )
+
+    await async_handle_webhook(hass, "webhook_id", request)
+    await hass.async_block_till_done()
+
+    assert webhook_called
+
+
+async def test_webhook_error_in_data(hass):
+    """Test that errors in webhook data are handled."""
+    webhook_called = False
+
+    async def handle_event(_):
+        nonlocal webhook_called
+        webhook_called = True
+
+    response = b'""webhook_activation"}'
+    request = MockRequest(content=response, mock_source="test")
+
+    async_dispatcher_connect(
+        hass,
+        "signal-netatmo-webhook-None",
+        handle_event,
+    )
+
+    await async_handle_webhook(hass, "webhook_id", request)
+    await hass.async_block_till_done()
+
+    assert not webhook_called
+
+
+async def test_webhook_climate_event(hass):
+    """Test that climate events are handled."""
+    webhook_called = False
+
+    async def handle_event(_):
+        nonlocal webhook_called
+        webhook_called = True
+
+    response = (
+        b'{"user_id": "123", "user": {"id": "123", "email": "foo@bar.com"},'
+        b'"home_id": "456", "event_type": "therm_mode",'
+        b'"home": {"id": "456", "therm_mode": "away"},'
+        b'"mode": "away", "previous_mode": "schedule", "push_type": "home_event_changed"}'
+    )
+    request = MockRequest(content=response, mock_source="test")
+
+    hass.data["netatmo"] = {
+        DATA_DEVICE_IDS: {},
+    }
+
+    async_dispatcher_connect(
+        hass,
+        "signal-netatmo-webhook-therm_mode",
+        handle_event,
+    )
+
+    await async_handle_webhook(hass, "webhook_id", request)
+    await hass.async_block_till_done()
+
+    assert webhook_called
+
+
+async def test_webhook_person_event(hass):
+    """Test that person events are handled."""
+    webhook_called = False
+
+    async def handle_event(_):
+        nonlocal webhook_called
+        webhook_called = True
+
+    response = (
+        b'{"user_id": "5c81004xxxxxxxxxx45f4",'
+        b'"persons": [{"id": "e2bf7xxxxxxxxxxxxea3", "face_id": "5d66xxxxxx9b9",'
+        b'"face_key": "89dxxxxx22", "is_known": true,'
+        b'"face_url": "https://netatmocameraimage.blob.core.windows.net/production/5xxx"}],'
+        b'"snapshot_id": "5d19bae867368a59e81cca89", "snapshot_key": "d3b3ae0229f7xb74cf8",'
+        b'"snapshot_url": "https://netatmocameraimage.blob.core.windows.net/production/5xxxx",'
+        b'"event_type": "person", "camera_id": "70:xxxxxx:a7", "device_id": "70:xxxxxx:a7",'
+        b'"home_id": "5c5dxxxxxxxd594", "home_name": "Boulogne Billan.",'
+        b'"event_id": "5d19bxxxxxxxxcca88",'
+        b'"message": "Boulogne Billan.: Benoit has been seen by Indoor Camera ",'
+        b'"push_type": "NACamera-person"}'
+    )
+    request = MockRequest(content=response, mock_source="test")
+
+    hass.data["netatmo"] = {
+        DATA_DEVICE_IDS: {},
+        DATA_PERSONS: {},
+    }
+
+    async_dispatcher_connect(
+        hass,
+        "signal-netatmo-webhook-person",
+        handle_event,
+    )
+
+    await async_handle_webhook(hass, "webhook_id", request)
+    await hass.async_block_till_done()
+
+    assert webhook_called