def test_scp_failed_ds_decode(self): """Test failure to decode the dataset""" # Hard to test directly as decode errors won't show up until the # dataset is actually used self.scp = DummyStorageSCP() self.scp.status = 0x0000 self.scp.start() ae = AE() ae.add_requested_context(HangingProtocolStorage, ExplicitVRLittleEndian) assoc = ae.associate('localhost', 11112) assert assoc.is_established req = C_STORE() req.MessageID = 1 req.AffectedSOPClassUID = DATASET.SOPClassUID req.AffectedSOPInstanceUID = DATASET.SOPInstanceUID req.Priorty = 0x0002 req.DataSet = BytesIO(b'\x08\x00\x01\x00\x04\x00\x00\x00\x00\x08\x00\x49') # Send C-STORE request to DIMSE and get response assoc.dimse.send_msg(req, 1) rsp, _ = assoc.dimse.receive_msg(True) assert rsp.Status == 0xC100 assert rsp.ErrorComment == 'Unable to decode the dataset' assoc.release() self.scp.stop()
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_send_c_store(self): """Check sending DIMSE C-STORE messages.""" # C-STORE-RQ primitive = C_STORE() primitive.MessageID = 1 primitive.AffectedSOPClassUID = '1.1.1' def test_callback(msg): """Callback""" self.assertEqual(msg.__class__.__name__, 'C_STORE_RQ') self.dimse.on_send_dimse_message = test_callback self.dimse.send_msg(primitive, 1) # C-STORE-RSP primitive = C_STORE() primitive.MessageIDBeingRespondedTo = 1 primitive.Status = 0x0000 def test_callback(msg): """Callback""" self.assertEqual(msg.__class__.__name__, 'C_STORE_RSP') self.dimse.on_send_dimse_message = test_callback self.dimse.send_msg(primitive, 1)
def test_callback_send_c_store(self): """Check callback for sending DIMSE C-STORE messages.""" # C-STORE-RQ primitive = C_STORE() primitive.MessageID = 1 primitive.AffectedSOPClassUID = '1.2.840.10008.5.1.4.1.1.2' # CT primitive.AffectedSOPInstanceUID = '1.1.2' primitive.Priority = 0x02 primitive.DataSet = BytesIO() # CT + no dataset self.dimse.send_msg(primitive, 1) primitive.AffectedSOPClassUID = '1.1.1' # No UID type, no dataset # MR + dataset primitive.AffectedSOPClassUID = '1.2.840.10008.5.1.4.1.1.4' # MR bytestream = BytesIO() bytestream.write(c_store_ds) primitive.DataSet = bytestream self.dimse.send_msg(primitive, 1) # C-STORE-RSP primitive = C_STORE() primitive.MessageIDBeingRespondedTo = 1 primitive.Status = 0x0000 self.dimse.send_msg(primitive, 1)
def test_assignment(self): """ Check assignment works correctly """ primitive = C_STORE() primitive.MessageID = 11 self.assertEqual(primitive.MessageID, 11) primitive.MessageIDBeingRespondedTo = 13 self.assertEqual(primitive.MessageIDBeingRespondedTo, 13) # AffectedSOPClassUID primitive.AffectedSOPClassUID = '1.1.1' self.assertEqual(primitive.AffectedSOPClassUID, UID('1.1.1')) self.assertTrue(isinstance(primitive.AffectedSOPClassUID, UID)) primitive.AffectedSOPClassUID = UID('1.1.2') self.assertEqual(primitive.AffectedSOPClassUID, UID('1.1.2')) self.assertTrue(isinstance(primitive.AffectedSOPClassUID, UID)) primitive.AffectedSOPClassUID = b'1.1.3' self.assertEqual(primitive.AffectedSOPClassUID, UID('1.1.3')) self.assertTrue(isinstance(primitive.AffectedSOPClassUID, UID)) # AffectedSOPInstanceUID primitive.AffectedSOPInstanceUID = b'1.2.1' self.assertEqual(primitive.AffectedSOPInstanceUID, UID('1.2.1')) self.assertTrue(isinstance(primitive.AffectedSOPClassUID, UID)) primitive.AffectedSOPInstanceUID = UID('1.2.2') self.assertEqual(primitive.AffectedSOPInstanceUID, UID('1.2.2')) self.assertTrue(isinstance(primitive.AffectedSOPClassUID, UID)) primitive.AffectedSOPInstanceUID = '1.2.3' self.assertEqual(primitive.AffectedSOPInstanceUID, UID('1.2.3')) self.assertTrue(isinstance(primitive.AffectedSOPClassUID, UID)) primitive.Priority = 0x02 self.assertEqual(primitive.Priority, 0x02) primitive.MoveOriginatorApplicationEntityTitle = 'UNITTEST_SCP' self.assertEqual(primitive.MoveOriginatorApplicationEntityTitle, b'UNITTEST_SCP ') primitive.MoveOriginatorApplicationEntityTitle = b'UNITTEST_SCP' self.assertEqual(primitive.MoveOriginatorApplicationEntityTitle, b'UNITTEST_SCP ') primitive.MoveOriginatorMessageID = 15 self.assertEqual(primitive.MoveOriginatorMessageID, 15) ref_ds = Dataset() ref_ds.PatientID = 1234567 primitive.DataSet = BytesIO(encode(ref_ds, True, True)) #self.assertEqual(primitive.DataSet, ref_ds) primitive.Status = 0x0000 self.assertEqual(primitive.Status, 0x0000) primitive.Status = 0xC123 self.assertEqual(primitive.Status, 0xC123) primitive.Status = 0xEE01 self.assertEqual(primitive.Status, 0xEE01)
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_assignment(self): """ Check assignment works correctly """ primitive = C_STORE() primitive.MessageID = 11 assert primitive.MessageID == 11 primitive.MessageIDBeingRespondedTo = 13 assert primitive.MessageIDBeingRespondedTo == 13 # 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) primitive.Priority = 0x02 assert primitive.Priority == 0x02 primitive.MoveOriginatorApplicationEntityTitle = 'UNITTEST_SCP' assert primitive.MoveOriginatorApplicationEntityTitle == b'UNITTEST_SCP ' primitive.MoveOriginatorApplicationEntityTitle = b'UNITTEST_SCP' assert primitive.MoveOriginatorApplicationEntityTitle == b'UNITTEST_SCP ' primitive.MoveOriginatorMessageID = 15 assert primitive.MoveOriginatorMessageID == 15 ref_ds = Dataset() ref_ds.PatientID = 1234567 primitive.DataSet = BytesIO(encode(ref_ds, True, True)) #assert primitive.DataSet, ref_ds) primitive.Status = 0x0000 assert primitive.Status == 0x0000 primitive.Status = 0xC123 assert primitive.Status == 0xC123 primitive.Status = 0xEE01 assert primitive.Status == 0xEE01
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_is_valid_request(self): """Test C_STORE.is_valid_request""" primitive = C_STORE() assert not primitive.is_valid_request primitive.MessageID = 1 assert not primitive.is_valid_request primitive.AffectedSOPClassUID = '1.2' assert not primitive.is_valid_request primitive.Priority = 2 assert not primitive.is_valid_request primitive.AffectedSOPInstanceUID = '1.2.1' assert not primitive.is_valid_request primitive.DataSet = BytesIO() assert primitive.is_valid_request
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 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_exceptions(self): """ Check incorrect types/values for properties raise exceptions """ primitive = C_STORE() # MessageID with self.assertRaises(TypeError): primitive.MessageID = 'halp' with self.assertRaises(TypeError): primitive.MessageID = 1.111 with self.assertRaises(ValueError): primitive.MessageID = 65536 with self.assertRaises(ValueError): primitive.MessageID = -1 # MessageIDBeingRespondedTo with self.assertRaises(TypeError): primitive.MessageIDBeingRespondedTo = 'halp' with self.assertRaises(TypeError): primitive.MessageIDBeingRespondedTo = 1.111 with self.assertRaises(ValueError): primitive.MessageIDBeingRespondedTo = 65536 with self.assertRaises(ValueError): primitive.MessageIDBeingRespondedTo = -1 # AffectedSOPClassUID with self.assertRaises(TypeError): primitive.AffectedSOPClassUID = 45.2 with self.assertRaises(TypeError): primitive.AffectedSOPClassUID = 100 with self.assertRaises(ValueError): primitive.AffectedSOPClassUID = 'abc' # AffectedSOPInstanceUID with self.assertRaises(TypeError): primitive.AffectedSOPInstanceUID = 45.2 with self.assertRaises(TypeError): primitive.AffectedSOPInstanceUID = 100 with self.assertRaises(ValueError): primitive.AffectedSOPInstanceUID = 'abc' # Priority with self.assertRaises(ValueError): primitive.Priority = 45.2 with self.assertRaises(ValueError): primitive.Priority = 'abc' with self.assertRaises(ValueError): primitive.Priority = -1 with self.assertRaises(ValueError): primitive.Priority = 3 # MoveOriginatorApplicationEntityTitle with self.assertRaises(TypeError): primitive.MoveOriginatorApplicationEntityTitle = 45.2 with self.assertRaises(TypeError): primitive.MoveOriginatorApplicationEntityTitle = 100 with self.assertRaises(ValueError): primitive.MoveOriginatorApplicationEntityTitle = '' with self.assertRaises(ValueError): primitive.MoveOriginatorApplicationEntityTitle = ' ' # MoveOriginatorMessageID with self.assertRaises(TypeError): primitive.MoveOriginatorMessageID = 'halp' with self.assertRaises(TypeError): primitive.MoveOriginatorMessageID = 1.111 with self.assertRaises(ValueError): primitive.MoveOriginatorMessageID = 65536 with self.assertRaises(ValueError): primitive.MoveOriginatorMessageID = -1 # DataSet with self.assertRaises(TypeError): primitive.DataSet = 'halp' with self.assertRaises(TypeError): primitive.DataSet = 1.111 with self.assertRaises(TypeError): primitive.DataSet = 50 with self.assertRaises(TypeError): primitive.DataSet = [30, 10] # Status with self.assertRaises(TypeError): primitive.Status = 19.4