From fb281c6bdecf3d93b016b52bbb5d0682046c70a2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Joakim=20S=C3=B8rensen?= <joasoe@gmail.com>
Date: Tue, 1 Jun 2021 15:09:23 +0200
Subject: [PATCH] Add arch to payload (#51330)

---
 homeassistant/components/analytics/analytics.py |  2 ++
 homeassistant/components/analytics/const.py     |  1 +
 tests/components/analytics/test_analytics.py    | 11 +++++++++--
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/homeassistant/components/analytics/analytics.py b/homeassistant/components/analytics/analytics.py
index e6e7ffac337..571ffd90f22 100644
--- a/homeassistant/components/analytics/analytics.py
+++ b/homeassistant/components/analytics/analytics.py
@@ -21,6 +21,7 @@ from .const import (
     ANALYTICS_ENDPOINT_URL_DEV,
     ATTR_ADDON_COUNT,
     ATTR_ADDONS,
+    ATTR_ARCH,
     ATTR_AUTO_UPDATE,
     ATTR_AUTOMATION_COUNT,
     ATTR_BASE,
@@ -157,6 +158,7 @@ class Analytics:
             payload[ATTR_SUPERVISOR] = {
                 ATTR_HEALTHY: supervisor_info[ATTR_HEALTHY],
                 ATTR_SUPPORTED: supervisor_info[ATTR_SUPPORTED],
+                ATTR_ARCH: supervisor_info[ATTR_ARCH],
             }
 
         if operating_system_info.get(ATTR_BOARD) is not None:
diff --git a/homeassistant/components/analytics/const.py b/homeassistant/components/analytics/const.py
index 16929a7131d..4688c578a00 100644
--- a/homeassistant/components/analytics/const.py
+++ b/homeassistant/components/analytics/const.py
@@ -16,6 +16,7 @@ LOGGER: logging.Logger = logging.getLogger(__package__)
 
 ATTR_ADDON_COUNT = "addon_count"
 ATTR_ADDONS = "addons"
+ATTR_ARCH = "arch"
 ATTR_AUTO_UPDATE = "auto_update"
 ATTR_AUTOMATION_COUNT = "automation_count"
 ATTR_BASE = "base"
diff --git a/tests/components/analytics/test_analytics.py b/tests/components/analytics/test_analytics.py
index 09f82e37fba..ee67a7e3935 100644
--- a/tests/components/analytics/test_analytics.py
+++ b/tests/components/analytics/test_analytics.py
@@ -132,7 +132,9 @@ async def test_send_base_with_supervisor(hass, caplog, aioclient_mock):
 
     with patch(
         "homeassistant.components.hassio.get_supervisor_info",
-        side_effect=Mock(return_value={"supported": True, "healthy": True}),
+        side_effect=Mock(
+            return_value={"supported": True, "healthy": True, "arch": "amd64"}
+        ),
     ), patch(
         "homeassistant.components.hassio.get_os_info",
         side_effect=Mock(return_value={"board": "blue", "version": "123"}),
@@ -157,7 +159,10 @@ async def test_send_base_with_supervisor(hass, caplog, aioclient_mock):
 
     assert f"'uuid': '{MOCK_UUID}'" in caplog.text
     assert f"'version': '{MOCK_VERSION}'" in caplog.text
-    assert "'supervisor': {'healthy': True, 'supported': True}" in caplog.text
+    assert (
+        "'supervisor': {'healthy': True, 'supported': True, 'arch': 'amd64'}"
+        in caplog.text
+    )
     assert "'operating_system': {'board': 'blue', 'version': '123'}" in caplog.text
     assert "'installation_type':" in caplog.text
     assert "'integration_count':" not in caplog.text
@@ -197,6 +202,7 @@ async def test_send_usage_with_supervisor(hass, caplog, aioclient_mock):
             return_value={
                 "healthy": True,
                 "supported": True,
+                "arch": "amd64",
                 "addons": [{"slug": "test_addon"}],
             }
         ),
@@ -303,6 +309,7 @@ async def test_send_statistics_with_supervisor(hass, caplog, aioclient_mock):
             return_value={
                 "healthy": True,
                 "supported": True,
+                "arch": "amd64",
                 "addons": [{"slug": "test_addon"}],
             }
         ),