fixing coverage

Author: adedamola-sode

src/techui_builder/generate.py

@@ -207,7 +207,7 @@ def _allocate_widget(
         if scrn_mapping["type"] == "embedded":
             height, width = self._get_screen_dimensions(str(scrn_path))
             new_widget = pwidget.EmbeddedDisplay(
-                name,
+                name.removeprefix(":").removesuffix(":"),
                 str(data_scrn_path),
                 0,
                 0,  # Change depending on the order

tests/test_files/widget.xml

@@ -1,6 +1,6 @@
 <?xml version="1.0" ?>
 <widget type="embedded" version="2.0.0">
-  <name>CAM:</name>
+  <name>CAM</name>
   <x>0</x>
   <y>0</y>
   <width>860</width>

tests/test_generate.py

@@ -221,6 +221,31 @@ def test_generator_allocate_widget(generator):
     assert str(widget) == xml_content
 
 
+def test_generator_allocate_widget_with_suffix(generator):
+    generator._initilise_name_suffix = Mock(return_value=(":CAM:", ":CAM:", "R"))
+
+    scrn_mapping = {
+        "file": "ADAravis/ADAravis_summary.bob",
+        "prefix": "$(P)$(R)",
+        "suffix": ":CAM:",
+        "type": "embedded",
+    }
+    component = Entity(
+        type="detectorPlugins.detectorPlugins",
+        P="BL23B-DI-MOD-02",
+        desc=None,
+        M=None,
+        R=None,
+    )
+    widget = generator._allocate_widget(scrn_mapping, component)
+    control_widget = Path("tests/test_files/widget.xml")
+
+    with open(control_widget) as f:
+        xml_content = f.read()
+
+    assert str(widget) == xml_content
+
+
 def test_generator_create_widget_related(generator):
     generator._get_screen_dimensions = Mock(return_value=(800, 1280))
     screen_name = "test"