scalability_tester.py 2.79 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
#!/usr/bin/env python

##############################################################################
#
# Copyright (c) 2011 Nexedi SA and Contributors. All Rights Reserved.
#                    Arnaud Fontaine <arnaud.fontaine@nexedi.com>
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################

31 32 33 34 35 36
from benchmark import CSVBenchmarkResult

class CSVScalabilityBenchmarkResult(CSVBenchmarkResult):
  def flush(self):
    super(CSVScalabilityBenchmarkResult, self).flush()
    self._argument_namespace.notify_method(self._result_filename,
Arnaud Fontaine's avatar
Arnaud Fontaine committed
37
                                           self._result_file.tell())
38

39 40
from erp5.utils.benchmark.performance_tester import PerformanceTester

41
class ScalabilityTester(PerformanceTester):
Arnaud Fontaine's avatar
Arnaud Fontaine committed
42
  def preRun(self, *args, **kwargs):
43 44
    pass

Arnaud Fontaine's avatar
Arnaud Fontaine committed
45
  def postRun(self, *args, **kwargs):
46 47
    pass

48 49 50 51 52 53 54 55 56
  def getResultClass(self):
    if not self._argument_namespace.erp5_publish_url:
      return CSVScalabilityBenchmarkResult

    return super(ScalabilityTester, self).getResultClass()

from slapos.tool.nosqltester import NoSQLTester

class RunScalabilityTester(NoSQLTester):
57
  def __init__(self):
58
    super(RunScalabilityTester, self).__init__()
59 60

  def _add_parser_arguments(self, parser):
61 62
    super(RunScalabilityTester, self)._add_parser_arguments(parser)
    ScalabilityTester._add_parser_arguments(parser)
63 64

  def _parse_arguments(self, parser):
65 66 67
    namespace = super(RunScalabilityTester, self)._parse_arguments(parser)
    ScalabilityTester._check_parsed_arguments(namespace)
    namespace.notify_method = self.send_result_availability_notification
68 69 70
    return namespace

  def run_tester(self):
71
    ScalabilityTester(self.argument_namespace).run()
72 73

def main():
74
  RunScalabilityTester().run()
75 76 77

if __name__ == '__main__':
  main()