2424 from http .server import BaseHTTPRequestHandler
2525 from http .server import HTTPServer
2626
27+
2728class TestGenerateText (unittest .TestCase ):
2829 def setUp (self ):
2930 self .registry = CollectorRegistry ()
@@ -79,16 +80,20 @@ def test_escaping(self):
7980 self .assertEqual (b'# HELP cc A\\ ncount\\ \\ er\n # TYPE cc counter\n cc{a="\\ \\ x\\ n\\ ""} 1.0\n ' , generate_latest (self .registry ))
8081
8182 def test_nonnumber (self ):
82- class MyNumber ():
83+
84+ class MyNumber (object ):
8385 def __repr__ (self ):
8486 return "MyNumber(123)"
87+
8588 def __float__ (self ):
8689 return 123.0
87- class MyCollector ():
90+
91+ class MyCollector (object ):
8892 def collect (self ):
8993 metric = Metric ("nonnumber" , "Non number" , 'untyped' )
9094 metric .add_sample ("nonnumber" , {}, MyNumber ())
9195 yield metric
96+
9297 self .registry .register (MyCollector ())
9398 self .assertEqual (b'# HELP nonnumber Non number\n # TYPE nonnumber untyped\n nonnumber 123.0\n ' , generate_latest (self .registry ))
9499
@@ -98,6 +103,7 @@ def setUp(self):
98103 self .registry = CollectorRegistry ()
99104 self .counter = Gauge ('g' , 'help' , registry = self .registry )
100105 self .requests = requests = []
106+
101107 class TestHandler (BaseHTTPRequestHandler ):
102108 def do_PUT (self ):
103109 if 'with_basic_auth' in self .requestline and self .headers ['authorization' ] != 'Basic Zm9vOmJhcg==' :
@@ -113,9 +119,11 @@ def do_PUT(self):
113119
114120 httpd = HTTPServer (('localhost' , 0 ), TestHandler )
115121 self .address = 'http://localhost:{0}' .format (httpd .server_address [1 ])
122+
116123 class TestServer (threading .Thread ):
117124 def run (self ):
118125 httpd .handle_request ()
126+
119127 self .server = TestServer ()
120128 self .server .daemon = True
121129 self .server .start ()
0 commit comments