From d0f55b2e7d9fa712faf260ab571787a8bb3e0d60 Mon Sep 17 00:00:00 2001
From: Khalim Conn-Kowlessar <kconnkowlessar@gmail.com>
Date: Mon, 3 Jul 2023 15:42:24 +0100
Subject: [PATCH] Fixed property tests

---
 model_data/Property.py            |  1 +
 model_data/tests/test_property.py | 22 +++++++++++++++-------
 2 files changed, 16 insertions(+), 7 deletions(-)

diff --git a/model_data/Property.py b/model_data/Property.py
index 8f542f67..3627a1d1 100644
--- a/model_data/Property.py
+++ b/model_data/Property.py
@@ -18,6 +18,7 @@ class Property(BaseUtility):
         "roof-description": "roof",
         "walls-description": "walls",
         "windows-description": "windows",
+        "lighting-description": "lighting"
     }
 
     floor = None
diff --git a/model_data/tests/test_property.py b/model_data/tests/test_property.py
index 964bda00..dd87e5d8 100644
--- a/model_data/tests/test_property.py
+++ b/model_data/tests/test_property.py
@@ -17,7 +17,8 @@ mock_epc_response = {
             "windows-description": "Windows Description",
             "mainheat-description": "Main Heating Description",
             "hotwater-description": "Hot Water Description",
-            "transaction-type": "rental"
+            "transaction-type": "rental",
+            "lighting-description": "Good Lighting Efficiency",
         },
         {
             "inspection-date": "2023-05-01",
@@ -27,7 +28,8 @@ mock_epc_response = {
             "windows-description": "Windows Description",
             "mainheat-description": "Main Heating Description",
             "hotwater-description": "Hot Water Description",
-            "transaction-type": "rental"
+            "transaction-type": "rental",
+            "lighting-description": "Good Lighting Efficiency",
         }
     ]
 }
@@ -38,21 +40,24 @@ mock_epc_response_dupe = {
             'inspection-date': '2023-06-01', 'some-other-key': 'some-value', 'roof-description': 'Roof Description',
             'walls-description': 'Walls Description', 'windows-description': 'Windows Description',
             'mainheat-description': 'Main Heating Description', 'hotwater-description': 'Hot Water Description',
-            "transaction-type": "rental"
+            "transaction-type": "rental",
+            "lighting-description": "Good Lighting Efficiency",
         },
         {
             'inspection-date': '2023-05-01', 'some-other-key': 'some-other-value',
             'roof-description': 'Roof Description', 'walls-description': 'Walls Description',
             'windows-description': 'Windows Description', 'mainheat-description': 'Main Heating Description',
             'hotwater-description': 'Hot Water Description',
-            "transaction-type": "rental"
+            "transaction-type": "rental",
+            "lighting-description": "Good Lighting Efficiency",
         },
         {
             'inspection-date': '2023-06-01', 'some-other-key': 'duplicate-date',
             'roof-description': 'Roof Description',
             'walls-description': 'Walls Description', 'windows-description': 'Windows Description',
             'mainheat-description': 'Main Heating Description', 'hotwater-description': 'Hot Water Description',
-            "transaction-type": "rental"
+            "transaction-type": "rental",
+            "lighting-description": "Good Lighting Efficiency",
         }
     ]
 }
@@ -99,14 +104,17 @@ class TestProperty:
             {"walls-description": "Walls Description"},
             {"windows-description": "Windows Description"},
             {"mainheat-description": "Main Heating Description"},
-            {"hotwater-description": "Hot Water Description"}
+            {"hotwater-description": "Hot Water Description"},
+            {"lighting-description": "Good Lighting Efficiency"},
+            {"low-energy-lighting": 0}
         ]))
         mock_cleaner.cleaned = {
             "roof-description": [{"original_description": "Roof Description"}],
             "walls-description": [{"original_description": "Walls Description"}],
             "windows-description": [{"original_description": "Windows Description"}],
             "mainheat-description": [{"original_description": "Main Heating Description"}],
-            "hotwater-description": [{"original_description": "Hot Water Description"}]
+            "hotwater-description": [{"original_description": "Hot Water Description"}],
+            "lighting-description": [{"original_description": "Good Lighting Efficiency"}]
         }
         return mock_cleaner