1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (c) 2019 Richard Hull and contributors
# See LICENSE.rst for details.
"""
Tests for the :py:class:`luma.core.interface.serial.ftdi_spi` class.
"""
from unittest.mock import Mock, call, patch
from luma.core.interface.serial import ftdi_spi
from helpers import fib, assert_only_cleans_whats_setup
@patch('pyftdi.spi.SpiController')
def test_init(mock_controller):
gpio = Mock()
gpio.width = 8
instance = Mock()
instance.get_port = Mock()
instance.get_gpio = Mock(return_value=gpio)
mock_controller.side_effect = [instance]
ftdi_spi(device='ftdi://dummy', bus_speed_hz=16000000, gpio_CS=3, gpio_DC=5, gpio_RST=6)
mock_controller.assert_called_with(cs_count=1)
instance.configure.assert_called_with('ftdi://dummy')
instance.get_port.assert_called_with(cs=0, freq=16000000, mode=0)
instance.get_gpio.assert_called_with()
gpio.set_direction.assert_called_with(0x60, 0x60)
@patch('pyftdi.spi.SpiController')
def test_command(mock_controller):
cmds = [3, 1, 4, 2]
gpio = Mock()
gpio.width = 8
port = Mock()
instance = Mock()
instance.get_port = Mock(return_value=port)
instance.get_gpio = Mock(return_value=gpio)
mock_controller.side_effect = [instance]
serial = ftdi_spi(device='ftdi://dummy', bus_speed_hz=16000000, gpio_CS=3, gpio_DC=5, gpio_RST=6)
serial.command(*cmds)
gpio.write.assert_has_calls([call(0x00), call(0x40), call(0x40)])
port.write.assert_called_once_with(cmds)
@patch('pyftdi.spi.SpiController')
def test_data(mock_controller):
data = list(fib(100))
gpio = Mock()
gpio.width = 8
port = Mock()
instance = Mock()
instance.get_port = Mock(return_value=port)
instance.get_gpio = Mock(return_value=gpio)
mock_controller.side_effect = [instance]
serial = ftdi_spi(device='ftdi://dummy', bus_speed_hz=16000000, gpio_CS=3, gpio_DC=5, gpio_RST=6)
serial.data(data)
gpio.write.assert_has_calls([call(0x00), call(0x40), call(0x60)])
port.write.assert_called_once_with(data)
@patch('pyftdi.spi.SpiController')
def test_cleanup(mock_controller):
gpio = Mock()
gpio.width = 8
port = Mock()
instance = Mock()
instance.get_port = Mock(return_value=port)
instance.get_gpio = Mock(return_value=gpio)
mock_controller.side_effect = [instance]
serial = ftdi_spi(device='ftdi://dummy', bus_speed_hz=16000000, gpio_CS=3, gpio_DC=5, gpio_RST=6)
serial.cleanup()
instance.terminate.assert_called_once_with()
assert_only_cleans_whats_setup(gpio)
|