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)
def test_conversion_rsp(self): """ Check conversion to a -RSP PDU produces the correct output """ primitive = C_STORE() primitive.MessageIDBeingRespondedTo = 5 primitive.AffectedSOPClassUID = '1.2.4.10' primitive.AffectedSOPInstanceUID = '1.2.4.5.7.8' primitive.Status = 0x0000 dimse_msg = C_STORE_RSP() dimse_msg.primitive_to_message(primitive) pdvs = dimse_msg.encode_msg(1, 16382) cs_pdv = pdvs[0].presentation_data_value_list[0][1] self.assertEqual(cs_pdv, c_store_rsp_cmd)
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_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)