def test_send_n_get(self): """Check sending DIMSE N-GET messages.""" # N-GET-RQ primitive = N_GET() primitive.MessageID = 1 primitive.AffectedSOPClassUID = '1.1.1' def test_callback(msg): """Callback""" self.assertEqual(msg.__class__.__name__, 'N_GET_RQ') self.dimse.on_send_dimse_message = test_callback self.dimse.send_msg(primitive, 1) # N-GET-RSP primitive = N_GET() primitive.MessageIDBeingRespondedTo = 1 primitive.Status = 0x0000 def test_callback(msg): """Callback""" self.assertEqual(msg.__class__.__name__, 'N_GET_RSP') self.dimse.on_send_dimse_message = test_callback self.dimse.send_msg(primitive, 1)
def test_callback_send_n_get(self): """Check callback for sending DIMSE N-GET messages.""" # N-GET-RQ primitive = N_GET() primitive.MessageID = 1 primitive.RequestedSOPClassUID = '1.1.1' primitive.RequestedSOPInstanceUID = '1.1.1.1' self.dimse.send_msg(primitive, 1) # Plus user defined primitive.AttributeIdentifierList = [(0x0000, 0x0000), (0xffff, 0xffff)] self.dimse.send_msg(primitive, 1) # N-GET-RSP # Mandatory elements primitive = N_GET() 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' self.dimse.send_msg(primitive, 1) # Conditional primitive.AttributeList = BytesIO(n_get_rsp_ds) self.dimse.send_msg(primitive, 1)
def test_callback_send_n_get(self): """Check callback for sending DIMSE N-GET messages.""" # N-GET-RQ primitive = N_GET() primitive.MessageID = 1 primitive.AffectedSOPClassUID = '1.1.1' self.dimse.send_msg(primitive, 1) # N-GET-RSP primitive = N_GET() primitive.MessageIDBeingRespondedTo = 1 primitive.Status = 0x0000 self.dimse.send_msg(primitive, 1)
def test_conversion_rsp(self): """ Check conversion to a -RSP PDU produces the correct output """ primitive = N_GET() 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_GET_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_get_rsp_cmd assert ds_pdv == n_get_rsp_ds
def test_is_valid_resposne(self): """Test N_GET.is_valid_response.""" primitive = N_GET() 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 ------- DIMSE message primitive One of the DIMSE message primitives from pynetdicom3.dimse_primitives generated from the current DIMSEMessage. """ # 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.ID return primitive
def test_callback_receive_n_get(self): """Check callback for receiving DIMSE N-GET messages.""" # N-GET-RQ primitive = N_GET() primitive.MessageID = 1 primitive.RequestedSOPClassUID = '1.1.1' primitive.RequestedSOPInstanceUID = '1.1.1.1' msg = N_GET_RQ() msg.primitive_to_message(primitive) msg.ID = 1 self.dimse.debug_receive_n_get_rq(msg) # Plus user defined primitive.AttributeIdentifierList = [(0x0000, 0x0000), (0xffff, 0xffff)] msg = N_GET_RQ() msg.primitive_to_message(primitive) msg.ID = 1 self.dimse.debug_receive_n_get_rq(msg) # N-GET-RSP # Mandatory elements primitive = N_GET() primitive.MessageIDBeingRespondedTo = 1 primitive.Status = 0x0000 msg = N_GET_RSP() msg.primitive_to_message(primitive) msg.ID = 1 self.dimse.debug_receive_n_get_rsp(msg) # User defined primitive.AffectedSOPClassUID = '1.1.1' primitive.AffectedSOPInstanceUID = '1.1.1.1' msg = N_GET_RSP() msg.primitive_to_message(primitive) msg.ID = 1 self.dimse.debug_receive_n_get_rsp(msg) # Conditional primitive.AttributeList = BytesIO(n_get_rsp_ds) msg = N_GET_RSP() msg.primitive_to_message(primitive) msg.ID = 1 self.dimse.debug_receive_n_get_rsp(msg)
def test_is_valid_request(self): """Test N_GET.is_valid_request""" primitive = N_GET() 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 primitive.is_valid_request
def test_conversion_rq(self): """ Check conversion to a -RQ PDU produces the correct output """ primitive = N_GET() 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' primitive.AttributeIdentifierList = [(0x7fe0, 0x0010), (0x0000, 0x0000), (0xFFFF, 0xFFFF)] dimse_msg = N_GET_RQ() dimse_msg.primitive_to_message(primitive) pdvs = [] for fragment in dimse_msg.encode_msg(1, 16382): pdvs.append(fragment) assert len(pdvs) == 1 cs_pdv = pdvs[0].presentation_data_value_list[0][1] assert cs_pdv == n_get_rq_cmd
def test_conversion_rsp(self): """ Check conversion to a -RSP PDU produces the correct output """ primitive = N_GET()
def test_exceptions(self): """ Check incorrect types/values for properties raise exceptions """ primitive = N_GET()
def test_assignment(self): """ Check assignment works correctly """ primitive = N_GET()
def message_to_primitive(self): """Convert the DIMSEMessage class to a DIMSE primitive. Returns ------- primitive One of the pynetdicom3.dimse_primitives primitives generated from the current DIMSEMessage. """ # 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() # elif 'C_CANCEL' in cls_type_name: # primitive = C_CANCEL() # 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 if cls_type_name == 'C_STORE_RQ': setattr(primitive, 'DataSet', self.data_set) elif cls_type_name in [ 'C_FIND_RQ', 'C_FIND_RSP', 'C_GET_RQ', 'C_GET_RSP', 'C_MOVE_RQ', 'C_MOVE_RSP' ]: setattr(primitive, 'Identifier', self.data_set) elif cls_type_name == 'N_EVENT_REPORT_RQ': setattr(primitive, 'EventInformation', self.data_set) elif cls_type_name == 'N_EVENT_REPORT_RSP': setattr(primitive, 'EventReply', self.data_set) elif cls_type_name in [ 'N_GET_RSP', 'N_SET_RSP', 'N_CREATE_RQ', 'N_CREATE_RSP' ]: setattr(primitive, 'AttributeList', self.data_set) elif cls_type_name == 'N_SET_RQ': setattr(primitive, 'ModificationList', self.data_set) elif cls_type_name == 'N_ACTION_RQ': setattr(primitive, 'ActionInformation', self.data_set) elif cls_type_name == 'N_ACTION_RSP': setattr(primitive, 'ActionReply', self.data_set) return primitive
def test_exceptions(self): """ Check incorrect types/values for properties raise exceptions """ primitive = N_GET() # 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' # AttributeIdentifierList with pytest.raises(ValueError): primitive.AttributeIdentifierList = Tag(0x0000, 0x0000) with pytest.raises(ValueError, match="Attribute Identifier List must"): primitive.AttributeIdentifierList = ['ijk', 'abc'] with pytest.raises(ValueError, match="Attribute Identifier List must"): primitive.AttributeIdentifierList = [] # 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] # Status with pytest.raises(TypeError): primitive.Status = 19.4
def test_assignment(self): """Check assignment works correctly""" primitive = N_GET() # 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' # AttributeIdentifierList primitive.AttributeIdentifierList = [ 0x00001000, (0x0000, 0x1000), Tag(0x7fe0, 0x0010) ] assert [Tag(0x0000, 0x1000), Tag(0x0000, 0x1000), Tag(0x7fe0, 0x0010)] == primitive.AttributeIdentifierList primitive.AttributeIdentifierList = [(0x7fe0, 0x0010)] assert [Tag(0x7fe0, 0x0010)] == primitive.AttributeIdentifierList primitive.AttributeIdentifierList = (0x7fe0, 0x0010) assert [Tag(0x7fe0, 0x0010)] == primitive.AttributeIdentifierList elem = DataElement((0x0000, 0x0005), 'AT', [Tag(0x0000, 0x1000)]) assert isinstance(elem.value, MutableSequence) primitive.AttributeIdentifierList = elem.value assert [Tag(0x0000, 0x1000)] == primitive.AttributeIdentifierList # 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
pass @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"""