apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-17 22:21:48 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Use uuid to name the metrics shared folder (#4025)

Browse Source 
* Use uuid to name the metrics shared folder

* Use uuid to name the metrics shared folder test case
This commit is contained in:
guozhuangzhuang
2025-09-10 16:58:13 +08:00
committed by GitHub
parent 2f473ba966
commit f06e3ee1fc
2 changed files with 39 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
tests/metrics/test_metrics.py
View File

@@ -14,12 +14,13 @@
# limitations under the License.
"""
import os
import unittest
from unittest.mock import patch
from prometheus_client import Gauge
from fastdeploy.metrics.metrics import get_filtered_metrics
from fastdeploy.metrics.metrics import cleanup_prometheus_files, get_filtered_metrics
class TestGetFilteredMetrics(unittest.TestCase):
@@ -68,5 +69,30 @@ class TestGetFilteredMetrics(unittest.TestCase):
self.assertIn("custom_metric_total", result)
class TestCleanupPrometheusFiles(unittest.TestCase):
def test_cleanup_prometheus_files(self):
prom_dir = cleanup_prometheus_files(is_main=True)
self.assertTrue(os.path.exists(prom_dir))
self.assertTrue(os.path.isdir(prom_dir))
prom_dir2 = cleanup_prometheus_files(is_main=True)
self.assertNotEqual(prom_dir, prom_dir2)
self.assertTrue(os.path.exists(prom_dir2))
self.assertTrue(os.path.isdir(prom_dir2))
prom_dir = cleanup_prometheus_files(is_main=True, instance_id="001")
self.assertTrue(os.path.exists(prom_dir))
self.assertTrue(os.path.isdir(prom_dir))
test_file = os.path.join(prom_dir, "test.txt")
with open(test_file, "w") as f:
f.write("hello")
self.assertTrue(os.path.exists(test_file))
prom_dir2 = cleanup_prometheus_files(is_main=True, instance_id="001")
self.assertEqual(prom_dir, prom_dir2)
self.assertTrue(os.path.exists(prom_dir2))
self.assertTrue(os.path.isdir(prom_dir2))
self.assertFalse(os.path.exists(test_file))
if __name__ == "__main__":
unittest.main()