def setUp(self): self._loop = asyncio.get_event_loop() self._loop.set_debug(True) self._temp_path = tempfile.mkdtemp(prefix='nuclio-test-py-wrapper') # write handler to temp path self._handler_path = self._write_handler(self._temp_path) # set PYTHONPATH to include temp path sys.path.append(self._temp_path) # generate socket path self._socket_path = os.path.join(self._temp_path, 'nuclio.sock') # create transport self._unix_stream_server = self._create_unix_stream_server(self._socket_path) # create logger self._logger = nuclio_sdk.Logger(logging.DEBUG) self._logger.set_handler('test-default', sys.stdout, nuclio_sdk.logger.HumanReadableFormatter()) self._platform_kind = 'test' self._default_test_handler = 'reverser:handler' # create a wrapper self._wrapper = wrapper.Wrapper(self._logger, self._loop, self._default_test_handler, self._socket_path, self._platform_kind, decode_event_strings=self._decode_event_strings) self._loop.run_until_complete(self._wrapper.initialize())
def test_face_prediction(): f = open("image.txt", "r") image = f.readline() f.close() logger = nuclio_sdk.Logger(level=logging.INFO) ctx = nuclio_sdk.Context(logger=logger) event = nuclio_sdk.Event(body=image) handler(ctx, event)
def __init__(self): self._logger = nuclio_sdk.Logger(logging.DEBUG, "mock_platform") self._logger.set_handler("default", sys.stdout, nuclio_sdk.logger.HumanReadableFormatter()) self._handler_contexts = {} self._call_function_mock = unittest.mock.MagicMock() self._kind = "test" # for tests that need a context self._context = nuclio_sdk.Context(self._logger, self)
def chain_call_function_mock(name, event, node=None, timeout=None, service_name_override=None): logger = nuclio_sdk.Logger(level=logging.DEBUG) logger.set_handler('default', sys.stdout, nuclio_sdk.logger.HumanReadableFormatter()) logger.debug_with("Call function mock called", name=name, service_name_override=service_name_override) if name == "face_prediction": nuclio_plat = nuclio_sdk.test.Platform() return nuclio_plat.call_handler(functions.face_prediction.handler, event) raise RuntimeError( 'Call function called with unexpected function name: {0}'.format(name))
def setUp(self): self._temp_path = tempfile.mkdtemp(prefix='nuclio-test-py-wrapper') # write handler to temp path self._handler_path = self._write_handler(self._temp_path) # set PYTHONPATH to include temp path sys.path.append(self._temp_path) # generate socket path self._socket_path = os.path.join(self._temp_path, 'nuclio.sock') # create transport self._unix_stream_server = self._create_unix_stream_server( self._socket_path) # create logger self._logger = nuclio_sdk.Logger(logging.DEBUG) self._logger.set_handler('test-default', sys.stdout, nuclio_sdk.logger.HumanReadableFormatter()) # create a wrapper self._wrapper = wrapper.Wrapper(self._logger, 'reverser:handler', self._socket_path, 'test')
def create_logger(level): """Create a logger that emits JSON to stdout""" return nuclio_sdk.Logger(level)