bumblebee-status/tests/mocks.py

# pylint: disable=C0103,C0111

import mock
import json
import shlex
import random
import subprocess

from bumblebee.input import I3BarInput
from bumblebee.output import Widget
from bumblebee.config import Config

def rand(cnt):
    return "".join(random.choice("abcdefghijklmnopqrstuvwxyz0123456789") for i in range(cnt))

def setup_test(test, Module):
    test._stdin, test._select, test.stdin, test.select = poll_mock("bumblebee.input")

    test.popen = MockPopen()

    test.config = Config()
    test.input = I3BarInput()
    test.engine = mock.Mock()
    test.engine.input = test.input
    test.input.need_event = True
    test.module = Module(engine=test.engine, config={ "config": test.config })
    for widget in test.module.widgets():
        widget.link_module(test.module)
        test.anyWidget = widget

def teardown_test(test):
    test._stdin.stop()
    test._select.stop()
    test.popen.cleanup()

def poll_mock(module=""):
    if len(module) > 0: module = "{}.".format(module)

    stdin = mock.patch("{}sys.stdin".format(module))
    select = mock.patch("{}select".format(module))
    poll = mock.Mock()

    stdin_mock = stdin.start()
    select_mock = select.start()

    stdin_mock.fileno.return_value = 1
    select_mock.poll.return_value = poll
    poll.poll.return_value = [(stdin_mock.fileno.return_value, 100)]

    return stdin, select, stdin_mock, select_mock

def mouseEvent(stdin, button, inp, module=None, instance=None):
    stdin.readline.return_value = json.dumps({
        "name": module.id if module else rand(10),
        "button": button,
        "instance": instance
    })
    inp.start()
    inp.stop()
    stdin.readline.assert_any_call()

class MockPopen(object):
    def __init__(self, module=""):
        if len(module) > 0: module = "{}.".format(module)
        self._patch = mock.patch("{}subprocess.Popen".format(module))
        self._popen = self._patch.start()
        self.mock = mock.Mock()
        # for a nicer, more uniform interface
        self.mock.popen = self._popen
        # for easier command execution checks
        self.mock.popen.assert_call = self.assert_call
        self._popen.return_value = self.mock

        self.mock.communicate.return_value = [ "", None ]
        self.mock.returncode = 0

    def assert_call(self, cmd):
        self.mock.popen.assert_any_call(shlex.split(cmd), env=mock.ANY, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

    def cleanup(self):
        self._patch.stop()

class MockInput(object):
    def __init__(self):
        self._callbacks = {}
    def start(self):
        pass

    def stop(self):
        pass

    def get_callback(self, uid):
        return self._callbacks.get(uid, None)

    def register_callback(self, obj, button, cmd):
        if not obj:
            return
        self._callbacks[obj.id] = {
            "button": button,
            "command": cmd,
        }

class MockOutput(object):
    def start(self):
        pass

    def stop(self):
        pass

    def draw(self, widget, engine, module):
        engine.stop()

    def begin(self):
        pass

    def flush(self):
        pass

    def end(self):
        pass

class MockEngine(object):
    def __init__(self):
        self.input = MockInput()

class MockWidget(Widget):
    def __init__(self, text):
        super(MockWidget, self).__init__(text)
        self.module = None
        self.attr_state = ["state-default"]
        self.id = rand(10)

        self.full_text(text)

#    def state(self):
#        return self.attr_state

    def update(self, widgets):
        pass

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4