Source code for labgrid.driver.fake

import logging

import attr

from ..factory import target_factory
from ..protocol import CommandProtocol, ConsoleProtocol, FileTransferProtocol, PowerProtocol
from .common import Driver
from .commandmixin import CommandMixin
from .consoleexpectmixin import ConsoleExpectMixin


@target_factory.reg_driver
@attr.s
[docs]class FakeConsoleDriver(ConsoleExpectMixin, Driver, ConsoleProtocol):
    txdelay = attr.ib(default=0.0, validator=attr.validators.instance_of(float))

    def __attrs_post_init__(self):
        super().__attrs_post_init__()
        self.logger = logging.getLogger("{}({})".format(self, self.target))

    def _read(self, *args):
        pass

    def _write(self, *args):
        pass

[docs]    def open(self):
        pass

[docs]    def close(self):
        pass


@target_factory.reg_driver
@attr.s
[docs]class FakeCommandDriver(CommandMixin, Driver, CommandProtocol):
    @Driver.check_active
[docs]    def run(self, *args):
        pass

    @Driver.check_active
[docs]    def run_check(self, *args):
        pass

    @Driver.check_active
[docs]    def get_status(self):
        pass


@target_factory.reg_driver
@attr.s
[docs]class FakeFileTransferDriver(Driver, FileTransferProtocol):
    @Driver.check_active
[docs]    def get(self, *args):
        pass

    @Driver.check_active
[docs]    def put(self, *args):
        pass

@target_factory.reg_driver
@attr.s
[docs]class FakePowerDriver(Driver, PowerProtocol):
    @Driver.check_active
[docs]    def on(self, *args):
        pass

    @Driver.check_active
[docs]    def off(self, *args):
        pass

    @Driver.check_active
[docs]    def cycle(self, *args):
        pass