Banner: Add unit tests

Signed-off-by: Yevhen Fastiuk <[email protected]>
sonic-net · May 14, 2024 · 8c8143d · 8c8143d
1 parent 24274a0
commit 8c8143d
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/hostcfgd/hostcfgd_test.py b/tests/hostcfgd/hostcfgd_test.py
@@ -312,3 +312,20 @@ def test_load(self):
         data = {}
         dns_cfg.load(data)
         dns_cfg.dns_update.assert_called()
+
+
+class TestBannerCfg:
+    def test_load(self):
+        banner_cfg = hostcfgd.BannerCfg()
+        banner_cfg.banner_message = mock.MagicMock()
+
+        data = {}
+        banner_cfg.load(data)
+        banner_cfg.banner_message.assert_called()
+
+    @mock.patch('hostcfgd.run_cmd')
+    def test_banner_message(self, mock_run_cmd):
+        banner_cfg = hostcfgd.BannerCfg()
+        banner_cfg.banner_message(None, {'test': 'test'})
+
+        mock_run_cmd.assert_has_calls([call(['systemctl', 'restart', 'banner-config'], True, True)])