def test_read_small_table():
in_q = mock.Mock()
in_q.get.return_value = messages.ReadTableMsg(
table_addr = domain.TableAddr(
device_addr = domain.DeviceAddr(
gateway_addr=mock.Mock(), unit=0xFF),
id = 1),
recording_id="unique-id")
out_q = mock.Mock()
conn = mock.Mock()
with mock.patch('jem_data.core.modbus.read_registers'):
table_reader._read_table(in_q, out_q, conn)
out_q.put.assert_called_once_with(mock.ANY)
def test_read_large_table():
in_q = mock.Mock()
in_q.get.return_value = messages.ReadTableMsg(
table_addr = domain.TableAddr(
device_addr = domain.DeviceAddr(
gateway_addr=mock.Mock(), unit=0xFF),
id = 6),
recording_id="unique_id")
out_q = mock.Mock()
conn = mock.Mock()
with mock.patch('jem_data.core.modbus.read_registers'):
table_reader._read_table(in_q, out_q, conn)
## Table 6 is large, and requires more than 1 call.
## Check that more than 1 sub-result is being pushed on the queue.
nose.assert_greater(len(out_q.put.mock_calls), 1)
