Merge pull request #180 from realvitya/fix_fortios_prompt

fix/fortios - fix prompt regex
scrapli · Oct 15, 2024 · 6f047fb · 6f047fb
2 parents 8f5d131 + 5defcb0
commit 6f047fb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/scrapli_community/fortinet/fortios/fortinet_fortios.py b/scrapli_community/fortinet/fortios/fortinet_fortios.py
@@ -17,7 +17,7 @@
         "async": AsyncFortinetFortiOSDriver,
     },
     "defaults": {
-        "comms_prompt_pattern": r"[\w_-]+ (\(\w+\) )?[$#]",
+        "comms_prompt_pattern": r"[\w_-]+ (\([\w-]+\) )?[$#]",
         "sync_on_open": default_sync_on_open,
         "async_on_open": default_async_on_open,
         "sync_on_close": default_sync_on_close,

diff --git a/tests/unit/fortinet/fortios/test_fortinet_fortios.py b/tests/unit/fortinet/fortios/test_fortinet_fortios.py
@@ -12,6 +12,7 @@
         "SCRAPLI-FORTIGATE $ ",  # remote admin root prompt
         "SCRAPLI-FORTIGATE (global) # ",  # global context
         "SCRAPLI-FORTIGATE (interface) # ",  # interface config context
+        "SCRAPLI-FORTIGATE (ha-mgmt-interfaces) # ",  # subsection with dashes
         "SCRAPLI-FORTIGATE (vdom) # ",  # vdom config context
         "SCRAPLI-FORTIGATE (testvdom) # ",  # a user defined vdom context
     ],