Python BaseDataHandler._find_new_data_check_attachment方法代码示例

# 需要导入模块: from ion.agents.data.handlers.base_data_handler import BaseDataHandler [as 别名]
# 或者: from ion.agents.data.handlers.base_data_handler.BaseDataHandler import _find_new_data_check_attachment [as 别名]

#.........这里部分代码省略.........
            self._bdh.set()

    def test_set_not_dictionary(self):
        with self.assertRaises(InstrumentParameterException) as cm:
            self._bdh.set("not_found")

    def test_set_polling(self):
        # TODO: Need to change back to enums instead of strings. Problem with BaseEnum.
        self._bdh.set({"POLLING_INTERVAL": 7200})
        self.assertEqual(self._bdh.get(["POLLING_INTERVAL"]), {"POLLING_INTERVAL": 7200})

    def test_set_not_found(self):
        with self.assertRaises(InstrumentParameterException) as cm:
            self._bdh.set({"not_found": 1})

    def test_execute_acquire_data_not_dictionary(self):
        with self.assertRaises(ConfigurationError) as cm:
            self._bdh.execute_acquire_data("not_found")

    @patch("ion.agents.data.handlers.base_data_handler.spawn")
    def test_execute_acquire_data_with_stream_id_not_new(self, mock):
        self._bdh._semaphore = Mock()
        self._bdh._glet_queue = Mock()
        self._bdh.execute_acquire_data({"stream_id": "test_stream_id", "constraints": "test_constraints"})
        self._stream_registrar.create_publisher.assert_called_once_with(stream_id="test_stream_id")

    def test_execute_acquire_data_with_stream_id_new_already_acquiring(self):
        self._bdh._semaphore = Mock()
        self._bdh._semaphore.acquire.return_value = False
        self._bdh.execute_acquire_data({"stream_id": "test_stream_id"})
        self._bdh._semaphore.acquire.assert_called_once_with(blocking=False)

    @patch("ion.agents.data.handlers.base_data_handler.ResourceRegistryServiceClient")
    def test__find_new_data_check_attachment(self, rr_cli_cls):
        rr_cli = rr_cli_cls.return_value
        attachment_1 = Mock()
        attachment_1._id = "attachment_1"
        attachment_1.keywords = ["NewDataCheck"]
        attachment_1.content = "content"
        rr_cli.find_attachments.return_value = [attachment_1]

        ret = self._bdh._find_new_data_check_attachment(res_id="res_id")
        self.assertEqual(ret, "content")
        rr_cli.find_attachments.assert_called_once_with(resource_id="res_id", include_content=False, id_only=False)

    @patch("ion.agents.data.handlers.base_data_handler.ResourceRegistryServiceClient")
    def test__find_new_data_check_attachment_no_newdatacheck(self, rr_cli_cls):
        rr_cli = rr_cli_cls.return_value
        attachment_1 = Mock()
        attachment_1._id = "attachment_1"
        attachment_1.keywords = ["not_found"]
        attachment_1.content = "content"
        rr_cli.find_attachments.return_value = [attachment_1]

        ret = self._bdh._find_new_data_check_attachment(res_id="res_id")
        self.assertEqual(ret, None)
        rr_cli.find_attachments.assert_called_once_with(resource_id="res_id", include_content=False, id_only=False)

    @patch("ion.agents.data.handlers.base_data_handler.ResourceRegistryServiceClient")
    def test__find_new_data_check_attachment_raise_notfound(self, rr_cli_cls):
        rr_cli = rr_cli_cls.return_value
        rr_cli.find_attachments.side_effect = NotFound

        with self.assertRaises(InstrumentException) as cm:
            self._bdh._find_new_data_check_attachment(res_id="not_found")

# 需要导入模块: from ion.agents.data.handlers.base_data_handler import BaseDataHandler [as 别名]
# 或者: from ion.agents.data.handlers.base_data_handler.BaseDataHandler import _find_new_data_check_attachment [as 别名]

#.........这里部分代码省略.........
        with self.assertRaises(ConfigurationError):
            self._bdh.execute_acquire_sample('not_found')

    @patch('ion.agents.data.handlers.base_data_handler.StandaloneStreamPublisher', spec=StandaloneStreamPublisher)
    @patch('ion.agents.data.handlers.base_data_handler.spawn')
    def test_execute_acquire_sample_with_stream_id_not_new(self, mock, StreamPublisher_mock):
        self._bdh._semaphore = Mock()
        self._bdh._glet_queue = Mock()
        stream_route = Mock(spec=StreamRoute)
        stream_route.exchange_point = sentinel.exchange_point
        stream_route.routing_key = sentinel.routing_key

        StreamPublisher_mock.return_value = MagicMock(spec=StandaloneStreamPublisher)

        self._bdh.execute_acquire_sample({'stream_id': 'test_stream_id',
                                          'constraints': 'test_constraints',
                                          'stream_route': stream_route})

    @patch('ion.agents.data.handlers.base_data_handler.StandaloneStreamPublisher', spec=StandaloneStreamPublisher)
    def test_execute_acquire_sample_with_stream_id_new_already_acquiring(self, StreamPublisher_mock):
        self._bdh._semaphore = Mock()
        self._bdh._semaphore.acquire.return_value = False
        stream_route = Mock(spec=StreamRoute)
        stream_route.exchange_point = sentinel.exchange_point
        stream_route.routing_key = sentinel.routing_key

        StreamPublisher_mock.return_value = MagicMock(spec=StandaloneStreamPublisher)

        self._bdh.execute_acquire_sample({'stream_id': 'test_stream_id',
                                          'stream_route': stream_route})
        self._bdh._semaphore.acquire.assert_called_once_with(blocking=False)

    @patch('ion.agents.data.handlers.base_data_handler.ResourceRegistryServiceClient')
    def test__find_new_data_check_attachment(self, rr_cli_cls):
        rr_cli = rr_cli_cls.return_value
        attachment_1 = Mock()
        attachment_1._id = 'attachment_1'
        attachment_1.keywords = ['NewDataCheck']
        attachment_1.content = msgpack.packb('content')
        rr_cli.find_attachments.return_value = [attachment_1]

        ret = self._bdh._find_new_data_check_attachment(res_id='res_id')
        self.assertEqual(ret, 'content')
        rr_cli.find_attachments.assert_called_once_with(resource_id='res_id',
            include_content=True,
            id_only=False)

    @patch('ion.agents.data.handlers.base_data_handler.ResourceRegistryServiceClient')
    def test__find_new_data_check_attachment_no_newdatacheck(self, rr_cli_cls):
        rr_cli = rr_cli_cls.return_value
        attachment_1 = Mock()
        attachment_1._id = 'attachment_1'
        attachment_1.keywords = ['not_found']
        attachment_1.content = 'content'
        rr_cli.find_attachments.return_value = [attachment_1]

        ret = self._bdh._find_new_data_check_attachment(res_id='res_id')
        self.assertEqual(ret, None)
        rr_cli.find_attachments.assert_called_once_with(resource_id='res_id',
            include_content=True,
            id_only=False)

    @patch('ion.agents.data.handlers.base_data_handler.ResourceRegistryServiceClient')
    def test__find_new_data_check_attachment_raise_notfound(self, rr_cli_cls):
        rr_cli = rr_cli_cls.return_value
        rr_cli.find_attachments.side_effect = NotFound