11from __future__ import unicode_literals
2+
23import os
34import shutil
45import tempfile
5- import time
66import unittest
77
8- import prometheus_client
9- from prometheus_client .core import *
10- from prometheus_client .multiprocess import *
8+ from prometheus_client import core
9+ from prometheus_client .core import (
10+ CollectorRegistry ,
11+ Counter ,
12+ Gauge ,
13+ Histogram ,
14+ Summary ,
15+ )
16+ from prometheus_client .multiprocess import (
17+ mark_process_dead ,
18+ MultiProcessCollector ,
19+ )
20+
1121
1222class TestMultiProcess (unittest .TestCase ):
1323 def setUp (self ):
1424 self .tempdir = tempfile .mkdtemp ()
1525 os .environ ['prometheus_multiproc_dir' ] = self .tempdir
16- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 123 )
26+ core ._ValueClass = core ._MultiProcessValue (lambda : 123 )
1727 self .registry = CollectorRegistry ()
1828 MultiProcessCollector (self .registry , self .tempdir )
1929
2030 def tearDown (self ):
2131 del os .environ ['prometheus_multiproc_dir' ]
2232 shutil .rmtree (self .tempdir )
23- prometheus_client . core ._ValueClass = prometheus_client . core ._MutexValue
33+ core ._ValueClass = core ._MutexValue
2434
2535 def test_counter_adds (self ):
2636 c1 = Counter ('c' , 'help' , registry = None )
27- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
37+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
2838 c2 = Counter ('c' , 'help' , registry = None )
2939 self .assertEqual (0 , self .registry .get_sample_value ('c' ))
3040 c1 .inc (1 )
@@ -33,7 +43,7 @@ def test_counter_adds(self):
3343
3444 def test_summary_adds (self ):
3545 s1 = Summary ('s' , 'help' , registry = None )
36- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
46+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
3747 s2 = Summary ('s' , 'help' , registry = None )
3848 self .assertEqual (0 , self .registry .get_sample_value ('s_count' ))
3949 self .assertEqual (0 , self .registry .get_sample_value ('s_sum' ))
@@ -44,7 +54,7 @@ def test_summary_adds(self):
4454
4555 def test_histogram_adds (self ):
4656 h1 = Histogram ('h' , 'help' , registry = None )
47- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
57+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
4858 h2 = Histogram ('h' , 'help' , registry = None )
4959 self .assertEqual (0 , self .registry .get_sample_value ('h_count' ))
5060 self .assertEqual (0 , self .registry .get_sample_value ('h_sum' ))
@@ -57,7 +67,7 @@ def test_histogram_adds(self):
5767
5868 def test_gauge_all (self ):
5969 g1 = Gauge ('g' , 'help' , registry = None )
60- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
70+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
6171 g2 = Gauge ('g' , 'help' , registry = None )
6272 self .assertEqual (0 , self .registry .get_sample_value ('g' , {'pid' : '123' }))
6373 self .assertEqual (0 , self .registry .get_sample_value ('g' , {'pid' : '456' }))
@@ -69,7 +79,7 @@ def test_gauge_all(self):
6979
7080 def test_gauge_liveall (self ):
7181 g1 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'liveall' )
72- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
82+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
7383 g2 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'liveall' )
7484 self .assertEqual (0 , self .registry .get_sample_value ('g' , {'pid' : '123' }))
7585 self .assertEqual (0 , self .registry .get_sample_value ('g' , {'pid' : '456' }))
@@ -83,7 +93,7 @@ def test_gauge_liveall(self):
8393
8494 def test_gauge_min (self ):
8595 g1 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'min' )
86- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
96+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
8797 g2 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'min' )
8898 self .assertEqual (0 , self .registry .get_sample_value ('g' ))
8999 g1 .set (1 )
@@ -92,7 +102,7 @@ def test_gauge_min(self):
92102
93103 def test_gauge_max (self ):
94104 g1 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'max' )
95- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
105+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
96106 g2 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'max' )
97107 self .assertEqual (0 , self .registry .get_sample_value ('g' ))
98108 g1 .set (1 )
@@ -101,7 +111,7 @@ def test_gauge_max(self):
101111
102112 def test_gauge_livesum (self ):
103113 g1 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'livesum' )
104- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
114+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
105115 g2 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'livesum' )
106116 self .assertEqual (0 , self .registry .get_sample_value ('g' ))
107117 g1 .set (1 )
@@ -119,7 +129,7 @@ def test_counter_across_forks(self):
119129 pid = 0
120130 def get_pid ():
121131 return pid
122- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (get_pid )
132+ core ._ValueClass = core ._MultiProcessValue (get_pid )
123133 c1 = Counter ('c' , 'help' , registry = None )
124134 self .assertEqual (0 , self .registry .get_sample_value ('c' ))
125135 c1 .inc (1 )
@@ -168,19 +178,12 @@ def setUp(self):
168178
169179 def test_unset_syncdir_env (self ):
170180 self .assertRaises (
171- ValueError ,
172- MultiProcessCollector ,
173- self .registry
174- )
181+ ValueError , MultiProcessCollector , self .registry )
175182
176183 def test_file_syncpath (self ):
177184 registry = CollectorRegistry ()
178185 self .assertRaises (
179- ValueError ,
180- MultiProcessCollector ,
181- registry ,
182- self .tmpfl
183- )
186+ ValueError , MultiProcessCollector , registry , self .tmpfl )
184187
185188 def tearDown (self ):
186189 os .remove (self .tmpfl )
0 commit comments