def test_constructor(self):
stdout = Mock()
stderr = Mock()
assert Worker(stdout=stdout).stdout is stdout
assert Worker(stderr=stderr).stderr is stderr
assert Worker(stdout=None).stdout is not stdout
assert Worker(stderr=None).stderr is not stderr
def test_setup_logging_no_log_level(self):
mock_log_handler = Mock()
worker_inst = Worker(
logfile="TEMP",
logformat="LOGFORMAT",
loghandlers=[mock_log_handler],
)
worker_inst._setup_logging()
self.setup_logging.assert_not_called()
def test_setup_logging(self, loghandlers):
worker_inst = Worker(
loglevel=5,
logfile='TEMP',
logging_config=None,
loghandlers=loghandlers,
)
worker_inst._setup_logging()
self.setup_logging.assert_called_once_with(
loglevel=5,
logfile='TEMP',
logging_config=None,
loghandlers=loghandlers or [],
)
def test_setup_logging(self, loghandlers):
worker_inst = Worker(
loglevel=5,
logfile="TEMP",
logformat="LOGFORMAT",
loghandlers=loghandlers,
)
worker_inst._setup_logging()
self.setup_logging.assert_called_once_with(
loglevel=5,
logfile="TEMP",
logformat="LOGFORMAT",
loghandlers=loghandlers,
)
async def test_on_first_start__override_logging(self):
worker = Worker(override_logging=False)
self._setup_for_on_first_start(worker)
await worker.on_first_start()
worker._setup_logging.assert_not_called()
def test_constructor_aligns_beacons(self):
x = Service()
y = Service()
Worker(x, y)
def worker(self):
return Worker(loglevel='INFO', logfile=None)
fh.write(graph.create_png())
@cached_property
def websockets(self) -> Websockets:
return Websockets(
port=self.websocket_port,
loop=self.loop,
beacon=self.beacon,
)
@cached_property
def webserver(self) -> Webserver:
return Webserver(
port=self.web_port,
bind=self.web_bind,
loop=self.loop,
beacon=self.beacon,
)
@cached_property
def user_cache(self) -> UserCache:
return UserCache(loop=self.loop, beacon=self.beacon)
app = App()
if __name__ == "__main__":
from mode.worker import Worker
Worker(app, loglevel="info", daemon=True).execute_from_commandline()
def worker(self):
return Worker(loglevel="INFO", logfile=None)
def _service(self) -> ServiceT:
return AppService(self, loop=self.loop)
@cached_property
def websockets(self) -> Websockets:
return Websockets(
port=self.websocket_port,
loop=self.loop,
beacon=self.beacon,
)
@cached_property
def webserver(self) -> Webserver:
return Webserver(
port=self.web_port,
bind=self.web_bind,
loop=self.loop,
beacon=self.beacon,
)
@cached_property
def user_cache(self) -> UserCache:
return UserCache(loop=self.loop, beacon=self.beacon)
app = App()
if __name__ == '__main__':
from mode.worker import Worker
Worker(app, loglevel='info').execute_from_commandline()