bumblebee-status/tests/modules/test_brightness.py

# pylint: disable=C0103,C0111

import mock
import unittest

try:
    from StringIO import StringIO
except ImportError:
    from io import StringIO

try:
    FileNotFoundError
except NameError:
    FileNotFoundError = IOError

import tests.mocks as mocks

from bumblebee.config import Config
from bumblebee.input import WHEEL_UP, WHEEL_DOWN
from bumblebee.modules.brightness import Module

class TestBrightnessModule(unittest.TestCase):
    def setUp(self):
        mocks.setup_test(self, Module)

    def tearDown(self):
        mocks.teardown_test(self)

    # def test_format(self):
    #     for widget in self.module.widgets():
    #         self.assertEquals(len(widget.full_text()), len("100%"))

    def test_wheel_up(self):
        mocks.mouseEvent(stdin=self.stdin, button=WHEEL_UP, inp=self.input, module=self.module)
        self.popen.assert_call("xbacklight +2%")

    def test_wheel_down(self):
        mocks.mouseEvent(stdin=self.stdin, button=WHEEL_DOWN, inp=self.input, module=self.module)
        self.popen.assert_call("xbacklight -2%")

    def test_custom_step(self):
        self.config.set("brightness.step", "10")
        module = Module(engine=self.engine, config={"config": self.config})
        mocks.mouseEvent(stdin=self.stdin, button=WHEEL_DOWN, inp=self.input, module=module)
        self.popen.assert_call("xbacklight -10%")

    @mock.patch('bumblebee.modules.brightness.open', create=True)
    def test_update(self, mock_open):
        mock_open.side_effect = [
                mock.mock_open(read_data="20").return_value,
                mock.mock_open(read_data="100").return_value
                ]
        self.module.update_all()
        self.assertEquals(self.module.brightness(self.anyWidget), "020%")
        self.assertEquals(len(self.module.brightness(self.anyWidget)), len("100%"))

    @mock.patch('bumblebee.modules.brightness.open')
    def test_error(self,mock_open):
        mock_open.side_effect = FileNotFoundError
        self.module.update_all()
        self.assertEquals(self.module.brightness(self.anyWidget), "n/a")

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