def test_conversion_rsp(self):
""" Check conversion to a -RSP PDU produces the correct output """
primitive = N_GET()
primitive.MessageIDBeingRespondedTo = 5
primitive.AffectedSOPClassUID = '1.2.4.10'
primitive.AffectedSOPInstanceUID = '1.2.4.5.7.8'
primitive.Status = 0x0000
ds = Dataset()
ds.PatientID = 'Test1101'
ds.PatientName = "Tube HeNe"
primitive.AttributeList = BytesIO(encode(ds, True, True))
dimse_msg = N_GET_RSP()
dimse_msg.primitive_to_message(primitive)
pdvs = []
for fragment in dimse_msg.encode_msg(1, 16382):
pdvs.append(fragment)
assert len(pdvs) == 2
cs_pdv = pdvs[0].presentation_data_value_list[0][1]
ds_pdv = pdvs[1].presentation_data_value_list[0][1]
assert cs_pdv == n_get_rsp_cmd
assert ds_pdv == n_get_rsp_ds
def test_callback_receive_n_set(self):
"""Check callback for receiving DIMSE N-SET messages."""
# N-SET-RQ
primitive = N_SET()
primitive.MessageID = 1
primitive.RequestedSOPClassUID = '1.1.1'
primitive.RequestedSOPInstanceUID = '1.1.1.1'
primitive.ModificationList = BytesIO(n_set_rq_ds)
msg = N_SET_RQ()
msg.primitive_to_message(primitive)
msg.ID = 1
self.dimse.debug_receive_n_set_rq(msg)
# N-SET-RSP
primitive = N_SET()
primitive.MessageIDBeingRespondedTo = 1
primitive.Status = 0x0000
msg = N_SET_RSP()
msg.primitive_to_message(primitive)
msg.ID = 1
self.dimse.debug_receive_n_set_rsp(msg)
# User defined
primitive.AffectedSOPClassUID = '1.1.1'
primitive.AffectedSOPInstanceUID = '1.1.1.1'
primitive.ModificationList = BytesIO(n_set_rsp_ds)
msg = N_GET_RSP()
msg.primitive_to_message(primitive)
msg.ID = 1
self.dimse.debug_receive_n_set_rsp(msg)
def test_callback_receive_n_get(self):
"""Check callback for receiving DIMSE N-GET messages."""
# N-GET-RQ
msg = N_GET_RQ()
self.dimse.debug_receive_n_get_rq(msg)
# N-GET-RSP
msg = N_GET_RSP()
self.dimse.debug_receive_n_get_rsp(msg)
def test_callback_receive_n_get(self):
"""Check callback for receiving DIMSE N-GET messages."""
# N-GET-RQ
primitive = N_GET()
primitive.MessageID = 1
primitive.RequestedSOPClassUID = '1.1.1'
primitive.RequestedSOPInstanceUID = '1.1.1.1'
msg = N_GET_RQ()
msg.primitive_to_message(primitive)
msg.ID = 1
self.dimse.debug_receive_n_get_rq(msg)
# Plus user defined
primitive.AttributeIdentifierList = [(0x0000, 0x0000),
(0xffff, 0xffff)]
msg = N_GET_RQ()
msg.primitive_to_message(primitive)
msg.ID = 1
self.dimse.debug_receive_n_get_rq(msg)
# N-GET-RSP
# Mandatory elements
primitive = N_GET()
primitive.MessageIDBeingRespondedTo = 1
primitive.Status = 0x0000
msg = N_GET_RSP()
msg.primitive_to_message(primitive)
msg.ID = 1
self.dimse.debug_receive_n_get_rsp(msg)
# User defined
primitive.AffectedSOPClassUID = '1.1.1'
primitive.AffectedSOPInstanceUID = '1.1.1.1'
msg = N_GET_RSP()
msg.primitive_to_message(primitive)
msg.ID = 1
self.dimse.debug_receive_n_get_rsp(msg)
# Conditional
primitive.AttributeList = BytesIO(n_get_rsp_ds)
msg = N_GET_RSP()
msg.primitive_to_message(primitive)
msg.ID = 1
self.dimse.debug_receive_n_get_rsp(msg)
def test_message_to_primitive_n_get(self):
"""Test converting N_GET_RQ and _RSP to primitive."""
# N-GET-RQ
msg = N_GET_RQ()
primitive = msg.message_to_primitive()
assert isinstance(primitive, N_GET)
# N-GET-RSP
msg = N_GET_RSP()
primitive = msg.message_to_primitive()
assert isinstance(primitive, N_GET)
def test_message_to_primitive_n_get(self):
"""Test converting N_GET_RQ and _RSP to primitive."""
# N-GET-RQ
msg = N_GET_RQ()
for data in [n_get_rq_cmd]:
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, N_GET)
assert primitive.RequestedSOPClassUID == UID(
'1.2.840.10008.5.1.4.1.1.2')
assert primitive.RequestedSOPInstanceUID == UID(
'1.2.392.200036.9116.2.6.1.48')
assert primitive.MessageID == 7
primitive.AttributeIdentifierList = [(0x7fe0, 0x0010),
(0x0000, 0x0000),
(0xFFFF, 0xFFFF)]
msg = N_GET_RSP()
for data in [n_get_rsp_cmd, n_get_rsp_ds]:
p_data = P_DATA()
p_data.presentation_data_value_list.append([0, data])
msg.decode_msg(p_data)
msg.decode_msg(p_data)
primitive = msg.message_to_primitive()
assert isinstance(primitive, N_GET)
assert primitive.AttributeIdentifierList is None
assert primitive.AffectedSOPClassUID == UID('1.2.4.10')
assert primitive.AffectedSOPInstanceUID == UID('1.2.4.5.7.8')
assert primitive.MessageIDBeingRespondedTo == 5
assert primitive.Status == 0x0000
ds = decode(primitive.AttributeList, True, True)
assert ds.PatientName == 'Tube HeNe'
assert ds.PatientID == 'Test1101'
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)