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_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_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)
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