def test_is_valid_resposne(self):
"""Test N_SET.is_valid_response."""
primitive = N_SET()
assert not primitive.is_valid_response
primitive.MessageIDBeingRespondedTo = 1
assert not primitive.is_valid_response
primitive.Status = 0x0000
assert primitive.is_valid_response
def message_to_primitive(self):
"""Convert the ``DIMSEMessage`` class to a DIMSE primitive.
Returns
-------
DIMSEPrimitive sub-class
One of the DIMSE message primitives from
:ref:`pynetdicom.dimse_primitives<api_dimse_primitives>` generated
from the current ``DIMSEMessage`` sub-class object.
"""
# pylint: disable=too-many-branches
cls_type_name = self.__class__.__name__
if 'C_ECHO' in cls_type_name:
primitive = C_ECHO()
elif 'C_STORE' in cls_type_name:
primitive = C_STORE()
elif 'C_FIND' in cls_type_name:
primitive = C_FIND()
elif 'C_GET' in cls_type_name:
primitive = C_GET()
elif 'C_MOVE' in cls_type_name:
primitive = C_MOVE()
elif 'C_CANCEL' in cls_type_name:
primitive = C_CANCEL()
elif 'N_EVENT' in cls_type_name:
primitive = N_EVENT_REPORT()
elif 'N_GET' in cls_type_name:
primitive = N_GET()
elif 'N_SET' in cls_type_name:
primitive = N_SET()
elif 'N_ACTION' in cls_type_name:
primitive = N_ACTION()
elif 'N_CREATE' in cls_type_name:
primitive = N_CREATE()
elif 'N_DELETE' in cls_type_name:
primitive = N_DELETE()
# Command Set
# For each parameter in the primitive, set the appropriate value
# from the Message's Command Set elements
for elem in self.command_set:
if hasattr(primitive, elem.keyword):
setattr(
primitive,
elem.keyword,
self.command_set.__getattr__(elem.keyword)
)
# Datasets
# Set the primitive's DataSet/Identifier/etc attribute
try:
dataset_keyword = _DATASET_KEYWORDS[cls_type_name]
setattr(primitive, dataset_keyword, self.data_set)
except KeyError:
pass
# Set the presentation context ID the message was set under
primitive._context_id = self.context_id
return primitive
def test_conversion_rsp(self):
""" Check conversion to a -RSP PDU produces the correct output """
primitive = N_SET()
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_SET_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_set_rsp_cmd
assert ds_pdv == n_set_rsp_ds
def test_conversion_rq(self):
""" Check conversion to a -RQ PDU produces the correct output """
primitive = N_SET()
primitive.MessageID = 7
primitive.RequestedSOPClassUID = '1.2.840.10008.5.1.4.1.1.2'
primitive.RequestedSOPInstanceUID = '1.2.392.200036.9116.2.6.1.48'
ds = Dataset()
ds.PatientID = 'Test1101'
ds.PatientName = "Tube HeNe"
primitive.ModificationList = BytesIO(encode(ds, True, True))
dimse_msg = N_SET_RQ()
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_set_rq_cmd
assert ds_pdv == n_set_rq_ds
def test_is_valid_request(self):
"""Test N_SET.is_valid_request"""
primitive = N_SET()
assert not primitive.is_valid_request
primitive.MessageID = 1
assert not primitive.is_valid_request
primitive.RequestedSOPClassUID = '1.2'
assert not primitive.is_valid_request
primitive.RequestedSOPInstanceUID = '1.2.1'
assert not primitive.is_valid_request
ds = Dataset()
ds.PatientID = 'Test1101'
ds.PatientName = "Tube HeNe"
primitive.ModificationList = BytesIO(encode(ds, True, True))
assert primitive.is_valid_request
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.context_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.context_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.context_id = 1
self.dimse.debug_receive_n_set_rsp(msg)
def test_callback_send_n_set(self):
"""Check callback for sending 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(b'\x00\x01')
self.dimse.send_msg(primitive, 1)
# N-SET-RSP
primitive = N_SET()
primitive.MessageIDBeingRespondedTo = 1
primitive.Status = 0x0000
self.dimse.send_msg(primitive, 1)
# User defined
primitive.AffectedSOPClassUID = '1.1.1'
primitive.AffectedSOPInstanceUID = '1.1.1.1'
primitive.AttributeList = BytesIO(b'\x00\x01')
self.dimse.send_msg(primitive, 1)
@staticmethod
def peek_next_pdu():
return 0x01
REFERENCE_MSG = [
(C_ECHO(), ('C_ECHO_RQ', 'C_ECHO_RSP')),
(C_STORE(), ('C_STORE_RQ', 'C_STORE_RSP')),
(C_FIND(), ('C_FIND_RQ', 'C_FIND_RSP')),
(C_GET(), ('C_GET_RQ', 'C_GET_RSP')),
(C_MOVE(), ('C_MOVE_RQ', 'C_MOVE_RSP')),
(C_CANCEL(), (None, 'C_CANCEL_RQ')),
(N_EVENT_REPORT(), ('N_EVENT_REPORT_RQ', 'N_EVENT_REPORT_RSP')),
(N_GET(), ('N_GET_RQ', 'N_GET_RSP')),
(N_SET(), ('N_SET_RQ', 'N_SET_RSP')),
(N_ACTION(), ('N_ACTION_RQ', 'N_ACTION_RSP')),
(N_CREATE(), ('N_CREATE_RQ', 'N_CREATE_RSP')),
(N_DELETE(), ('N_DELETE_RQ', 'N_DELETE_RSP')),
]
class TestDIMSEProvider(object):
"""Test DIMSE service provider operations."""
def setup(self):
"""Set up"""
self.dimse = DIMSEServiceProvider(DummyDUL(), 1)
def test_receive_not_pdata(self):
"""Test we get back None if not a P_DATA"""
assert self.dimse.receive_msg(True) == (None, None)
def _set_scp(self, req, context):
"""Implementation of the N-SET SCP.
Parameters
----------
req : dimse_primitives.N_SET
The N-SET request primitive sent by the peer.
context : presentation.PresentationContext
The presentation context that the SCP is operating under.
Notes
-----
**N-SET Request**
*Parameters*
| (M) Message ID
| (M) Requested SOP Class UID
| (M) Requested SOP Instance UID
| (M) Modification List
**N-SET Response**
*Parameters*
| (M) Message ID Being Responded To
| (U) Attribute List
| (U) Affected SOP Class UID
| (U) Affected SOP Instance UID
| (M) Status
*Status*
Success
| ``0x0000`` - Success
Failure
| ``0x0105`` - No such attribute
| ``0x0106`` - Invalid attribute value
| ``0x0110`` - Processing failure
| ``0x0112`` - SOP Instance not recognised
| ``0x0116`` - Attribute value out of range
| ``0x0117`` - Invalid object instance
| ``0x0118`` - No such SOP Class
| ``0x0119`` - Class-Instance conflict
| ``0x0121`` - Missing attribute value
| ``0x0124`` - Refused: not authorised
| ``0x0210`` - Duplicate invocation
| ``0x0211`` - Unrecognised operation
| ``0x0212`` - Mistyped argument
| ``0x0213`` - Resource limitation
References
----------
* DICOM Standard, Part 4, `Annex F <http://dicom.nema.org/medical/dicom/current/output/html/part04.html#chapter_F>`_
* DICOM Standard, Part 7, Sections
`10.1.3 <http://dicom.nema.org/medical/dicom/current/output/html/part07.html#sect_10.1.3>`_,
`10.3.3 <http://dicom.nema.org/medical/dicom/current/output/html/part07.html#sect_10.3.3>`_
and `Annex C <http://dicom.nema.org/medical/dicom/current/output/html/part07.html#chapter_C>`_
"""
# Build N-CREATE response primitive
rsp = N_SET()
rsp.MessageIDBeingRespondedTo = req.MessageID
rsp.AffectedSOPClassUID = req.RequestedSOPClassUID
rsp.AffectedSOPInstanceUID = req.RequestedSOPInstanceUID
try:
status, ds = evt.trigger(
self.assoc,
evt.EVT_N_SET,
{'request' : req, 'context' : context.as_tuple}
)
except Exception as exc:
LOGGER.error(
"Exception in the handler bound to 'evt.EVT_N_SET"
)
LOGGER.exception(exc)
rsp.Status = 0x0110
self.dimse.send_msg(rsp, context.context_id)
return
# Validate rsp_status and set rsp.Status accordingly
rsp = self.validate_status(status, rsp)
if rsp.Status in self.statuses:
status = self.statuses[rsp.Status]
else:
# Unknown status
self.dimse.send_msg(rsp, context.context_id)
return
if status[0] in (STATUS_SUCCESS, STATUS_WARNING) and ds:
# If success or warning then there may be a dataset
transfer_syntax = context.transfer_syntax[0]
bytestream = encode(ds,
transfer_syntax.is_implicit_VR,
transfer_syntax.is_little_endian)
bytestream = BytesIO(bytestream)
if bytestream.getvalue() == b'':
LOGGER.error(
"Failed to encode the N-SET response's 'Attribute "
"List' dataset"
)
# Processing failure
rsp.Status = 0x0110
else:
rsp.AttributeList = bytestream
# Send response primitive
self.dimse.send_msg(rsp, context.context_id)
def test_exceptions(self):
""" Check incorrect types/values for properties raise exceptions """
primitive = N_SET()
# MessageID
with pytest.raises(TypeError):
primitive.MessageID = 'halp'
with pytest.raises(TypeError):
primitive.MessageID = 1.111
with pytest.raises(ValueError):
primitive.MessageID = 65536
with pytest.raises(ValueError):
primitive.MessageID = -1
# MessageIDBeingRespondedTo
with pytest.raises(TypeError):
primitive.MessageIDBeingRespondedTo = 'halp'
with pytest.raises(TypeError):
primitive.MessageIDBeingRespondedTo = 1.111
with pytest.raises(ValueError):
primitive.MessageIDBeingRespondedTo = 65536
with pytest.raises(ValueError):
primitive.MessageIDBeingRespondedTo = -1
# AffectedSOPClassUID
with pytest.raises(TypeError):
primitive.AffectedSOPClassUID = 45.2
with pytest.raises(TypeError):
primitive.AffectedSOPClassUID = 100
with pytest.raises(ValueError):
primitive.AffectedSOPClassUID = 'abc'
# AffectedSOPInstanceUID
with pytest.raises(TypeError):
primitive.AffectedSOPInstanceUID = 45.2
with pytest.raises(TypeError):
primitive.AffectedSOPInstanceUID = 100
with pytest.raises(ValueError):
primitive.AffectedSOPInstanceUID = 'abc'
# RequestedSOPClassUID
with pytest.raises(TypeError):
primitive.RequestedSOPClassUID = 45.2
with pytest.raises(TypeError):
primitive.RequestedSOPClassUID = 100
with pytest.raises(ValueError):
primitive.RequestedSOPClassUID = 'abc'
# RequestedSOPInstanceUID
with pytest.raises(TypeError):
primitive.RequestedSOPInstanceUID = 45.2
with pytest.raises(TypeError):
primitive.RequestedSOPInstanceUID = 100
with pytest.raises(ValueError):
primitive.RequestedSOPInstanceUID = 'abc'
# AttributeList
msg = r"'AttributeList' parameter must be a BytesIO object"
with pytest.raises(TypeError, match=msg):
primitive.AttributeList = 'halp'
with pytest.raises(TypeError):
primitive.AttributeList = 1.111
with pytest.raises(TypeError):
primitive.AttributeList = 50
with pytest.raises(TypeError):
primitive.AttributeList = [30, 10]
# ModificationList
msg = r"'ModificationList' parameter must be a BytesIO object"
with pytest.raises(TypeError, match=msg):
primitive.ModificationList = 'halp'
with pytest.raises(TypeError):
primitive.ModificationList = 1.111
with pytest.raises(TypeError):
primitive.ModificationList = 50
with pytest.raises(TypeError):
primitive.ModificationList = [30, 10]
# Status
with pytest.raises(TypeError):
primitive.Status = 19.4
def test_assignment(self):
""" Check assignment works correctly """
primitive = N_SET()
# AffectedSOPClassUID
primitive.AffectedSOPClassUID = '1.1.1'
assert primitive.AffectedSOPClassUID == UID('1.1.1')
assert isinstance(primitive.AffectedSOPClassUID, UID)
primitive.AffectedSOPClassUID = UID('1.1.2')
assert primitive.AffectedSOPClassUID == UID('1.1.2')
assert isinstance(primitive.AffectedSOPClassUID, UID)
primitive.AffectedSOPClassUID = b'1.1.3'
assert primitive.AffectedSOPClassUID == UID('1.1.3')
assert isinstance(primitive.AffectedSOPClassUID, UID)
# AffectedSOPInstanceUID
primitive.AffectedSOPInstanceUID = b'1.2.1'
assert primitive.AffectedSOPInstanceUID == UID('1.2.1')
assert isinstance(primitive.AffectedSOPClassUID, UID)
primitive.AffectedSOPInstanceUID = UID('1.2.2')
assert primitive.AffectedSOPInstanceUID == UID('1.2.2')
assert isinstance(primitive.AffectedSOPClassUID, UID)
primitive.AffectedSOPInstanceUID = '1.2.3'
assert primitive.AffectedSOPInstanceUID == UID('1.2.3')
assert isinstance(primitive.AffectedSOPClassUID, UID)
# AttributeList
ref_ds = Dataset()
ref_ds.PatientID = '1234567'
primitive.AttributeList = BytesIO(encode(ref_ds, True, True))
ds = decode(primitive.AttributeList, True, True)
assert ds.PatientID == '1234567'
#ModificationList
ref_ds = Dataset()
ref_ds.PatientID = '123456'
primitive.ModificationList = BytesIO(encode(ref_ds, True, True))
ds = decode(primitive.ModificationList, True, True)
assert ds.PatientID == '123456'
# MessageID
primitive.MessageID = 11
assert 11 == primitive.MessageID
# MessageIDBeingRespondedTo
primitive.MessageIDBeingRespondedTo = 13
assert 13 == primitive.MessageIDBeingRespondedTo
# RequestedSOPClassUID
primitive.RequestedSOPClassUID = '1.1.1'
assert primitive.RequestedSOPClassUID == UID('1.1.1')
assert isinstance(primitive.RequestedSOPClassUID, UID)
primitive.RequestedSOPClassUID = UID('1.1.2')
assert primitive.RequestedSOPClassUID == UID('1.1.2')
assert isinstance(primitive.RequestedSOPClassUID, UID)
primitive.RequestedSOPClassUID = b'1.1.3'
assert primitive.RequestedSOPClassUID == UID('1.1.3')
assert isinstance(primitive.RequestedSOPClassUID, UID)
# RequestedSOPInstanceUID
primitive.RequestedSOPInstanceUID = b'1.2.1'
assert primitive.RequestedSOPInstanceUID == UID('1.2.1')
assert isinstance(primitive.RequestedSOPInstanceUID, UID)
primitive.RequestedSOPInstanceUID = UID('1.2.2')
assert primitive.RequestedSOPInstanceUID == UID('1.2.2')
assert isinstance(primitive.RequestedSOPInstanceUID, UID)
primitive.RequestedSOPInstanceUID = '1.2.3'
assert primitive.RequestedSOPInstanceUID == UID('1.2.3')
assert isinstance(primitive.RequestedSOPInstanceUID, UID)
# Status
primitive.Status = 0x0000
assert primitive.Status == 0x0000
@staticmethod
def peek_next_pdu():
return 0x01
REFERENCE_MSG = [
(C_ECHO(), ("C_ECHO_RQ", "C_ECHO_RSP")),
(C_STORE(), ("C_STORE_RQ", "C_STORE_RSP")),
(C_FIND(), ("C_FIND_RQ", "C_FIND_RSP")),
(C_GET(), ("C_GET_RQ", "C_GET_RSP")),
(C_MOVE(), ("C_MOVE_RQ", "C_MOVE_RSP")),
(C_CANCEL(), (None, "C_CANCEL_RQ")),
(N_EVENT_REPORT(), ("N_EVENT_REPORT_RQ", "N_EVENT_REPORT_RSP")),
(N_GET(), ("N_GET_RQ", "N_GET_RSP")),
(N_SET(), ("N_SET_RQ", "N_SET_RSP")),
(N_ACTION(), ("N_ACTION_RQ", "N_ACTION_RSP")),
(N_CREATE(), ("N_CREATE_RQ", "N_CREATE_RSP")),
(N_DELETE(), ("N_DELETE_RQ", "N_DELETE_RSP")),
]
class TestDIMSEProvider:
"""Test DIMSE service provider operations."""
def setup(self):
"""Set up"""
self.dimse = DIMSEServiceProvider(DummyAssociation())
def test_receive_not_pdata(self):
"""Test we get back None if not a P_DATA"""
assert self.dimse.get_msg(True) == (None, None)