Merge pull request #74 from pimoroni/support-disable-led-button

Support disabling LED/button for #70

Author: Gadgetoid

examples/automatic.py

@@ -89,7 +89,7 @@ def set_automatic(status):
     last_change = 0
 
 
-fanshim = FanShim()
+fanshim = FanShim(disable_button=args.nobutton, disable_led=args.noled)
 fanshim.set_hold_time(1.0)
 fanshim.set_fan(False)
 armed = True

library/fanshim/__init__.py

@@ -8,7 +8,7 @@
 
 
 class FanShim():
-    def __init__(self, pin_fancontrol=18, pin_button=17, button_poll_delay=0.05):
+    def __init__(self, pin_fancontrol=18, pin_button=17, button_poll_delay=0.05, disable_button=False, disable_led=False):
         """FAN Shim.
 
         :param pin_fancontrol: BCM pin for fan on/off
@@ -24,24 +24,36 @@ def __init__(self, pin_fancontrol=18, pin_button=17, button_poll_delay=0.05):
         self._button_hold_time = 2.0
         self._t_poll = None
 
+        self._disable_button = disable_button
+        self._disable_led = disable_led
+
         GPIO.setwarnings(False)
         GPIO.setmode(GPIO.BCM)
         GPIO.setup(self._pin_fancontrol, GPIO.OUT)
-        GPIO.setup(self._pin_button, GPIO.IN, pull_up_down=GPIO.PUD_UP)
 
-        self._led = apa102.APA102(1, 15, 14, None, brightness=0.05)
+        if not self._disable_button:
+            GPIO.setup(self._pin_button, GPIO.IN, pull_up_down=GPIO.PUD_UP)
+
+        if not self._disable_led:
+            self._led = apa102.APA102(1, 15, 14, None, brightness=0.05)
 
         atexit.register(self._cleanup)
 
     def start_polling(self):
         """Start button polling."""
+        if self._disable_button:
+            return
+
         if self._t_poll is None:
             self._t_poll = Thread(target=self._run)
             self._t_poll.daemon = True
             self._t_poll.start()
 
     def stop_polling(self):
         """Stop button polling."""
+        if self._disable_button:
+            return
+
         if self._t_poll is not None:
             self._running = False
             self._t_poll.join()
@@ -112,6 +124,9 @@ def set_light(self, r, g, b, brightness=None):
         :param b: Blue (0-255)
 
         """
+        if self._disable_led:
+            return
+
         self._led.set_pixel(0, r, g, b)
         if brightness is not None:
             self._led.set_brightness(0, brightness)

library/tests/conftest.py

@@ -0,0 +1,55 @@
+"""Test configuration.
+These allow the mocking of various Python modules
+that might otherwise have runtime side-effects.
+"""
+import sys
+import mock
+import pytest
+
+
+@pytest.fixture(scope='function', autouse=False)
+def FanShim():
+    import fanshim
+    yield fanshim.FanShim
+    del sys.modules['fanshim']
+
+
+@pytest.fixture(scope='function', autouse=False)
+def GPIO():
+    """Mock RPi.GPIO module."""
+    GPIO = mock.MagicMock()
+    # Fudge for Python < 37 (possibly earlier)
+    sys.modules['RPi'] = mock.Mock()
+    sys.modules['RPi'].GPIO = GPIO
+    sys.modules['RPi.GPIO'] = GPIO
+    yield GPIO
+    del sys.modules['RPi']
+    del sys.modules['RPi.GPIO']
+
+
+@pytest.fixture(scope='function', autouse=False)
+def apa102():
+    """Mock APA102 module."""
+    apa102 = mock.MagicMock()
+    sys.modules['apa102'] = apa102
+    yield apa102
+    del sys.modules['apa102']
+
+
+@pytest.fixture(scope='function', autouse=False)
+def spidev():
+    """Mock spidev module."""
+    spidev = mock.MagicMock()
+    sys.modules['spidev'] = spidev
+    yield spidev
+    del sys.modules['spidev']
+
+
+@pytest.fixture(scope='function', autouse=False)
+def atexit():
+    """Mock atexit module."""
+    atexit = mock.MagicMock()
+    sys.modules['atexit'] = atexit
+    yield atexit
+    del sys.modules['atexit']
+

library/tests/test_setup.py

@@ -2,11 +2,37 @@
 import sys
 
 
-def test_setup():
-    sys.modules['RPi'] = mock.Mock()
-    sys.modules['RPi.GPIO'] = mock.Mock()
-    sys.modules['plasma'] = mock.Mock()
-
-    from fanshim import FanShim
+def test_setup(GPIO, apa102, FanShim):
     fanshim = FanShim()
-    del fanshim
+
+    GPIO.setwarnings.assert_called_once_with(False)
+    GPIO.setmode.assert_called_once_with(GPIO.BCM)
+    GPIO.setup.assert_has_calls([
+        mock.call(fanshim._pin_fancontrol, GPIO.OUT),
+        mock.call(fanshim._pin_button, GPIO.IN, pull_up_down=GPIO.PUD_UP)
+    ])
+
+    apa102.APA102.assert_called_once_with(1, 15, 14, None, brightness=0.05)
+
+
+def test_button_disable(GPIO, apa102, FanShim):
+    fanshim = FanShim(disable_button=True)
+
+    GPIO.setwarnings.assert_called_once_with(False)
+    GPIO.setmode.assert_called_once_with(GPIO.BCM)
+    GPIO.setup.assert_called_once_with(fanshim._pin_fancontrol, GPIO.OUT)
+
+
+def test_led_disable(GPIO, apa102, FanShim):
+    fanshim = FanShim(disable_led=True)
+
+    GPIO.setwarnings.assert_called_once_with(False)
+    GPIO.setmode.assert_called_once_with(GPIO.BCM)
+    GPIO.setup.assert_has_calls([
+        mock.call(fanshim._pin_fancontrol, GPIO.OUT),
+        mock.call(fanshim._pin_button, GPIO.IN, pull_up_down=GPIO.PUD_UP)
+    ])
+
+    assert not apa102.APA102.called
+
+    fanshim.set_light(0, 0, 0)

library/tox.ini

@@ -1,5 +1,5 @@
 [tox]
-envlist = py{27,35},qa
+envlist = py{27,35,37},qa
 skip_missing_interpreters = True
 
 [testenv]