1
1
from __future__ import unicode_literals
2
72A8
td>+
2
3
import os
3
4
import shutil
4
5
import tempfile
5
- import time
6
6
import unittest
7
7
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
+
11
21
12
22
class TestMultiProcess (unittest .TestCase ):
13
23
def setUp (self ):
14
24
self .tempdir = tempfile .mkdtemp ()
15
25
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 )
17
27
self .registry = CollectorRegistry ()
18
28
MultiProcessCollector (self .registry , self .tempdir )
19
29
20
30
def tearDown (self ):
21
31
del os .environ ['prometheus_multiproc_dir' ]
22
32
shutil .rmtree (self .tempdir )
23
- prometheus_client . core ._ValueClass = prometheus_client . core ._MutexValue
33
+ core ._ValueClass = core ._MutexValue
24
34
25
35
def test_counter_adds (self ):
26
36
c1 = Counter ('c' , 'help' , registry = None )
27
- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
37
+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
28
38
c2 = Counter ('c' , 'help' , registry = None )
29
39
self .assertEqual (0 , self .registry .get_sample_value ('c' ))
30
40
c1 .inc (1 )
@@ -33,7 +43,7 @@ def test_counter_adds(self):
33
43
34
44
def test_summary_adds (self ):
35
45
s1 = Summary ('s' , 'help' , registry = None )
36
- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
46
+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
37
47
s2 = Summary ('s' , 'help' , registry = None )
38
48
self .assertEqual (0 , self .registry .get_sample_value ('s_count' ))
39
49
self .assertEqual (0 , self .registry .get_sample_value ('s_sum' ))
@@ -44,7 +54,7 @@ def test_summary_adds(self):
44
54
45
55
def test_histogram_adds (self ):
46
56
h1 = Histogram ('h' , 'help' , registry = None )
47
- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
57
+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
48
58
h2 = Histogram ('h' , 'help' , registry = None )
49
59
self .assertEqual (0 , self .registry .get_sample_value ('h_count' ))
50
60
self .assertEqual (0 , self .registry .get_sample_value ('h_sum' ))
@@ -57,7 +67,7 @@ def test_histogram_adds(self):
57
67
58
68
def test_gauge_all (self ):
59
69
g1 = Gauge ('g' , 'help' , registry = None )
60
- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (lambda : 456 )
70
+ core ._ValueClass = core ._MultiProcessValue (lambda : 456 )
61
71
g2 = Gauge ('g' , 'help' , registry = None )
62
72
self .assertEqual (0 , self .registry .get_sample_value ('g' , {'pid' : '123' }))
63
73
self .assertEqual (0 , self .registry .get_sample_value ('g' , {'pid' : '456' }))
@@ -69,7 +79,7 @@ def test_gauge_all(self):
69
79
70
80
def test_gauge_liveall (self ):
71
81
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 )
73
83
g2 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'liveall' )
74
84
self .assertEqual (0 , self .registry .get_sample_value ('g' , {'pid' : '123' }))
75
85
self .assertEqual (0 , self .registry .get_sample_value ('g' , {'pid' : '456' }))
@@ -83,7 +93,7 @@ def test_gauge_liveall(self):
83
93
84
94
def test_gauge_min (self ):
85
95
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 )
87
97
g2 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'min' )
88
98
self .assertEqual (0 , self .registry .get_sample_value ('g' ))
89
99
g1 .set (1 )
@@ -92,7 +102,7 @@ def test_gauge_min(self):
92
102
93
103
def test_gauge_max (self ):
94
104
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 )
96
106
g2 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'max' )
97
107
self .assertEqual (0 , self .registry .get_sample_value ('g' ))
98
108
g1 .set (1 )
@@ -101,7 +111,7 @@ def test_gauge_max(self):
101
111
102
112
def test_gauge_livesum (self ):
103
113
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 )
105
115
g2 = Gauge ('g' , 'help' , registry = None , multiprocess_mode = 'livesum' )
106
116
self .assertEqual (0 , self .registry .get_sample_value ('g' ))
107
117
g1 .set (1 )
@@ -119,7 +129,7 @@ def test_counter_across_forks(self):
119
129
pid = 0
120
130
def get_pid ():
121
131
return pid
122
- prometheus_client . core ._ValueClass = prometheus_client . core ._MultiProcessValue (get_pid )
132
+ core ._ValueClass = core ._MultiProcessValue (get_pid )
123
133
c1 = Counter ('c' , 'help' , registry = None )
124
134
self .assertEqual (0 , self .registry .get_sample_value ('c' ))
125
135
c1 .inc (1 )
@@ -168,19 +178,12 @@ def setUp(self):
168
178
169
179
def test_unset_syncdir_env (self ):
170
180
self .assertRaises (
171
- ValueError ,
172
- MultiProcessCollector ,
173
- self .registry
174
- )
181
+ ValueError , MultiProcessCollector , self .registry )
175
182
176
183
def test_file_syncpath (self ):
177
184
registry = CollectorRegistry ()
178
185
self .assertRaises (
179
- ValueError ,
180
- MultiProcessCollector ,
181
- registry ,
182
- self .tmpfl
183
- )
186
+ ValueError , MultiProcessCollector , registry , self .tmpfl )
184
187
185
188
def tearDown (self ):
186
189
os .remove (self .tmpfl )
0 commit comments