def test_callback_receive_c_find(self): """Check callback for receiving DIMSE C-FIND messages.""" # C-FIND-RQ primitive = C_FIND() 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.Identifier = BytesIO() # No dataset msg = C_FIND_RQ() msg.primitive_to_message(primitive) self.dimse.debug_receive_c_find_rq(msg) # Dataset ref_ds = Dataset() ref_ds.PatientID = 'Test1101' ref_ds.PatientName = "Tube HeNe" primitive.Identifier = BytesIO(encode(ref_ds, True, True)) msg = C_FIND_RQ() msg.primitive_to_message(primitive) # Dataset self.dimse.debug_receive_c_find_rq(msg) # C-FIND-RSP primitive = C_FIND() 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.Identifier = BytesIO() # No dataset primitive.Status = 0x0000 # Must be for pending msg = C_FIND_RSP() msg.primitive_to_message(primitive) self.dimse.debug_receive_c_find_rsp(msg) primitive.Identifier = BytesIO(encode(ref_ds, True, True)) msg = C_FIND_RSP() msg.primitive_to_message(primitive) # Dataset msg.data_set = BytesIO(encode(ref_ds, True, True)) self.dimse.debug_receive_c_find_rsp(msg) # Non-pending status msg.data_set.Status = 0x0001 self.dimse.debug_receive_c_find_rsp(msg) # C-CANCEL-FIND-RQ self.dimse.debug_receive_c_cancel_rq(msg)
def test_callback_send_c_find(self): """Check callback for sending DIMSE C-FIND messages.""" # C-FIND-RQ primitive = C_FIND() primitive.MessageID = 1 primitive.AffectedSOPClassUID = '1.1.1' primitive.Priority = 0x02 primitive.Identifier = BytesIO() # No dataset self.dimse.send_msg(primitive, 1) # Dataset bytestream = BytesIO() bytestream.write(c_store_ds) primitive.Identifier = bytestream self.dimse.send_msg(primitive, 1) # C-FIND-RSP primitive = C_FIND() primitive.MessageIDBeingRespondedTo = 1 primitive.Status = 0x0000 primitive.AffectedSOPClassUID = '1.1.1' # No dataset self.dimse.send_msg(primitive, 1) # Dataset bytestream = BytesIO() bytestream.write(c_store_ds) primitive.Identifier = bytestream primitive.Status = 0xFF00 # Only has dataset when 0xFF00 or 0xFF01 self.dimse.send_msg(primitive, 1)
def test_bad_req_identifier(self): """Test SCP handles a bad request identifier""" self.scp = DummyFindSCP() self.scp.statuses = [0xFF00] self.scp.identifiers = [self.query] self.scp.start() ae = AE() ae.add_requested_context( ProductCharacteristicsQueryInformationModelFind, ExplicitVRLittleEndian) ae.acse_timeout = 5 ae.dimse_timeout = 5 assoc = ae.associate('localhost', 11112) assert assoc.is_established req = C_FIND() req.MessageID = 1 req.AffectedSOPClassUID = ProductCharacteristicsQueryInformationModelFind.uid req.Priority = 2 req.Identifier = BytesIO( b'\x08\x00\x01\x00\x04\x00\x00\x00\x00\x08\x00\x49') assoc.dimse.send_msg(req, 1) rsp, _ = assoc.dimse.receive_msg(True) assert rsp.Status == 0xC310 assoc.release() self.scp.stop()
def test_is_valid_request(self): """Test C_FIND.is_valid_request""" primitive = C_FIND() 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.Identifier = BytesIO() assert primitive.is_valid_request
def test_conversion_rq(self): """ Check conversion to a -RQ PDU produces the correct output """ primitive = C_FIND() primitive.MessageID = 7 primitive.AffectedSOPClassUID = '1.2.840.10008.5.1.4.1.1.2' primitive.Priority = 0x02 ref_identifier = Dataset() ref_identifier.PatientID = '*' ref_identifier.QueryRetrieveLevel = "PATIENT" primitive.Identifier = BytesIO(encode(ref_identifier, True, True)) dimse_msg = C_FIND_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_find_rq_cmd) self.assertEqual(ds_pdv, c_find_rq_ds)
def test_conversion_rsp(self): """ Check conversion to a -RSP PDU produces the correct output """ primitive = C_FIND() primitive.MessageIDBeingRespondedTo = 5 primitive.AffectedSOPClassUID = '1.2.840.10008.5.1.4.1.1.2' primitive.Status = 0xFF00 ref_identifier = Dataset() ref_identifier.QueryRetrieveLevel = "PATIENT" ref_identifier.RetrieveAETitle = validate_ae_title("FINDSCP") ref_identifier.PatientName = "ANON^A^B^C^D" primitive.Identifier = BytesIO(encode(ref_identifier, True, True)) dimse_msg = C_FIND_RSP() 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_find_rsp_cmd) self.assertEqual(ds_pdv, c_find_rsp_ds)
def test_assignment(self): """ Check assignment works correctly """ primitive = C_FIND() 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)) primitive.Priority = 0x02 self.assertEqual(primitive.Priority, 0x02) ref_ds = Dataset() ref_ds.PatientID = '*' ref_ds.QueryRetrieveLevel = "PATIENT" primitive.Identifier = 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_assignment(self): """ Check assignment works correctly """ primitive = C_FIND() 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) primitive.Priority = 0x02 assert primitive.Priority == 0x02 ref_ds = Dataset() ref_ds.PatientID = '*' ref_ds.QueryRetrieveLevel = "PATIENT" primitive.Identifier = 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_exceptions(self): """ Check incorrect types/values for properties raise exceptions """ primitive = C_FIND() # 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' # 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 # Identifier with self.assertRaises(TypeError): primitive.Identifier = 'halp' with self.assertRaises(TypeError): primitive.Identifier = 1.111 with self.assertRaises(TypeError): primitive.Identifier = 50 with self.assertRaises(TypeError): primitive.Identifier = [30, 10] # Status with self.assertRaises(TypeError): primitive.Status = 19.4