def test_includes_basic_default_arguments(self):
parser = argparse.ArgumentParser()
gpr.add_basic_perfscale_arguments(parser)
parsed_args = parser.parse_args(
['an-env', '/bin/juju', '/tmp/logs', 'an-env-mod'])
self.assertEqual(parsed_args, get_default_args())
def parse_args(argv):
"""Parse all arguments."""
parser = argparse.ArgumentParser(
description="Perfscale bundle deployment test.")
add_basic_perfscale_arguments(parser)
parser.add_argument('--bundle-name',
help='Bundle to deploy during test run.',
default='cs:~landscape/bundle/landscape-scalable')
return parser.parse_args(argv)
def test_must_be_enabled_when_requested(self):
parser = argparse.ArgumentParser()
gpr.add_basic_perfscale_arguments(parser)
args = parser.parse_args(['--enable-ha'])
client = Mock()
gpr.maybe_enable_ha(client, args)
client.enable_ha.assert_called_once_with()
client.wait_for_ha.assert_called_once_with()
def test_must_not_be_enabled_by_default(self):
parser = argparse.ArgumentParser()
gpr.add_basic_perfscale_arguments(parser)
args = parser.parse_args([])
client = Mock()
gpr.maybe_enable_ha(client, args)
self.assertEqual(0, client.enable_ha.call_count)
self.assertEqual(0, client.wait_for_ha.call_count)
def test_includes_basic_default_arguments(self):
parser = argparse.ArgumentParser()
gpr.add_basic_perfscale_arguments(parser)
parsed_args = parser.parse_args([
'an-env',
'/bin/juju',
'/tmp/logs',
'an-env-mod'])
self.assertEqual(parsed_args, get_default_args())
def test_adds_perfscale_arguments(self):
parser = argparse.ArgumentParser()
gpr.add_basic_perfscale_arguments(parser)
parsed_args = parser.parse_args([])
self.assertEqual(parsed_args.enable_ha, False)
self.assertEqual(parsed_args.enable_pprof, False)
parsed_args = parser.parse_args(['--enable-ha', '--enable-pprof'])
self.assertEqual(parsed_args.enable_ha, True)
self.assertEqual(parsed_args.enable_pprof, True)
def parse_args(argv):
"""Parse all arguments."""
parser = argparse.ArgumentParser(
description="Perfscale bundle deployment test.")
add_basic_perfscale_arguments(parser)
parser.add_argument('--model-count',
type=int,
help='Number of models to create.',
default=100)
return parser.parse_args(argv)
def test_must_be_enabled_when_requested(self):
parser = argparse.ArgumentParser()
gpr.add_basic_perfscale_arguments(parser)
args = parser.parse_args(['--enable-ha'])
client = Mock()
gpr.maybe_enable_ha(client, args)
client.enable_ha.assert_called_once_with()
client.wait_for_ha.assert_called_once_with()
def test_must_not_be_enabled_by_default(self):
parser = argparse.ArgumentParser()
gpr.add_basic_perfscale_arguments(parser)
args = parser.parse_args([])
client = Mock()
gpr.maybe_enable_ha(client, args)
self.assertEqual(0, client.enable_ha.call_count)
self.assertEqual(0, client.wait_for_ha.call_count)
def test_adds_perfscale_arguments(self):
parser = argparse.ArgumentParser()
gpr.add_basic_perfscale_arguments(parser)
parsed_args = parser.parse_args([])
self.assertEqual(parsed_args.enable_ha, False)
self.assertEqual(parsed_args.enable_pprof, False)
parsed_args = parser.parse_args(['--enable-ha', '--enable-pprof'])
self.assertEqual(parsed_args.enable_ha, True)
self.assertEqual(parsed_args.enable_pprof, True)
def parse_args(argv):
"""Parse all arguments."""
parser = argparse.ArgumentParser(
description='Perfscale xplod charm stree test.')
add_basic_perfscale_arguments(parser)
parser.add_argument('--deploy-amount',
help='The amount of deploys of xplod charm to do.',
type=int,
default=7)
return parser.parse_args(argv)
def parse_args(argv):
"""Parse all arguments."""
parser = argparse.ArgumentParser(
description='Perfscale Controller Stress test.')
add_basic_perfscale_arguments(parser)
parser.add_argument(
'--deploy-amount',
help='The amount of deploys to do.',
type=int,
default=1)
return parser.parse_args(argv)
