chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,25 @@
|
||||
import logging
|
||||
import sys
|
||||
|
||||
import pytest
|
||||
|
||||
from ray.dashboard.utils import close_logger_file_descriptor
|
||||
|
||||
|
||||
def test_close_logger_file_descriptor():
|
||||
logger_format = "%(message)s"
|
||||
logger = logging.getLogger("test_job_id")
|
||||
|
||||
job_driver_log_path = "/tmp/ray.log"
|
||||
job_driver_handler = logging.FileHandler(job_driver_log_path)
|
||||
job_driver_formatter = logging.Formatter(logger_format)
|
||||
job_driver_handler.setFormatter(job_driver_formatter)
|
||||
logger.addHandler(job_driver_handler)
|
||||
|
||||
assert job_driver_handler.stream.closed is False
|
||||
close_logger_file_descriptor(logger)
|
||||
assert job_driver_handler.stream is None
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(pytest.main(["-v", __file__]))
|
||||
Reference in New Issue
Block a user