Modified tests

Author: adedamola-sode

tests/test_generate.py

@@ -1,6 +1,6 @@
 from pathlib import Path
+from unittest.mock import patch
 
-# from techui_builder.builder import Builder
 from techui_builder.models import Entity
 
 
@@ -39,7 +39,8 @@ def test_generator_is_list_of_dicts(generator):
     assert generator._is_list_of_dicts(not_list_of_dicts) is False
 
 
-def test_generator_allocate_widget(generator):
+@patch("techui_builder.generate.Generator._initialise_name_suffix")
+def test_generator_allocate_widget(mock_initilise_name_suffix, generator):
     scrn_mapping = {
         "file": "ADAravis/ADAravis_summary.bob",
         "prefix": "$(P)$(R)",
@@ -48,9 +49,8 @@ def test_generator_allocate_widget(generator):
     component = Entity(
         type="ADAravis.aravisCamera", P="BL23B-DI-MOD-02", desc=None, M=None, R="CAM:"
     )
-
+    mock_initilise_name_suffix.return_value = ("CAM:", "CAM:", "R")
     widget = generator._allocate_widget(scrn_mapping, component)
-    print(widget)
     control_widget = Path("tests/test_files/widget.xml")
 
     with open(control_widget) as f: