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_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.context_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.context_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.context_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.context_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.context_id = 1 self.dimse.debug_receive_n_get_rsp(msg)
def test_message_to_primitive_n_get(self): """Test converting N_GET_RQ and _RSP to primitive.""" # N-GET-RQ msg = N_GET_RQ() for data in [n_get_rq_cmd]: 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, N_GET) assert primitive.RequestedSOPClassUID == UID( "1.2.840.10008.5.1.4.1.1.2") assert primitive.RequestedSOPInstanceUID == UID( "1.2.392.200036.9116.2.6.1.48") assert primitive.MessageID == 7 primitive.AttributeIdentifierList = [ (0x7FE0, 0x0010), (0x0000, 0x0000), (0xFFFF, 0xFFFF), ] msg = N_GET_RSP() for data in [n_get_rsp_cmd, n_get_rsp_ds]: p_data = P_DATA() p_data.presentation_data_value_list.append([0, data]) msg.decode_msg(p_data) msg.decode_msg(p_data) primitive = msg.message_to_primitive() assert isinstance(primitive, N_GET) assert primitive.AttributeIdentifierList is None assert primitive.AffectedSOPClassUID == UID("1.2.4.10") assert primitive.AffectedSOPInstanceUID == UID("1.2.4.5.7.8") assert primitive.MessageIDBeingRespondedTo == 5 assert primitive.Status == 0x0000 ds = decode(primitive.AttributeList, True, True) assert ds.PatientName == "Tube HeNe" assert ds.PatientID == "Test1101"