rename deploy_test to test.py

.drone.yml

@@ -16,7 +16,7 @@ steps:
     image: vireakouk/netdevops-drone-nornir-scrapli:latest
     commands:
       - pip install pybatfish rich
-      - python deploy_test.py
+      - python test.py
     when:
       branch:
         - dev
@@ -44,7 +44,6 @@ steps:
       - git remote -v
       - git push --all
       - git push --tags
-
     when:
       branch:
         - dev
@@ -66,8 +65,7 @@ steps:
     image: vireakouk/netdevops-drone-nornir-scrapli:latest
     commands:
       - pip install pybatfish rich
-      - python backup_config.py
-      - python deploy_test.py
+      - python test.py
     when:
       branch:
         - main
@@ -97,7 +95,6 @@ steps:
           🌐 Commit Link: {{ commit.link }}
           🌐 Build Link: {{ build.link }}
         {{/success}}
-      
     when:
       status: [ success, failure ]
       branch: [ main, dev ]

test.py

@@ -0,0 +1,73 @@
+## cloned from Julio Perez batfish scripts ##
+
+from datetime import datetime
+from pybatfish.client.commands import (
+    bf_session,
+    bf_init_snapshot,
+    bf_set_network,
+)
+from pybatfish.client.session import Session
+from pybatfish.question import load_questions
+from pybatfish.client.asserts import (
+    assert_no_duplicate_router_ids,
+    assert_no_incompatible_bgp_sessions,
+    assert_no_unestablished_bgp_sessions,
+)
+from rich.console import Console
+
+
+console = Console(color_system="truecolor")
+
+
+def test_duplicate_rtr_ids(snap):
+    """Testing for duplicate router IDs"""
+    console.print(
+        ":white_exclamation_mark: [bold yellow]Testing for duplicate router IDs[/bold yellow] :white_exclamation_mark:"
+    )
+    assert_no_duplicate_router_ids(
+        snapshot=snap,
+        protocols={"ospf", "bgp"},
+    )
+    console.print(
+        ":green_heart: [bold green]No duplicate router IDs found[/bold green] :green_heart:"
+    )
+
+
+def test_bgp_compatibility(snap):
+    """Testing for incompatible BGP sessions"""
+    console.print(
+        ":white_exclamation_mark: [bold yellow]Testing for incompatible BGP sessions[/bold yellow] :white_exclamation_mark:"
+    )
+    assert_no_incompatible_bgp_sessions(
+        snapshot=snap,
+    )
+    console.print(
+        ":green_heart: [bold green]All BGP sessions compatible![/bold green] :green_heart:"
+    )
+
+
+def test_bgp_unestablished(snap):
+    """Testing for BGP sessions that are not established"""
+    console.print(
+        ":white_exclamation_mark: [bold yellow]Testing for unestablished BGP sessions[/bold yellow] :white_exclamation_mark:"
+    )
+    assert_no_unestablished_bgp_sessions(
+        snapshot=snap,
+    )
+    console.print(
+        ":green_heart: [bold green]All BGP sessions are established![/bold green] :green_heart:"
+    )
+
+
+if __name__ == "__main__":
+    NETWORK_NAME = "eos_lab"
+    SNAPSHOT_NAME = f"snapshot-{datetime.today().strftime('%Y-%m-%d')}"
+    SNAPSHOT_DIR = "./snapshots"
+
+    bf_session.host = "192.168.0.200"
+    bf_set_network(NETWORK_NAME)
+    init_snap = bf_init_snapshot(SNAPSHOT_DIR, name=SNAPSHOT_NAME, overwrite=True)
+    load_questions()
+    test_duplicate_rtr_ids(init_snap)
+    test_bgp_compatibility(init_snap)
+    test_bgp_unestablished(init_snap)