def test_parse_response_with_wrong_header(self):
protocol = IEC60488()
with pytest.raises(ValueError) as excinfo:
protocol.parse_response(b'DATA,DATA', 'HEADER')
assert str(excinfo.value) == 'Response header mismatch'
def test_query(self):
protocol = IEC60488()
transport = MockTransport(responses=[b'DATA,DATA\n'])
assert protocol.query(transport, 'HEADER') == ['DATA', 'DATA']
assert transport.messages[0] == b'HEADER\n'
def test_parse_response_without_header_prefix_and_multiple_data(self):
protocol = IEC60488()
assert protocol.parse_response(b'DATA,DATA,DATA') == [
'DATA', 'DATA', 'DATA'
]
def test_parse_response_with_header_and_prefix(self):
protocol = IEC60488(resp_prefix='PREFIX:', resp_header_sep=' ')
assert protocol.parse_response(b'PREFIX:HEADER DATA,DATA',
'HEADER') == ['DATA', 'DATA']
def test_write_with_data(self):
protocol = IEC60488()
transport = MockTransport()
protocol.write(transport, 'HEADER', 'D1', 'D2', 'D3')
assert transport.messages[0] == b'HEADER D1,D2,D3\n'
def test_write_without_data(self):
protocol = IEC60488()
transport = MockTransport()
protocol.write(transport, 'HEADER')
assert transport.messages[0] == b'HEADER\n'
def test_create_message_with_multiple_data(self):
protocol = IEC60488(msg_prefix='PREFIX:')
assert protocol.create_message('HEADER', 'D1', 'D2',
'D3') == b'PREFIX:HEADER D1,D2,D3\n'
def test_create_message_with_single_data(self):
protocol = IEC60488(msg_prefix='PREFIX:')
assert protocol.create_message('HEADER',
'DATA') == b'PREFIX:HEADER DATA\n'
def test_create_message_without_data(self):
protocol = IEC60488(msg_prefix='PREFIX:')
assert protocol.create_message('HEADER') == b'PREFIX:HEADER\n'