def test_callback_receive_n_set(self):
"""Check callback for receiving DIMSE N-SET messages."""
# N-SET-RQ
primitive = N_SET()
primitive.MessageID = 1
primitive.RequestedSOPClassUID = '1.1.1'
primitive.RequestedSOPInstanceUID = '1.1.1.1'
primitive.ModificationList = BytesIO(n_set_rq_ds)
msg = N_SET_RQ()
msg.primitive_to_message(primitive)
msg.context_id = 1
self.dimse.debug_receive_n_set_rq(msg)
# N-SET-RSP
primitive = N_SET()
primitive.MessageIDBeingRespondedTo = 1
primitive.Status = 0x0000
msg = N_SET_RSP()
msg.primitive_to_message(primitive)
msg.context_id = 1
self.dimse.debug_receive_n_set_rsp(msg)
# User defined
primitive.AffectedSOPClassUID = '1.1.1'
primitive.AffectedSOPInstanceUID = '1.1.1.1'
primitive.ModificationList = BytesIO(n_set_rsp_ds)
msg = N_GET_RSP()
msg.primitive_to_message(primitive)
msg.context_id = 1
self.dimse.debug_receive_n_set_rsp(msg)
def test_conversion_rq(self):
""" Check conversion to a -RQ PDU produces the correct output """
primitive = N_SET()
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'
ds = Dataset()
ds.PatientID = 'Test1101'
ds.PatientName = "Tube HeNe"
primitive.ModificationList = BytesIO(encode(ds, True, True))
dimse_msg = N_SET_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_set_rq_cmd
assert ds_pdv == n_set_rq_ds
def test_is_valid_request(self):
"""Test N_SET.is_valid_request"""
primitive = N_SET()
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 not primitive.is_valid_request
ds = Dataset()
ds.PatientID = 'Test1101'
ds.PatientName = "Tube HeNe"
primitive.ModificationList = BytesIO(encode(ds, True, True))
assert primitive.is_valid_request
def test_callback_send_n_set(self):
"""Check callback for sending DIMSE N-SET messages."""
# N-SET-RQ
primitive = N_SET()
primitive.MessageID = 1
primitive.RequestedSOPClassUID = '1.1.1'
primitive.RequestedSOPInstanceUID = '1.1.1.1'
primitive.ModificationList = BytesIO(b'\x00\x01')
self.dimse.send_msg(primitive, 1)
# N-SET-RSP
primitive = N_SET()
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'
primitive.AttributeList = BytesIO(b'\x00\x01')
self.dimse.send_msg(primitive, 1)
def test_exceptions(self):
""" Check incorrect types/values for properties raise exceptions """
primitive = N_SET()
# 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'
# 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]
# ModificationList
msg = r"'ModificationList' parameter must be a BytesIO object"
with pytest.raises(TypeError, match=msg):
primitive.ModificationList = 'halp'
with pytest.raises(TypeError):
primitive.ModificationList = 1.111
with pytest.raises(TypeError):
primitive.ModificationList = 50
with pytest.raises(TypeError):
primitive.ModificationList = [30, 10]
# Status
with pytest.raises(TypeError):
primitive.Status = 19.4
def test_assignment(self):
""" Check assignment works correctly """
primitive = N_SET()
# 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'
#ModificationList
ref_ds = Dataset()
ref_ds.PatientID = '123456'
primitive.ModificationList = BytesIO(encode(ref_ds, True, True))
ds = decode(primitive.ModificationList, True, True)
assert ds.PatientID == '123456'
# 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