__init__.py

import os.path
import json

PERSON_KEY = "person_per_hour"
ORDER_KEY = "sale_order_per_hour"

class ERP5_scalability():

  def getTestList(self):
    return ['createPerson', 'createSaleOrder']

  def getTestPath(self):
    return 'example/'

  def getUsersFilePath(self):
    return 'example/scalabilityUsers'

  def getUserQuantity(self, test_number):
    return [20, 30, 40, 50, 60][test_number]

  # Test duration in seconds
  def getTestDuration(self, test_number):
    return 8*self.getUserQuantity(test_number)

  def getTestRepetition(self, test_number):
    return 3

  def getScalabilityTestUrl(self, instance_information_dict):
    frontend_address = instance_information_dict['frontend-url-list'][0]
    return "%s/erp5" % frontend_address

  def getScalabilityTestMetricUrl(self, instance_information_dict, **kw):
    frontend_address = instance_information_dict['frontend-url-list'][0]
    metrics_url = frontend_address.replace("https://",
                    "https://%s:%s@" % (instance_information_dict['user'],
                                        instance_information_dict['password']))
    return metrics_url + "/erp5/ERP5Site_getScalabilityTestMetric"

  def getScalabilityTestOutput(self, metric_list):
    """
    From the list of metrics taken during a test run, select the best metric
    for the test output by a specific criteria
    """
    if not metric_list: return None
    output_json = json.loads(metric_list[0])
    for metric in metric_list:
      metric_json = json.loads(metric)
      if metric_json[PERSON_KEY] > output_json[PERSON_KEY]:
        output_json[PERSON_KEY] = metric_json[PERSON_KEY]
      if metric_json[ORDER_KEY] > output_json[ORDER_KEY]:
        output_json[ORDER_KEY] = metric_json[ORDER_KEY]
    return "Person: %s doc/hour; SaleOrder: %s doc/hour;" % (
            str(output_json[PERSON_KEY]), str(output_json[ORDER_KEY]))

  def getBootstrapScalabilityTestUrl(self, instance_information_dict, count=0, **kw):
    frontend_address = instance_information_dict['frontend-url-list'][0]
    bootstrap_url = frontend_address.replace("https://",
                      "https://%s:%s@" % (instance_information_dict['user'],
                                          instance_information_dict['password']))
    bootstrap_url += "/erp5/ERP5Site_bootstrapScalabilityTest"
    bootstrap_url += "?user_quantity=%i" % self.getUserQuantity(count)
    return bootstrap_url