QA: Add basic tests.

Author: Gadgetoid

.gitignore

@@ -1,22 +1,20 @@
-__pycache__/
+build/
+_build/
+*.o
+*.so
+*.a
 *.py[cod]
+*.egg-info
 dist/
-sdist/
-env/
-build/
-develop-eggs/
-eggs/
-*.egg-info/
-.installed.cfg
-*.egg
+__pycache__
+.DS_Store
 *.deb
 *.dsc
 *.build
 *.changes
 *.orig.*
-library/debian/
 packaging/*tar.xz
-pip-log.txt
-pip-delete-this-directory.txt
-.DS_Store
-
+library/debian/
+.coverage
+.pytest_cache
+.tox

tests/conftest.py

@@ -0,0 +1,47 @@
+import sys
+
+import mock
+import pytest
+
+
+# MockSMbus borrowed from https://github.com/pimoroni/i2cdevice-python
+class MockSMBus:
+    def __init__(self, i2c_bus, default_registers=None):
+        self.regs = [0 for _ in range(255)]
+        if default_registers is not None:
+            for index in default_registers.keys():
+                self.regs[index] = default_registers.get(index)
+
+    def write_i2c_block_data(self, i2c_address, register, values):
+        self.regs[register:register + len(values)] = values
+
+    def read_i2c_block_data(self, i2c_address, register, length):
+        return self.regs[register:register + length]
+
+    def read_byte_data(self, i2c_address, register):
+        return self.read_i2c_block_data(i2c_address, register, 1)[0]
+
+    def write_byte_data(self, i2c_address, register, value):
+        self.write_i2c_block_data(i2c_address, register, [value])
+
+
+@pytest.fixture(scope="function", autouse=True)
+def cleanup():
+    yield
+    del sys.modules["cap1xxx"]
+
+
+@pytest.fixture(scope="function")
+def smbus2():
+    sys.modules["smbus2"] = mock.MagicMock()
+    yield sys.modules["smbus2"]
+    del sys.modules["smbus2"]
+
+
+@pytest.fixture(scope="function")
+def gpio():
+    sys.modules["RPi"] = mock.Mock()
+    sys.modules["RPi.GPIO"] = mock.MagicMock()
+    yield
+    del sys.modules["RPi"]
+    del sys.modules["RPi.GPIO"]

tests/test_io.py

@@ -0,0 +1,29 @@
+from conftest import MockSMBus
+
+
+def test_inputs(gpio, smbus2):
+    import cap1xxx
+
+    smbus2.SMBus.return_value = MockSMBus(1, default_registers={
+        cap1xxx.R_PRODUCT_ID: cap1xxx.PID_CAP1208,
+        cap1xxx.R_INPUT_STATUS: 0b10101010
+    })
+
+    cap1208 = cap1xxx.Cap1208()
+    assert cap1208.get_input_status() == [ "none", "press", "none", "press", "none", "press", "none", "press"]
+
+
+def test_outputs(gpio, smbus2):
+    import cap1xxx
+
+    mock = MockSMBus(1, default_registers={
+        cap1xxx.R_PRODUCT_ID: cap1xxx.PID_CAP1188,
+        cap1xxx.R_INPUT_STATUS: 0b10101010
+    })
+
+    smbus2.SMBus.return_value = mock
+
+    cap1188 = cap1xxx.Cap1188()
+    cap1188.set_led_state(0, True)
+
+    assert mock.read_byte_data(cap1xxx.DEFAULT_ADDR, cap1xxx.R_LED_OUTPUT_CON) == 0b00000001

tests/test_setup.py

@@ -0,0 +1,26 @@
+import pytest
+
+
+def test_setup(gpio, smbus2):
+    import cap1xxx
+
+    smbus2.SMBus(1).read_byte_data.return_value = cap1xxx.PID_CAP1208
+    cap1208 = cap1xxx.Cap1208()
+    assert cap1208.number_of_leds == 8
+
+    smbus2.SMBus(1).read_byte_data.return_value = cap1xxx.PID_CAP1188
+    cap1188 = cap1xxx.Cap1188()
+    assert cap1188.number_of_leds == 8
+
+    smbus2.SMBus(1).read_byte_data.return_value = cap1xxx.PID_CAP1166
+    cap1166 = cap1xxx.Cap1166()
+    assert cap1166.number_of_leds == 6
+
+
+def test_setup_invalid_pid(gpio, smbus2):
+    import cap1xxx
+
+    smbus2.SMBus(1).read_byte_data.return_value = 0x00
+
+    with pytest.raises(RuntimeError):
+        _ = cap1xxx.Cap1208()