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
|
#!/usr/bin/env python
import unittest
from pymodbus.datastore import *
from pymodbus.exceptions import NoSuchSlaveException
from pymodbus.compat import iteritems
class ModbusServerSingleContextTest(unittest.TestCase):
''' This is the unittest for the pymodbus.datastore.ModbusServerContext
using a single slave context.
'''
def setUp(self):
''' Sets up the test environment '''
self.slave = ModbusSlaveContext()
self.context = ModbusServerContext(slaves=self.slave, single=True)
def tearDown(self):
''' Cleans up the test environment '''
del self.context
def testSingleContextGets(self):
''' Test getting on a single context '''
for id in range(0, 0xff):
self.assertEqual(self.slave, self.context[id])
def testSingleContextDeletes(self):
''' Test removing on multiple context '''
def _test():
del self.context[0x00]
self.assertRaises(NoSuchSlaveException, _test)
def testSingleContextIter(self):
''' Test iterating over a single context '''
expected = (0, self.slave)
for slave in self.context:
self.assertEqual(slave, expected)
def testSingleContextDefault(self):
''' Test that the single context default values work '''
self.context = ModbusServerContext()
slave = self.context[0x00]
self.assertEqual(slave, {})
def testSingleContextSet(self):
''' Test a setting a single slave context '''
slave = ModbusSlaveContext()
self.context[0x00] = slave
actual = self.context[0x00]
self.assertEqual(slave, actual)
class ModbusServerMultipleContextTest(unittest.TestCase):
''' This is the unittest for the pymodbus.datastore.ModbusServerContext
using multiple slave contexts.
'''
def setUp(self):
''' Sets up the test environment '''
self.slaves = dict((id, ModbusSlaveContext()) for id in range(10))
self.context = ModbusServerContext(slaves=self.slaves, single=False)
def tearDown(self):
''' Cleans up the test environment '''
del self.context
def testMultipleContextGets(self):
''' Test getting on multiple context '''
for id in range(0, 10):
self.assertEqual(self.slaves[id], self.context[id])
def testMultipleContextDeletes(self):
''' Test removing on multiple context '''
del self.context[0x00]
self.assertRaises(NoSuchSlaveException, lambda: self.context[0x00])
def testMultipleContextIter(self):
''' Test iterating over multiple context '''
for id, slave in self.context:
self.assertEqual(slave, self.slaves[id])
self.assertTrue(id in self.context)
def testMultipleContextDefault(self):
''' Test that the multiple context default values work '''
self.context = ModbusServerContext(single=False)
self.assertRaises(NoSuchSlaveException, lambda: self.context[0x00])
def testMultipleContextSet(self):
''' Test a setting multiple slave contexts '''
slaves = dict((id, ModbusSlaveContext()) for id in range(10))
for id, slave in iteritems(slaves):
self.context[id] = slave
for id, slave in iteritems(slaves):
actual = self.context[id]
self.assertEqual(slave, actual)
#---------------------------------------------------------------------------#
# Main
#---------------------------------------------------------------------------#
if __name__ == "__main__":
unittest.main()
|
