def test_is_valid_request(self):
"""Test C_MOVE.is_valid_request"""
primitive = C_MOVE()
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.MoveDestination = "1234567890123456"
assert not primitive.is_valid_request
primitive.Identifier = BytesIO()
assert primitive.is_valid_request
def test_send_c_move_rsp_no_affected_sop(self):
"""Test the handler for C-MOVE rsp with no Affected SOP Class UID."""
self.ae = ae = AE()
ae.add_supported_context('1.2.840.10008.1.1')
ae.add_requested_context('1.2.840.10008.1.1')
scp = ae.start_server(('', 11112), block=False)
assoc = ae.associate('localhost', 11112)
assert assoc.is_established
msg = C_MOVE()
msg.MessageIDBeingRespondedTo = 1
msg.Status = 0x0000
msg.NumberOfRemainingSuboperations = 0
msg.NumberOfCompletedSuboperations = 0
msg.NumberOfFailedSuboperations = 0
msg.NumberOfWarningSuboperations = 0
assoc.dimse.send_msg(msg, 1)
assoc.release()
scp.shutdown()
def test_aet_short_true(self):
"""Test using short AE titles."""
primitive = C_MOVE()
_config.USE_SHORT_DIMSE_AET = True
primitive.MoveDestination = "A"
aet = primitive.MoveDestination
assert "A" == primitive.MoveDestination
primitive.MoveDestination = " ABCD FG "
assert primitive.MoveDestination == " ABCD FG "
primitive.MoveDestination = " ABCD FG "
assert primitive.MoveDestination == " ABCD FG "
primitive.MoveDestination = "ABCDEFGHIJKLMNO"
assert "ABCDEFGHIJKLMNO" == primitive.MoveDestination
primitive.MoveDestination = "ABCDEFGHIJKLMNOP"
assert "ABCDEFGHIJKLMNOP" == primitive.MoveDestination
msg = "Invalid 'Move Destination' value 'ABCDEFGHIJKLMNOPQ'"
with pytest.raises(ValueError, match=msg):
primitive.MoveDestination = "ABCDEFGHIJKLMNOPQ"
assert "ABCDEFGHIJKLMNOP" == primitive.MoveDestination
def test_aet_short_true(self):
"""Test using short AE titles."""
primitive = C_MOVE()
_config.USE_SHORT_DIMSE_AET = True
primitive.MoveDestination = b'A'
aet = primitive.MoveDestination
assert b'A ' == primitive.MoveDestination
primitive.MoveDestination = b'ABCDEFGHIJKLMNO'
assert b'ABCDEFGHIJKLMNO ' == primitive.MoveDestination
primitive.MoveDestination = b'ABCDEFGHIJKLMNOP'
assert b'ABCDEFGHIJKLMNOP' == primitive.MoveDestination
primitive.MoveDestination = b'ABCDEFGHIJKLMNOPQ'
assert b'ABCDEFGHIJKLMNOP' == primitive.MoveDestination
def test_callback_receive_c_move(self):
"""Check callback for receiving DIMSE C-MOVE messages."""
# C-MOVE-RQ
msg = C_MOVE_RQ()
self.dimse.debug_receive_c_move_rq(msg)
# C-MOVE-RSP
primitive = C_MOVE()
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()
primitive.NumberOfCompletedSuboperations = 1
primitive.NumberOfWarningSuboperations = 3
primitive.NumberOfFailedSuboperations = 4
# No dataset, remaining subops
primitive.Status = 0x0000 # Must be for pending
msg = C_MOVE_RSP()
msg.primitive_to_message(primitive)
self.dimse.debug_receive_c_move_rsp(msg)
# Dataset
ref_ds = Dataset()
ref_ds.PatientID = 'Test1101'
ref_ds.PatientName = "Tube HeNe"
primitive.Identifier = BytesIO(encode(ref_ds, True, True))
primitive.NumberOfRemainingSuboperations = 2
msg = C_GET_RSP()
msg.primitive_to_message(primitive)
# Dataset
self.dimse.debug_receive_c_move_rsp(msg)
# C-CANCEL-MOVE-RQ
self.dimse.debug_receive_c_cancel_rq(msg)
def test_callback_send_c_move(self):
"""Check callback for sending DIMSE C-MOVE messages."""
# C-MOVE-RQ
primitive = C_MOVE()
primitive.MessageID = 1
primitive.AffectedSOPClassUID = '1.1.1'
primitive.MoveDestination = b'TESTSCP'
# No dataset
primitive.Identifier = BytesIO()
self.dimse.send_msg(primitive, 1)
# Dataset
bytestream = BytesIO()
bytestream.write(c_store_ds)
primitive.Identifier = bytestream
self.dimse.send_msg(primitive, 1)
# C-MOVE-RSP
primitive = C_MOVE()
primitive.MessageIDBeingRespondedTo = 1
primitive.Status = 0x0000
# No dataset
primitive.Identifier = BytesIO()
self.dimse.send_msg(primitive, 1)
# Dataset
bytestream = BytesIO()
bytestream.write(c_store_ds)
primitive.Identifier = bytestream
self.dimse.send_msg(primitive, 1)
@staticmethod
def receive_pdu():
"""Dummy Receive method to test DIMSEServiceProvider.Receive"""
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_exceptions(self):
""" Check incorrect types/values for properties raise exceptions """
primitive = C_MOVE()
# 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
# NumberOfRemainingSuboperations
with pytest.raises(TypeError):
primitive.NumberOfRemainingSuboperations = 'halp'
with pytest.raises(TypeError):
primitive.NumberOfRemainingSuboperations = 1.111
with pytest.raises(ValueError):
primitive.NumberOfRemainingSuboperations = -1
# NumberOfCompletedSuboperations
with pytest.raises(TypeError):
primitive.NumberOfCompletedSuboperations = 'halp'
with pytest.raises(TypeError):
primitive.NumberOfCompletedSuboperations = 1.111
with pytest.raises(ValueError):
primitive.NumberOfCompletedSuboperations = -1
# NumberOfFailedSuboperations
with pytest.raises(TypeError):
primitive.NumberOfFailedSuboperations = 'halp'
with pytest.raises(TypeError):
primitive.NumberOfFailedSuboperations = 1.111
with pytest.raises(ValueError):
primitive.NumberOfFailedSuboperations = -1
# NumberOfWarningSuboperations
with pytest.raises(TypeError):
primitive.NumberOfWarningSuboperations = 'halp'
with pytest.raises(TypeError):
primitive.NumberOfWarningSuboperations = 1.111
with pytest.raises(ValueError):
primitive.NumberOfWarningSuboperations = -1
# AffectedSOPClassUID
with pytest.raises(TypeError):
primitive.AffectedSOPClassUID = 45.2
with pytest.raises(TypeError):
primitive.AffectedSOPClassUID = 100
# Priority
with pytest.raises(ValueError):
primitive.Priority = 45.2
with pytest.raises(ValueError):
primitive.Priority = 'abc'
with pytest.raises(ValueError):
primitive.Priority = -1
with pytest.raises(ValueError):
primitive.Priority = 3
# MoveDestination
with pytest.raises(TypeError):
primitive.MoveDestination = 45.2
with pytest.raises(TypeError):
primitive.MoveDestination = 100
with pytest.raises(ValueError):
primitive.MoveDestination = ''
with pytest.raises(ValueError):
primitive.MoveDestination = ' '
# Identifier
msg = r"'Identifier' parameter must be a BytesIO object"
with pytest.raises(TypeError, match=msg):
primitive.Identifier = 'halp'
with pytest.raises(TypeError):
primitive.Identifier = 1.111
with pytest.raises(TypeError):
primitive.Identifier = 50
with pytest.raises(TypeError):
primitive.Identifier = [30, 10]
# Status
with pytest.raises(TypeError):
primitive.Status = 19.4
def test_assignment(self):
""" Check assignment works correctly """
primitive = C_MOVE()
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
primitive.MoveDestination = 'UNITTEST_SCP'
assert primitive.MoveDestination == b'UNITTEST_SCP '
ref_ds = Dataset()
ref_ds.PatientID = 1234567
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
@staticmethod
def receive_pdu():
"""Dummy Receive method to test DIMSEServiceProvider.Receive"""
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:
"""Test DIMSE service provider operations."""
def setup(self):
"""Set up"""
self.dimse = DIMSEServiceProvider(DummyAssociation())
