From 36632f64f1dd89dd34903db62cd705b66f53b113 Mon Sep 17 00:00:00 2001
From: Martin Hjelmare <marhje52@gmail.com>
Date: Sun, 6 Sep 2020 11:54:08 +0200
Subject: [PATCH] Mock tradfri lights correctly (#39706)

---
 tests/components/tradfri/test_light.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/components/tradfri/test_light.py b/tests/components/tradfri/test_light.py
index 653a9ce62df..cf11d42411e 100644
--- a/tests/components/tradfri/test_light.py
+++ b/tests/components/tradfri/test_light.py
@@ -135,6 +135,10 @@ def mock_light(test_features=None, test_state=None, light_number=0):
         reachable=True,
         observe=Mock(),
         device_info=dev_info_mock,
+        has_light_control=True,
+        has_socket_control=False,
+        has_blind_control=False,
+        has_signal_repeater_control=False,
     )
     _mock_light.name = f"tradfri_light_{light_number}"