rdrf #2449 get fields on tl

muccg · Apr 12, 2023 · 6cbcc05 · 6cbcc05
1 parent f88fbf5
commit 6cbcc05
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rdrf/dashboards/testing.py b/rdrf/dashboards/testing.py
@@ -56,6 +56,7 @@ def test_normal_case(self):
 
     def create_traffic_light(self, data):
         from dashboards.components.tl import TrafficLights
+        from dashboards.components.tl import get_fields
 
         vis_config = Mock()
 
@@ -95,7 +96,9 @@ def create_traffic_light(self, data):
         vis_config.config = config_dict
         patient = Mock()
 
-        return TrafficLights("crc test tl", vis_config, data, patient, None)
+        tl = TrafficLights("crc test tl", vis_config, data, patient, None)
+        tl.fields = get_fields(tl.config)
+        return tl
 
     def create_dataframe(self):
         return None