Ejemplo n.º 1
0
    def test_conversion_rq(self):
        """ Check conversion to a -RQ PDU produces the correct output """
        primitive = C_MOVE()
        primitive.MessageID = 7
        primitive.AffectedSOPClassUID = '1.2.840.10008.5.1.4.1.1.2'
        primitive.Priority = 0x02
        primitive.MoveDestination = validate_ae_title("MOVE_SCP")

        ref_identifier = Dataset()
        ref_identifier.PatientID = '*'
        ref_identifier.QueryRetrieveLevel = "PATIENT"

        primitive.Identifier = BytesIO(encode(ref_identifier, True, True))

        dimse_msg = C_MOVE_RQ()
        dimse_msg.primitive_to_message(primitive)

        pdvs = []
        for fragment in dimse_msg.encode_msg(1, 16382):
            pdvs.append(fragment)

        cs_pdv = pdvs[0].presentation_data_value_list[0][1]
        ds_pdv = pdvs[1].presentation_data_value_list[0][1]
        assert cs_pdv == c_move_rq_cmd
        assert ds_pdv == c_move_rq_ds
Ejemplo n.º 2
0
    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)
Ejemplo n.º 3
0
    def test_aet_short_false(self):
        """Test using long AE titles."""
        primitive = C_MOVE()

        _config.USE_SHORT_DIMSE_AET = False

        primitive.MoveDestination = "A"
        assert "A" == primitive.MoveDestination
Ejemplo n.º 4
0
    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
Ejemplo n.º 5
0
    def test_aet_deprecation(self):
        """Test deprecation warning for Move Destination from bytes."""
        primitive = C_MOVE()

        msg = ("The use of bytes with 'Move Destination' is deprecated, "
               "use an ASCII str instead")
        with pytest.warns(DeprecationWarning, match=msg):
            primitive.MoveDestination = b'TEST'

        assert primitive.MoveDestination == 'TEST'
Ejemplo n.º 6
0
 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 = b'1234567890123456'
     assert not primitive.is_valid_request
     primitive.Identifier = BytesIO()
     assert primitive.is_valid_request
Ejemplo n.º 7
0
    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
Ejemplo n.º 8
0
    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
Ejemplo n.º 9
0
    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