From 0323a9c4e6f5ee0bb5cb8b33432d2f5eb739ce9e Mon Sep 17 00:00:00 2001
From: Joost Lekkerkerker <joostlek@outlook.com>
Date: Fri, 28 Feb 2025 23:03:57 +0100
Subject: [PATCH] Add SmartThings Viper device info (#139548)

---
 .../components/smartthings/entity.py          |  9 ++++
 .../smartthings/snapshots/test_init.ambr      | 50 +++++++++----------
 2 files changed, 34 insertions(+), 25 deletions(-)

diff --git a/homeassistant/components/smartthings/entity.py b/homeassistant/components/smartthings/entity.py
index 0d6ee32b473..f86f3a68f0e 100644
--- a/homeassistant/components/smartthings/entity.py
+++ b/homeassistant/components/smartthings/entity.py
@@ -53,6 +53,15 @@ class SmartThingsEntity(Entity):
                     "sw_version": ocf.firmware_version,
                 }
             )
+        if (viper := device.device.viper) is not None:
+            self._attr_device_info.update(
+                {
+                    "manufacturer": viper.manufacturer_name,
+                    "model": viper.model_name,
+                    "hw_version": viper.hardware_version,
+                    "sw_version": viper.software_version,
+                }
+            )
 
     async def async_added_to_hass(self) -> None:
         """Subscribe to updates."""
diff --git a/tests/components/smartthings/snapshots/test_init.ambr b/tests/components/smartthings/snapshots/test_init.ambr
index 0b5aeb57c18..e0d93553121 100644
--- a/tests/components/smartthings/snapshots/test_init.ambr
+++ b/tests/components/smartthings/snapshots/test_init.ambr
@@ -86,8 +86,8 @@
     'is_new': False,
     'labels': set({
     }),
-    'manufacturer': None,
-    'model': None,
+    'manufacturer': 'Arlo',
+    'model': 'VMC4041PB',
     'model_id': None,
     'name': '2nd Floor Hallway',
     'name_by_user': None,
@@ -108,7 +108,7 @@
     }),
     'disabled_by': None,
     'entry_type': None,
-    'hw_version': None,
+    'hw_version': 'WoCurtain3-WoCurtain3',
     'id': <ANY>,
     'identifiers': set({
       tuple(
@@ -119,8 +119,8 @@
     'is_new': False,
     'labels': set({
     }),
-    'manufacturer': None,
-    'model': None,
+    'manufacturer': 'WonderLabs Company',
+    'model': 'WoCurtain3',
     'model_id': None,
     'name': 'Curtain 1A',
     'name_by_user': None,
@@ -471,7 +471,7 @@
     }),
     'disabled_by': None,
     'entry_type': None,
-    'hw_version': None,
+    'hw_version': '250206213001',
     'id': <ANY>,
     'identifiers': set({
       tuple(
@@ -482,15 +482,15 @@
     'is_new': False,
     'labels': set({
     }),
-    'manufacturer': None,
-    'model': None,
+    'manufacturer': 'ecobee',
+    'model': 'aresSmart-ecobee3_remote_sensor',
     'model_id': None,
     'name': 'Child Bedroom',
     'name_by_user': None,
     'primary_config_entry': <ANY>,
     'serial_number': None,
     'suggested_area': None,
-    'sw_version': None,
+    'sw_version': '250206213001',
     'via_device_id': None,
   })
 # ---
@@ -504,7 +504,7 @@
     }),
     'disabled_by': None,
     'entry_type': None,
-    'hw_version': None,
+    'hw_version': '250206151734',
     'id': <ANY>,
     'identifiers': set({
       tuple(
@@ -515,15 +515,15 @@
     'is_new': False,
     'labels': set({
     }),
-    'manufacturer': None,
-    'model': None,
+    'manufacturer': 'ecobee',
+    'model': 'aresSmart-thermostat',
     'model_id': None,
     'name': 'Main Floor',
     'name_by_user': None,
     'primary_config_entry': <ANY>,
     'serial_number': None,
     'suggested_area': None,
-    'sw_version': None,
+    'sw_version': '250206151734',
     'via_device_id': None,
   })
 # ---
@@ -603,7 +603,7 @@
     }),
     'disabled_by': None,
     'entry_type': None,
-    'hw_version': None,
+    'hw_version': 'LTG002',
     'id': <ANY>,
     'identifiers': set({
       tuple(
@@ -614,15 +614,15 @@
     'is_new': False,
     'labels': set({
     }),
-    'manufacturer': None,
-    'model': None,
+    'manufacturer': 'Signify Netherlands B.V.',
+    'model': 'Hue ambiance spot',
     'model_id': None,
     'name': 'Bathroom spot',
     'name_by_user': None,
     'primary_config_entry': <ANY>,
     'serial_number': None,
     'suggested_area': None,
-    'sw_version': None,
+    'sw_version': '1.122.2',
     'via_device_id': None,
   })
 # ---
@@ -636,7 +636,7 @@
     }),
     'disabled_by': None,
     'entry_type': None,
-    'hw_version': None,
+    'hw_version': 'LCA001',
     'id': <ANY>,
     'identifiers': set({
       tuple(
@@ -647,15 +647,15 @@
     'is_new': False,
     'labels': set({
     }),
-    'manufacturer': None,
-    'model': None,
+    'manufacturer': 'Signify Netherlands B.V.',
+    'model': 'Hue color lamp',
     'model_id': None,
     'name': 'Standing light',
     'name_by_user': None,
     'primary_config_entry': <ANY>,
     'serial_number': None,
     'suggested_area': None,
-    'sw_version': None,
+    'sw_version': '1.122.2',
     'via_device_id': None,
   })
 # ---
@@ -735,7 +735,7 @@
     }),
     'disabled_by': None,
     'entry_type': None,
-    'hw_version': None,
+    'hw_version': 'SKY40147',
     'id': <ANY>,
     'identifiers': set({
       tuple(
@@ -746,15 +746,15 @@
     'is_new': False,
     'labels': set({
     }),
-    'manufacturer': None,
-    'model': None,
+    'manufacturer': 'Sensibo',
+    'model': 'skyplus',
     'model_id': None,
     'name': 'Office',
     'name_by_user': None,
     'primary_config_entry': <ANY>,
     'serial_number': None,
     'suggested_area': None,
-    'sw_version': None,
+    'sw_version': 'SKY40147',
     'via_device_id': None,
   })
 # ---