[tests] Fix unit tests (at least on my system)

This commit is contained in:
Tobi-wan Kenobi 2016-12-10 19:45:13 +01:00
parent 921afe475f
commit 716bafa90e
2 changed files with 9 additions and 1 deletions

View file

@ -8,5 +8,4 @@ python:
install: install:
- pip install psutil - pip install psutil
- pip install netifaces - pip install netifaces
- pip install mock
script: nosetests -v tests/ script: nosetests -v tests/

View file

@ -2,11 +2,19 @@
import unittest import unittest
import importlib import importlib
import mock
from bumblebee.engine import all_modules from bumblebee.engine import all_modules
from bumblebee.config import Config from bumblebee.config import Config
from tests.util import assertWidgetAttributes, MockEngine from tests.util import assertWidgetAttributes, MockEngine
class MockCommunicate(object):
def __init__(self):
self.returncode = 0
def communicate(self):
return (str.encode("1"), "error")
class TestGenericModules(unittest.TestCase): class TestGenericModules(unittest.TestCase):
def setUp(self): def setUp(self):
engine = MockEngine() engine = MockEngine()
@ -30,6 +38,7 @@ class TestGenericModules(unittest.TestCase):
@mock.patch("subprocess.Popen") @mock.patch("subprocess.Popen")
def test_update(self, mock_output): def test_update(self, mock_output):
mock_output.return_value = MockCommunicate()
for mod in self.objects: for mod in self.objects:
widgets = self.objects[mod].widgets() widgets = self.objects[mod].widgets()
self.objects[mod].update(widgets) self.objects[mod].update(widgets)