def test_is_cancelled_no_match(self): """Test is_cancelled with no matching C-CANCEL.""" assoc = DummyAssoc() cancel = C_CANCEL() cancel.MessageIDBeingRespondedTo = 5 assoc.dimse.cancel_req[5] = cancel assoc.dimse.cancel_req[3] = cancel service = ServiceClass(assoc) assert service.is_cancelled(1) is False assert service.is_cancelled(2) is False
def test_assignment(self): """ Check assignment works correctly """ primitive = C_CANCEL() primitive.MessageIDBeingRespondedTo = 13 assert primitive.MessageIDBeingRespondedTo == 13 with pytest.raises(ValueError): primitive.MessageIDBeingRespondedTo = 100000 with pytest.raises(TypeError): primitive.MessageIDBeingRespondedTo = 'test'
def test_is_cancelled_match(self): """Test is_cancelled with matching C-CANCEL.""" assoc = DummyAssoc() cancel = C_CANCEL() cancel.MessageIDBeingRespondedTo = 5 assoc.dimse.cancel_req[4] = C_GET() assoc.dimse.cancel_req[3] = cancel service = ServiceClass(assoc) assert service.is_cancelled(1) is False assert service.is_cancelled(2) is False assert service.is_cancelled(3) is True service = ServiceClass(assoc) assert service.is_cancelled(1) is False assert service.is_cancelled(2) is False assert service.is_cancelled(3) is False assert cancel not in assoc.dimse.cancel_req.values()
def message_to_primitive(self): """Convert the ``DIMSEMessage`` class to a DIMSE primitive. Returns ------- DIMSEPrimitive sub-class One of the DIMSE message primitives from :ref:`pynetdicom.dimse_primitives<api_dimse_primitives>` generated from the current ``DIMSEMessage`` sub-class object. """ # 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.context_id return primitive
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 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())