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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
#!/usr/bin/env python
"""
Bit Message Test Fixture
--------------------------------
This fixture tests the functionality of all the
bit based request/response messages:
* Read/Write Discretes
* Read Coils
"""
import unittest, struct
from pymodbus.bit_read_message import *
from pymodbus.bit_read_message import ReadBitsRequestBase
from pymodbus.bit_read_message import ReadBitsResponseBase
from pymodbus.exceptions import *
from pymodbus.pdu import ModbusExceptions
from pymodbus.compat import iteritems
from .modbus_mocks import MockContext
res = [True] * 21
res.extend([False] * 3)
#---------------------------------------------------------------------------#
# Fixture
#---------------------------------------------------------------------------#
class ModbusBitMessageTests(unittest.TestCase):
#-----------------------------------------------------------------------#
# Setup/TearDown
#-----------------------------------------------------------------------#
def setUp(self):
"""
Initializes the test environment and builds request/result
encoding pairs
"""
pass
def tearDown(self):
""" Cleans up the test environment """
pass
def testReadBitBaseClassMethods(self):
""" Test basic bit message encoding/decoding """
handle = ReadBitsRequestBase(1, 1)
msg = "ReadBitRequest(1,1)"
self.assertEqual(msg, str(handle))
handle = ReadBitsResponseBase([1,1])
msg = "ReadBitResponse(2)"
self.assertEqual(msg, str(handle))
def testBitReadBaseRequestEncoding(self):
""" Test basic bit message encoding/decoding """
for i in range(20):
handle = ReadBitsRequestBase(i, i)
result = struct.pack('>HH',i, i)
self.assertEqual(handle.encode(), result)
handle.decode(result)
self.assertEqual((handle.address, handle.count), (i,i))
def testBitReadBaseResponseEncoding(self):
""" Test basic bit message encoding/decoding """
for i in range(20):
input = [True] * i
handle = ReadBitsResponseBase(input)
result = handle.encode()
handle.decode(result)
self.assertEqual(handle.bits[:i], input)
def testBitReadBaseResponseHelperMethods(self):
""" Test the extra methods on a ReadBitsResponseBase """
input = [False] * 8
handle = ReadBitsResponseBase(input)
for i in [1,3,5]: handle.setBit(i, True)
for i in [1,3,5]: handle.resetBit(i)
for i in range(8):
self.assertEqual(handle.getBit(i), False)
def testBitReadBaseRequests(self):
""" Test bit read request encoding """
messages = {
ReadBitsRequestBase(12, 14) : b'\x00\x0c\x00\x0e',
ReadBitsResponseBase([1,0,1,1,0]) : b'\x01\x0d'
}
for request, expected in iteritems(messages):
self.assertEqual(request.encode(), expected)
def testBitReadMessageExecuteValueErrors(self):
""" Test bit read request encoding """
context = MockContext()
requests = [
ReadCoilsRequest(1,0x800),
ReadDiscreteInputsRequest(1,0x800),
]
for request in requests:
result = request.execute(context)
self.assertEqual(ModbusExceptions.IllegalValue,
result.exception_code)
def testBitReadMessageExecuteAddressErrors(self):
""" Test bit read request encoding """
context = MockContext()
requests = [
ReadCoilsRequest(1,5),
ReadDiscreteInputsRequest(1,5),
]
for request in requests:
result = request.execute(context)
self.assertEqual(ModbusExceptions.IllegalAddress, result.exception_code)
def testBitReadMessageExecuteSuccess(self):
""" Test bit read request encoding """
context = MockContext()
context.validate = lambda a,b,c: True
requests = [
ReadCoilsRequest(1,5),
ReadDiscreteInputsRequest(1,5),
]
for request in requests:
result = request.execute(context)
self.assertEqual(result.bits, [True] * 5)
def testBitReadMessageGetResponsePDU(self):
requests = {
ReadCoilsRequest(1,5): 3,
ReadCoilsRequest(1, 8): 3,
ReadCoilsRequest(0, 16): 4,
ReadDiscreteInputsRequest(1, 21): 5,
ReadDiscreteInputsRequest(1, 24): 5,
ReadDiscreteInputsRequest(1, 1900): 240,
}
for request, expected in iteritems(requests):
pdu_len = request.get_response_pdu_size()
self.assertEqual(pdu_len, expected)
#---------------------------------------------------------------------------#
# Main
#---------------------------------------------------------------------------#
if __name__ == "__main__":
unittest.main()
|
