def test_conversion_rq(self):
""" Check conversion to a -RQ PDU produces the correct output """
primitive = N_CREATE()
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'
ds = Dataset()
ds.PatientID = 'Test1101'
ds.PatientName = "Tube HeNe"
primitive.AttributeList = BytesIO(encode(ds, True, True))
dimse_msg = N_CREATE_RQ()
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_create_rq_cmd
assert ds_pdv == n_create_rq_ds
def test_empty_uid_n_create(self):
"""Test converting N_CREATE_RQ with empty UID value."""
msg = N_CREATE_RQ()
for data in [n_create_rq_cmd_empty]:
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_CREATE)
assert primitive.AffectedSOPClassUID == UID("1.2.3.4")
assert primitive.MessageID == 7
assert primitive.AffectedSOPInstanceUID is None
def test_attribute_list(self):
"""Test Event.attribute_list."""
request = N_CREATE_RQ()
request.AttributeList = self.bytestream
event = Event(None, evt.EVT_N_CREATE, {
'request': request,
'context': self.context.as_tuple
})
assert event._hash is None
assert event._decoded is None
ds = event.attribute_list
assert event._hash == hash(request.AttributeList)
assert isinstance(ds, Dataset)
assert ds.PatientName == 'TEST^Test'
ds.PatientID = '1234567'
assert event.attribute_list.PatientID == '1234567'
# Test hash mismatch
event._hash = None
assert 'PatientID' not in event.attribute_list
def test_message_to_primitive_n_create(self):
"""Test converting N_CREATE_RQ and _RSP to primitive."""
# N-CREATE-RQ
msg = N_CREATE_RQ()
for data in [n_create_rq_cmd, n_create_rq_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, N_CREATE)
assert primitive.AffectedSOPClassUID == UID(
"1.2.840.10008.5.1.4.1.1.2")
assert primitive.AffectedSOPInstanceUID == UID(
"1.2.392.200036.9116.2.6.1.48")
assert primitive.MessageID == 7
ds = decode(primitive.AttributeList, True, True)
assert ds.PatientName == "Tube HeNe"
assert ds.PatientID == "Test1101"
# N-ACTION-RSP
msg = N_CREATE_RSP()
for data in [n_create_rsp_cmd, n_create_rsp_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, N_CREATE)
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"
def test_callback_receive_n_create(self):
"""Check callback for receiving DIMSE N-CREATE messages."""
# N-CREATE-RQ
primitive = N_CREATE()
primitive.MessageID = 1
primitive.AffectedSOPClassUID = '1.1.1'
msg = N_CREATE_RQ()
msg.primitive_to_message(primitive)
msg.context_id = 1
self.dimse.debug_receive_n_create_rq(msg)
# User defined
primitive.AffectedSOPInstanceUID = '1.1.1.1'
primitive.AttributeList = BytesIO(n_create_rq_ds)
msg = N_CREATE_RQ()
msg.primitive_to_message(primitive)
msg.context_id = 1
self.dimse.debug_receive_n_create_rq(msg)
# N-CREATE-RSP
primitive = N_CREATE()
primitive.MessageIDBeingRespondedTo = 1
primitive.Status = 0x0000
msg = N_CREATE_RSP()
msg.primitive_to_message(primitive)
msg.context_id = 1
self.dimse.debug_receive_n_create_rsp(msg)
# User defined
primitive.AffectedSOPClassUID = '1.1.1'
primitive.AffectedSOPInstanceUID = '1.1.1.1'
primitive.AttributeList = BytesIO(n_create_rsp_ds)
msg = N_CREATE_RSP()
msg.primitive_to_message(primitive)
msg.context_id = 1
self.dimse.debug_receive_n_create_rsp(msg)
