tests/common.py

   1 # -*- coding: utf-8 -*-
   2 import os
   3 import time
   4 import unittest2
   5 import xmlrpclib
   6
   7 import openerp
   8
   9 ADDONS_PATH = os.environ['OPENERP_ADDONS_PATH']
  10 PORT = int(os.environ['OPENERP_PORT'])
  11 DB = os.environ['OPENERP_DATABASE']
  12
  13 HOST = '127.0.0.1'
  14
  15 ADMIN_USER = 'admin'
  16 ADMIN_USER_ID = 1
  17 ADMIN_PASSWORD = 'admin'
  18
  19 common_proxy_60 = None
  20 db_proxy_60 = None
  21 object_proxy_60 = None
  22
  23 common_proxy_61 = None
  24 db_proxy_61 = None
  25 model_proxy_61 = None
  26 model_uri_61 = None
  27
  28 def setUpModule():
  29     """
  30     Start the OpenERP server similary to the openerp-server script and
  31     setup some xmlrpclib proxies.
  32     """
  33     openerp.tools.config['addons_path'] = ADDONS_PATH
  34     openerp.tools.config['xmlrpc_port'] = PORT
  35     openerp.service.start_services()
  36
  37     global common_proxy_60
  38     global db_proxy_60
  39     global object_proxy_60
  40
  41     # Use the old (pre 6.1) API.
  42     url = 'http://%s:%d/xmlrpc/' % (HOST, PORT)
  43     common_proxy_60 = xmlrpclib.ServerProxy(url + 'common')
  44     db_proxy_60 = xmlrpclib.ServerProxy(url + 'db')
  45     object_proxy_60 = xmlrpclib.ServerProxy(url + 'object')
  46
  47     global common_proxy_61
  48     global db_proxy_61
  49     global model_proxy_61
  50     global model_uri_61
  51
  52     # Use the new (6.1) API.
  53     model_uri_61 = 'http://%s:%d/openerp/xmlrpc/1/' % (HOST, PORT)
  54     common_proxy_61 = xmlrpclib.ServerProxy(model_uri_61 + 'common')
  55     db_proxy_61 = xmlrpclib.ServerProxy(model_uri_61 + 'db')
  56     model_proxy_61 = xmlrpclib.ServerProxy(model_uri_61 + 'model/' + DB)
  57
  58     # Ugly way to ensure the server is listening.
  59     time.sleep(2)
  60
  61 def tearDownModule():
  62     """ Shutdown the OpenERP server similarly to a single ctrl-c. """
  63     openerp.service.stop_services()