Remove hardcoded version number from tests

Supervisor · May 17, 2011 · 9867362 · 9867362
1 parent 50f2962
commit 9867362
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/supervisor/tests/test_xmlrpc.py b/src/supervisor/tests/test_xmlrpc.py
@@ -87,7 +87,8 @@ def test_continue_request_methodsuccess(self):
         self.assertEqual(len(request.producers), 1)
         xml_response = request.producers[0]
         response = xmlrpclib.loads(xml_response)
-        self.assertEqual(response[0][0], '3.0')
+        from supervisor.rpcinterface import API_VERSION
+        self.assertEqual(response[0][0], API_VERSION)
         self.assertEqual(request._done, True)
         self.assertEqual(request.headers['Content-Type'], 'text/xml')
         self.assertEqual(request.headers['Content-Length'], len(xml_response))
@@ -117,7 +118,8 @@ def test_continue_request_no_params_in_request(self):
         xml_response = request.producers[0]
         import xmlrpclib
         response = xmlrpclib.loads(xml_response)
-        self.assertEqual(response[0][0], '3.0')
+        from supervisor.rpcinterface import API_VERSION
+        self.assertEqual(response[0][0], API_VERSION)
         self.assertEqual(request._done, True)
         self.assertEqual(request.headers['Content-Type'], 'text/xml')
         self.assertEqual(request.headers['Content-Length'], len(xml_response))