File: conftest.py

package info (click to toggle)
python-scrapli-replay 2023.7.30-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 456 kB
  • sloc: python: 2,625; makefile: 43
file content (41 lines) | stat: -rw-r--r-- 1,169 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import pytest

from scrapli.driver.core.cisco_iosxe import IOSXEDriver
from scrapli_replay.server.collector import ScrapliCollector
from scrapli_replay.server.server import BaseSSHServerSession


@pytest.fixture(scope="function")
def scrapli_conn():
    scrapli_conn = IOSXEDriver(host="localhost")
    return scrapli_conn


@pytest.fixture(scope="function")
def basic_collector(scrapli_conn):
    channel_inputs = ["some input"]
    interact_events = [[("some input", "some pattern", False)]]
    paging_indicator = "--More--"
    paging_escape_string = "non-standard"

    collector = ScrapliCollector(
        channel_inputs=channel_inputs,
        interact_events=interact_events,
        paging_indicator=paging_indicator,
        paging_escape_string=paging_escape_string,
        scrapli_connection=scrapli_conn,
    )

    return collector


@pytest.fixture(scope="function")
def basic_server():
    server = BaseSSHServerSession(
        collect_data={
            "on_open_inputs": ["openstuff"],
            "initial_privilege_level": "veryprivvy",
            "privilege_level_prompts": {"veryprivvy": "veryprivvyprompt"},
        }
    )
    return server