bumblebee-status/tests/core/test_decorators.py

import unittest

import core.decorators
import core.widget
import core.module
import core.config


class TestModule(core.module.Module):
    def __init__(self, config=None, theme=None):
        config = core.config.Config([])
        super().__init__(config, theme, core.widget.Widget(self.get))
        self.text = ""

    @core.decorators.scrollable
    def get(self, widget):
        return self.text


class config(unittest.TestCase):
    def setUp(self):
        self.module = TestModule()
        self.widget = self.module.widget()
        self.width = 10
        self.module.set("width", self.width)

    def test_no_text(self):
        self.assertEqual("", self.module.text)
        self.assertEqual("", self.module.get(self.widget))

    def test_smaller(self):
        self.module.text = "test"
        self.assertLess(len(self.module.text), self.width)
        self.assertEqual("test", self.module.get(self.widget))

    def test_bigger(self):
        self.module.text = "abcdefghijklmnopqrst"
        self.assertGreater(len(self.module.text), self.width)
        self.assertEqual(self.module.text[: self.width], self.module.get(self.widget))

    def test_bounce(self):
        self.module.text = "abcd"
        self.module.set("width", 2)
        self.assertEqual("ab", self.module.get(self.widget))
        self.assertEqual("bc", self.module.get(self.widget))
        self.assertEqual("cd", self.module.get(self.widget))
        self.assertEqual("bc", self.module.get(self.widget))
        self.assertEqual("ab", self.module.get(self.widget))
        self.assertEqual("bc", self.module.get(self.widget))
        self.assertEqual("cd", self.module.get(self.widget))
        self.assertEqual("bc", self.module.get(self.widget))
        self.assertEqual("ab", self.module.get(self.widget))

    def test_nobounce(self):
        self.module.set("scrolling.bounce", False)
        self.module.text = "abcd"
        self.module.set("width", 2)
        self.assertEqual("ab", self.module.get(self.widget))
        self.assertEqual("bc", self.module.get(self.widget))
        self.assertEqual("cd", self.module.get(self.widget))
        self.assertEqual("ab", self.module.get(self.widget))
        self.assertEqual("bc", self.module.get(self.widget))
        self.assertEqual("cd", self.module.get(self.widget))


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