Ejemplo n.º 1
0
    def test_create_driver(self):
        driver = logging.make_driver()
        request = Subject()
        sink = logging.Sink(request=request)
        source = driver.call(sink)

        self.assertIsNone(source)
Ejemplo n.º 2
0
def main():
    run(
        Component(call=deepspeech_server, input=DeepspeechSource),
        DeepspeechDrivers(deepspeech=deepspeech.make_driver(),
                          httpd=httpd.make_driver(),
                          argv=argv.make_driver(),
                          logging=logging.make_driver(),
                          file=file.make_driver()))
Ejemplo n.º 3
0
    def test_create_driver(self):
        driver = logging.make_driver()
        request = Subject()
        sink = logging.Sink(request=request)
        source = driver.call(sink)

        self.assertIsInstance(source, logging.Source)
        self.assertIsNotNone(source.response)
Ejemplo n.º 4
0
    def test_log(self, mock_get_logger):
        logger = MagicMock(spec=std_logging.Logger)
        mock_get_logger.return_value = logger

        driver = logging.make_driver()
        request = Subject()
        sink = logging.Sink(request=request)
        driver.call(sink)

        request.on_next(
            logging.Log(logger='foo', level='DEBUG', message='foo msg'))
        mock_get_logger.assert_called_with('foo')
        logger.log.assert_called_with('DEBUG', 'foo msg')
Ejemplo n.º 5
0
def main():
    loop = asyncio.get_event_loop()
    # loop.set_debug(True)
    aio_scheduler = AsyncIOScheduler(loop=loop)
    run(
        Component(call=partial(deepspeech_server, aio_scheduler),
                  input=DeepspeechSource),
        DeepspeechDrivers(deepspeech=deepspeech.make_driver(),
                          httpd=httpd.make_driver(),
                          argv=argv.make_driver(),
                          logging=logging.make_driver(),
                          file=file.make_driver()),
        loop=loop,
    )
Ejemplo n.º 6
0
    def test_set_level(self, mock_get_logger, mock_stream_handler):
        logger = MagicMock(spec=std_logging.Logger)
        stream_handler = MagicMock(spec=std_logging.Handler)
        mock_get_logger.return_value = logger
        mock_stream_handler.return_value = stream_handler

        driver = logging.make_driver()
        request = Subject()
        sink = logging.Sink(request=request)
        driver.call(sink)

        request.on_next(logging.SetLevel(logger='foo', level='DEBUG'))
        mock_get_logger.assert_called_with('foo')
        logger.setLevel.assert_called_with(std_logging.DEBUG)
        mock_stream_handler.assert_called_with()
        stream_handler.setLevel.assert_called_with(std_logging.DEBUG)