def test_conversion_rq(self):
""" Check conversion to a -RQ PDU produces the correct output """
primitive = C_STORE()
primitive.MessageID = 7
primitive.AffectedSOPClassUID = '1.2.840.10008.5.1.4.1.1.2'
primitive.AffectedSOPInstanceUID = '1.2.392.200036.9116.2.6.1.48.' \
'1215709044.1459316254.522441'
primitive.Priority = 0x02
primitive.MoveOriginatorApplicationEntityTitle = 'UNITTEST_SCP'
primitive.MoveOriginatorMessageID = 3
ref_ds = Dataset()
ref_ds.PatientID = 'Test1101'
ref_ds.PatientName = "Tube HeNe"
primitive.DataSet = BytesIO(encode(ref_ds, True, True))
dimse_msg = C_STORE_RQ()
dimse_msg.primitive_to_message(primitive)
pdvs = dimse_msg.encode_msg(1, 16382)
cs_pdv = pdvs[0].presentation_data_value_list[0][1]
ds_pdv = pdvs[1].presentation_data_value_list[0][1]
self.assertEqual(cs_pdv, c_store_rq_cmd_b)
self.assertEqual(ds_pdv, c_store_rq_ds_b)
def test_message_to_primitive_c_store(self):
"""Test converting C_STORE_RQ and _RSP to C_STORE primitive."""
msg = C_STORE_RQ()
for data in [c_store_rq_cmd, c_store_ds]:
p_data = P_DATA()
p_data.presentation_data_value_list.append([0, data])
msg.decode_msg(p_data)
primitive = msg.message_to_primitive()
assert isinstance(primitive, C_STORE)
assert isinstance(primitive.DataSet, BytesIO)
assert primitive.AffectedSOPClassUID == UID('1.1.1')
assert primitive.AffectedSOPInstanceUID == UID('1.2.1')
assert primitive.Priority == 2
assert primitive.MoveOriginatorApplicationEntityTitle == b'UNITTEST '
assert primitive.MoveOriginatorMessageID == 3
ds = decode(primitive.DataSet, True, True)
assert ds.PatientName == 'Tube^HeNe'
assert ds.PatientID == 'Test1101'
msg = C_STORE_RSP()
p_data = P_DATA()
p_data.presentation_data_value_list.append([0, c_store_rsp_cmd])
msg.decode_msg(p_data)
primitive = msg.message_to_primitive()
assert isinstance(primitive, C_STORE)
assert primitive.DataSet is None
for elem in msg.command_set:
if hasattr(primitive, elem.keyword):
item = getattr(primitive, elem.keyword)
assert item == elem.value
def test_fragment_pdv(self):
"""Test that the PDV fragmenter is working correctly."""
dimse_msg = C_STORE_RQ()
frag = dimse_msg._bytestream_to_pdv
result = frag(c_echo_rsp_cmd, 1000)
self.assertEqual(len(result), 1)
self.assertEqual(result[0], c_echo_rsp_cmd)
self.assertTrue(isinstance(result[0], bytes))
self.assertTrue(result[-1] != b'')
result = frag(c_echo_rsp_cmd, 10)
self.assertEqual(len(result), 20)
self.assertEqual(result[0], c_echo_rsp_cmd[:4])
self.assertTrue(isinstance(result[0], bytes))
self.assertTrue(result[-1] != b'')
byteio = BytesIO()
byteio.write(c_echo_rsp_cmd)
result = frag(byteio, 1000)
self.assertEqual(len(result), 1)
self.assertEqual(result[0], c_echo_rsp_cmd)
self.assertTrue(isinstance(result[0], bytes))
with self.assertRaises(TypeError):
frag([], 10)
with self.assertRaises(TypeError):
frag(c_echo_rsp_cmd, 'test')
with self.assertRaises(ValueError):
frag(c_echo_rsp_cmd, 0)
def test_fragment_pdv(self):
"""Test that the PDV fragmenter is working correctly."""
dimse_msg = C_STORE_RQ()
frag = dimse_msg._generate_pdv_fragments
result = []
for fragment in frag(c_echo_rsp_cmd, 1000):
result.append(fragment)
assert len(result) == 1
assert result[0] == c_echo_rsp_cmd
assert isinstance(result[0], bytes)
assert result[-1] != b''
result = []
for fragment in frag(c_echo_rsp_cmd, 10):
result.append(fragment)
assert len(result) == 20
assert result[0] == c_echo_rsp_cmd[:4]
assert isinstance(result[0], bytes)
assert result[-1] != b''
with pytest.raises(ValueError):
next(frag(c_echo_rsp_cmd, 6))
def test_primitive_to_message(self):
"""Test converting a DIMSE primitive to a DIMSE message."""
primitive = C_STORE()
primitive.MessageID = 7
primitive.AffectedSOPClassUID = '1.1.1'
primitive.AffectedSOPInstanceUID = '1.2.1'
primitive.Priority = 0x02
msg = C_STORE_RQ()
msg.primitive_to_message(primitive)
# Test unused command set elements are deleted
assert not 'MoveOriginatorApplicationEntityTitle' in msg.command_set
# Test raise error for unknown DIMSE message type
msg.__class__.__name__ = 'TestClass'
with pytest.raises(ValueError):
msg.primitive_to_message(primitive)
# Reset name to avoid errors in other tests
msg.__class__.__name__ = 'C_STORE_RQ'
def test_encode(self):
"""Test encoding of a DIMSE message."""
primitive = C_STORE()
primitive.MessageID = 7
primitive.AffectedSOPClassUID = '1.1.1'
primitive.AffectedSOPInstanceUID = '1.2.1'
primitive.Priority = 0x02
primitive.MoveOriginatorApplicationEntityTitle = 'UNITTEST'
primitive.MoveOriginatorMessageID = 3
# Test encode without dataset
dimse_msg = C_STORE_RQ()
dimse_msg.primitive_to_message(primitive)
p_data_list = []
for pdata in dimse_msg.encode_msg(12, 16):
p_data_list.append(pdata)
assert p_data_list[0].presentation_data_value_list[0][1][
0:1] == b'\x01'
assert p_data_list[-1].presentation_data_value_list[0][1][
0:1] == b'\x03'
assert dimse_msg.ID == 12
# Test encode with dataset
ds = Dataset()
ds.PatientID = 'Test1101'
ds.PatientName = 'Tube^HeNe'
primitive.DataSet = BytesIO(encode(ds, True, True))
dimse_msg = C_STORE_RQ()
dimse_msg.primitive_to_message(primitive)
p_data_list = []
for pdata in dimse_msg.encode_msg(13, 10):
p_data_list.append(pdata)
assert p_data_list[0].presentation_data_value_list[0][1][
0:1] == b'\x01'
assert p_data_list[-1].presentation_data_value_list[0][1][
0:1] == b'\x02'
assert p_data_list[-2].presentation_data_value_list[0][1][
0:1] == b'\x00'
assert p_data_list[-10].presentation_data_value_list[0][1][
0:1] == b'\x03'
assert dimse_msg.ID == 13
p_data_list = []
for pdata in dimse_msg.encode_msg(1, 31682):
p_data_list.append(pdata)
assert p_data_list[0].presentation_data_value_list[0][
1] == c_store_rq_cmd
assert p_data_list[1].presentation_data_value_list[0][1] == c_store_ds
def test_decode(self):
"""Test decoding of a DIMSE message."""
primitive = C_STORE()
primitive.MessageID = 7
primitive.AffectedSOPClassUID = '1.1.1'
primitive.AffectedSOPInstanceUID = '1.2.1'
primitive.Priority = 0x02
primitive.MoveOriginatorApplicationEntityTitle = b'UNITTEST'
primitive.MoveOriginatorMessageID = 3
ds = Dataset()
ds.PatientID = 'Test1101'
ds.PatientName = 'Tube^HeNe'
primitive.DataSet = BytesIO(encode(ds, True, True))
dimse_msg = C_STORE_RQ()
dimse_msg.primitive_to_message(primitive)
# CMD: (1x4, 3x1), DS: (0x1, 2x1)
p_data = dimse_msg.encode_msg(12, 24)
# Command set decoding
dimse_msg.decode_msg(next(p_data)) # MCHB 1
dimse_msg.decode_msg(next(p_data)) # MCHB 1
dimse_msg.decode_msg(next(p_data)) # MCHB 1
dimse_msg.decode_msg(next(p_data)) # MCHB 1
dimse_msg.decode_msg(next(p_data)) # MCHB 3 - end of command set
assert dimse_msg.__class__ == C_STORE_RQ
# Test decoded command set
cs = dimse_msg.command_set
assert cs.CommandGroupLength == 102
assert cs.AffectedSOPClassUID == UID('1.1.1')
assert cs.AffectedSOPInstanceUID == UID('1.2.1')
assert cs.Priority == 2
assert cs.CommandDataSetType == 1
assert cs.CommandField == 1
assert cs.MoveOriginatorApplicationEntityTitle == 'UNITTEST '
assert cs.MoveOriginatorMessageID == 3
# Test decoded dataset
dimse_msg.decode_msg(next(p_data)) # MCHB 1 # MCHB 0
dimse_msg.decode_msg(next(p_data)) # MCHB 1 # MCHB 2
assert dimse_msg.data_set.getvalue() == c_store_ds[1:]
# Test returns false
msg = C_STORE_RSP()
assert not msg.decode_msg(c_store_rsp_cmd)
def test_decode(self):
"""Test decoding of a DIMSE message."""
primitive = C_STORE()
primitive.MessageID = 7
primitive.AffectedSOPClassUID = '1.1.1'
primitive.AffectedSOPInstanceUID = '1.2.1'
primitive.Priority = 0x02
primitive.MoveOriginatorApplicationEntityTitle = b'UNITTEST'
primitive.MoveOriginatorMessageID = 3
ds = Dataset()
ds.PatientID = 'Test1101'
ds.PatientName = 'Tube^HeNe'
primitive.DataSet = BytesIO(encode(ds, True, True))
dimse_msg = C_STORE_RQ()
dimse_msg.primitive_to_message(primitive)
# CMD: (1x4, 3x1), DS: (0x1, 2x1)
p_data_list = dimse_msg.encode_msg(12, 24)
# Command set decoding
for pdv in p_data_list[:4]:
dimse_msg.decode_msg(pdv) # MCHB 1
dimse_msg.decode_msg(p_data_list[4]) # MCHB 3 - end of command set
self.assertEqual(dimse_msg.__class__, C_STORE_RQ)
# Test decoded command set
cs = dimse_msg.command_set
self.assertTrue(cs.CommandGroupLength == 102)
self.assertTrue(cs.AffectedSOPClassUID == UID('1.1.1'))
self.assertTrue(cs.AffectedSOPInstanceUID == UID('1.2.1'))
self.assertTrue(cs.Priority == 2)
self.assertTrue(cs.CommandDataSetType == 1)
self.assertTrue(cs.CommandField == 1)
# Bug in pydicom -> AEs not stripped of trailing spaces
self.assertTrue(cs.MoveOriginatorApplicationEntityTitle == b'UNITTEST ')
self.assertTrue(cs.MoveOriginatorMessageID == 3)
# Test decoded dataset
dimse_msg.decode_msg(p_data_list[5]) # MCHB 0
dimse_msg.decode_msg(p_data_list[6]) # MCHB 2
self.assertTrue(dimse_msg.data_set.getvalue() == c_store_ds[1:])
# Test returns false
msg = C_STORE_RSP()
self.assertFalse(msg.decode_msg(c_store_rsp_cmd))
def send_msg(self, primitive, context_id):
"""Send a DIMSE-C or DIMSE-N message to the peer AE.
Parameters
----------
primitive : pynetdicom3.dimse_primitives
The DIMSE service primitive to send to the peer.
context_id : int
The ID of the presentation context to be sent under.
"""
if primitive.__class__ == C_ECHO:
if primitive.MessageID is not None:
dimse_msg = C_ECHO_RQ()
else:
dimse_msg = C_ECHO_RSP()
elif primitive.__class__ == C_STORE:
if primitive.MessageID is not None:
dimse_msg = C_STORE_RQ()
else:
dimse_msg = C_STORE_RSP()
elif primitive.__class__ == C_FIND:
if primitive.MessageID is not None:
dimse_msg = C_FIND_RQ()
else:
dimse_msg = C_FIND_RSP()
elif primitive.__class__ == C_GET:
if primitive.MessageID is not None:
dimse_msg = C_GET_RQ()
else:
dimse_msg = C_GET_RSP()
elif primitive.__class__ == C_MOVE:
if primitive.MessageID is not None:
dimse_msg = C_MOVE_RQ()
else:
dimse_msg = C_MOVE_RSP()
elif primitive.__class__ == C_CANCEL:
dimse_msg = C_CANCEL_RQ()
elif primitive.__class__ == N_EVENT_REPORT:
if primitive.MessageID is not None:
dimse_msg = N_EVENT_REPORT_RQ()
else:
dimse_msg = N_EVENT_REPORT_RSP()
elif primitive.__class__ == N_GET:
if primitive.MessageID is not None:
dimse_msg = N_GET_RQ()
else:
dimse_msg = N_GET_RSP()
elif primitive.__class__ == N_SET:
if primitive.MessageID is not None:
dimse_msg = N_SET_RQ()
else:
dimse_msg = N_SET_RSP()
elif primitive.__class__ == N_ACTION:
if primitive.MessageID is not None:
dimse_msg = N_ACTION_RQ()
else:
dimse_msg = N_ACTION_RSP()
elif primitive.__class__ == N_CREATE:
if primitive.MessageID is not None:
dimse_msg = N_CREATE_RQ()
else:
dimse_msg = N_CREATE_RSP()
elif primitive.__class__ == N_DELETE:
if primitive.MessageID is not None:
dimse_msg = N_DELETE_RQ()
else:
dimse_msg = N_DELETE_RSP()
# Convert DIMSE primitive to DIMSE Message
dimse_msg.primitive_to_message(primitive)
# Callbacks
self.on_send_dimse_message(dimse_msg)
# Split the full messages into P-DATA chunks,
# each below the max_pdu size
pdata_pdu_list = dimse_msg.encode_msg(context_id,
self.maximum_pdu_size)
# Send the P-DATA PDUs to the peer via the DUL provider
for pdata_pdu in pdata_pdu_list:
self.dul.send_pdu(pdata_pdu)
def test_callback_receive_c_store(self):
"""Check callback for sending DIMSE C-STORE messages."""
# C-STORE-RQ
primitive = C_STORE()
primitive.MessageID = 7
primitive.AffectedSOPClassUID = '1.2.840.10008.5.1.4.1.1.2'
primitive.AffectedSOPInstanceUID = '1.2.392.200036.9116.2.6.1.48.' \
'1215709044.1459316254.522441'
primitive.Priority = 0x02
primitive.MoveOriginatorApplicationEntityTitle = 'UNITTEST_SCP'
primitive.MoveOriginatorMessageID = 3
primitive.DataSet = BytesIO()
# No dataset
msg = C_STORE_RQ()
msg.primitive_to_message(primitive)
self.dimse.debug_receive_c_store_rq(msg)
# Dataset
ref_ds = Dataset()
ref_ds.PatientID = 'Test1101'
ref_ds.PatientName = "Tube HeNe"
primitive.DataSet = BytesIO(encode(ref_ds, True, True))
msg = C_STORE_RQ()
msg.primitive_to_message(primitive)
# Dataset
self.dimse.debug_receive_c_store_rq(msg)
# C-STORE-RSP
primitive = C_STORE()
primitive.MessageIDBeingRespondedTo = 7
primitive.AffectedSOPClassUID = '1.2.840.10008.5.1.4.1.1.2'
primitive.AffectedSOPInstanceUID = '1.2.392.200036.9116.2.6.1.48.' \
'1215709044.1459316254.522441'
primitive.MoveOriginatorApplicationEntityTitle = 'UNITTEST_SCP'
primitive.MoveOriginatorMessageID = 3
primitive.DataSet = BytesIO()
# Check statuses + no dataset
for status in [0x0000, 0xb000, 0xb007, 0xb006, 0xa700, 0xa900, 0xc000]:
primitive.Status = status
msg = C_STORE_RSP()
msg.primitive_to_message(primitive)
self.dimse.debug_receive_c_store_rsp(msg)
# Dataset
ref_ds = Dataset()
ref_ds.PatientID = 'Test1101'
ref_ds.PatientName = "Tube HeNe"
msg = C_STORE_RSP()
msg.primitive_to_message(primitive)
# Dataset
msg.data_set = BytesIO(encode(ref_ds, True, True))
self.dimse.debug_receive_c_store_rsp(msg)