11from __future__ import unicode_literals
22
33from concurrent .futures import ThreadPoolExecutor
4- import inspect
54import time
65
76import pytest
1211 HistogramMetricFamily , Info , InfoMetricFamily , Metric , Sample ,
1312 StateSetMetricFamily , Summary , SummaryMetricFamily , UntypedMetricFamily ,
1413)
14+ from prometheus_client .decorator import getargspec
1515
1616try :
1717 import unittest2 as unittest
@@ -46,7 +46,7 @@ def f(r):
4646 else :
4747 raise TypeError
4848
49- self .assertEqual ((["r" ], None , None , None ), inspect . getargspec (f ))
49+ self .assertEqual ((["r" ], None , None , None ), getargspec (f ))
5050
5151 try :
5252 f (False )
@@ -107,7 +107,7 @@ def test_inprogress_function_decorator(self):
107107 def f ():
108108 self .assertEqual (1 , self .registry .get_sample_value ('g' ))
109109
110- self .assertEqual (([], None , None , None ), inspect . getargspec (f ))
110+ self .assertEqual (([], None , None , None ), getargspec (f ))
111111
112112 f ()
113113 self .assertEqual (0 , self .registry .get_sample_value ('g' ))
@@ -134,7 +134,7 @@ def test_time_function_decorator(self):
134134 def f ():
135135 time .sleep (.001 )
136136
137- self .assertEqual (([], None , None , None ), inspect . getargspec (f ))
137+ self .assertEqual (([], None , None , None ), getargspec (f ))
138138
139139 f ()
140140 self .assertNotEqual (0 , self .registry .get_sample_value ('g' ))
@@ -204,7 +204,7 @@ def test_function_decorator(self):
204204 def f ():
205205 pass
206206
207- self .assertEqual (([], None , None , None ), inspect . getargspec (f ))
207+ self .assertEqual (([], None , None , None ), getargspec (f ))
208208
209209 f ()
210210 self .assertEqual (1 , self .registry .get_sample_value ('s_count' ))
@@ -345,7 +345,7 @@ def test_function_decorator(self):
345345 def f ():
346346 pass
347347
348- self .assertEqual (([], None , None , None ), inspect . getargspec (f ))
348+ self .assertEqual (([], None , None , None ), getargspec (f ))
349349
350350 f ()
351351 self .assertEqual (1 , self .registry .get_sample_value ('h_count' ))
@@ -753,7 +753,7 @@ def test_restricted_registry(self):
753753
754754 m = Metric ('s' , 'help' , 'summary' )
755755 m .samples = [Sample ('s_sum' , {}, 7 )]
756- self .assertEquals ([m ], registry .restricted_registry (['s_sum' ]).collect ())
756+ self .assertEqual ([m ], registry .restricted_registry (['s_sum' ]).collect ())
757757
758758 def test_target_info_injected (self ):
759759 registry = CollectorRegistry (target_info = {'foo' : 'bar' })
@@ -777,11 +777,11 @@ def test_target_info_restricted_registry(self):
777777
778778 m = Metric ('s' , 'help' , 'summary' )
779779 m .samples = [Sample ('s_sum' , {}, 7 )]
780- self .assertEquals ([m ], registry .restricted_registry (['s_sum' ]).collect ())
780+ self .assertEqual ([m ], registry .restricted_registry (['s_sum' ]).collect ())
781781
782782 m = Metric ('target' , 'Target metadata' , 'info' )
783783 m .samples = [Sample ('target_info' , {'foo' : 'bar' }, 1 )]
784- self .assertEquals ([m ], registry .restricted_registry (['target_info' ]).collect ())
784+ self .assertEqual ([m ], registry .restricted_registry (['target_info' ]).collect ())
785785
786786
787787if __name__ == '__main__' :
0 commit comments