def test_widget(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config.merge({
"provisioning":
"test",
ScenarioExecutor.EXEC: [{
"throughput": 10,
"hold-for": 30,
"scenario": {
"default-address": "http://blazedemo.com/",
"requests": ["/"]
}
}]
})
obj.execution = obj.engine.config['execution'][0]
obj.settings.merge({
"path":
os.path.join(os.path.dirname(__file__), '..', "phantom.sh"),
})
obj.prepare()
obj.startup()
obj.get_widget()
self.assertTrue(isinstance(obj.widget.progress, urwid.ProgressBar))
self.assertEqual(obj.widget.duration, 30)
self.assertEqual(obj.widget.widgets[0].text,
"Target: http://blazedemo.com:80")
obj.check()
obj.shutdown()
def test_improved_request_building(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge(
yaml.load(
open(__dir__() +
"/../yaml/phantom_improved_request.yml").read()))
obj.execution = obj.engine.config['execution'][0]
obj.settings.merge({
"path":
os.path.join(os.path.dirname(__file__), '..', "phantom.sh"),
})
obj.prepare()
with open(obj.pbench.schedule_file) as fds:
config = fds.readlines()
get_requests = [
req_str.split(" ")[1] for req_str in config
if req_str.startswith("GET")
]
self.assertEqual(len(get_requests), 2)
for get_req in get_requests:
self.assertEqual(
dict(parse.parse_qsl(parse.urlsplit(get_req).query)), {
"get_param1": "value1",
"get_param2": "value2"
})
def test_diagnostics(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config.merge({
"provisioning":
"test",
ScenarioExecutor.EXEC: [{
"throughput": 10,
"hold-for": 30,
"scenario": {
"default-address": "http://blazedemo.com/",
"requests": ["/"]
}
}]
})
obj.execution = obj.engine.config['execution'][0]
obj.settings.merge({
"path": RESOURCES_DIR + "pbench/phantom.sh",
})
obj.prepare()
obj.startup()
for _ in range(3):
obj.check()
obj.shutdown()
obj.post_process()
self.assertIsNotNone(obj.get_error_diagnostics())
def test_widget(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config.merge({
"provisioning": "test",
ScenarioExecutor.EXEC: [
{
"throughput": 10,
"hold-for": 30,
"scenario": {
"default-address": "http://blazedemo.com/",
"requests": ["/"]
}
}
]})
obj.execution = obj.engine.config['execution'][0]
obj.settings.merge({
"path": os.path.join(os.path.dirname(__file__), '..', "phantom.sh"),
})
obj.prepare()
obj.startup()
obj.get_widget()
self.assertTrue(isinstance(obj.widget.progress, urwid.ProgressBar))
self.assertEqual(obj.widget.duration, 30)
self.assertEqual(obj.widget.widgets[0].text, "Target: http://blazedemo.com:80")
obj.check()
obj.shutdown()
def test_pbench_payload_relpath(self):
"Verify that enhanced pbench preserves relative script path"
script_path = "tests/data/pbench.src"
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge({
ScenarioExecutor.EXEC: {
"executor": "pbench",
"scenario": {"script": "tests/data/pbench.src"}
},
"provisioning": "test",
})
obj.execution = obj.engine.config['execution']
obj.settings.merge({
"path": os.path.join(os.path.dirname(__file__), '..', "phantom.sh"),
"enhanced": True,
})
obj.prepare()
pbench_conf = os.path.join(obj.engine.artifacts_dir, "pbench.conf")
with open(pbench_conf) as conf_fds:
config = conf_fds.read()
self.assertIn(script_path, config)
def test_pbench_payload_relpath(self):
"Verify that enhanced pbench preserves relative script path"
script_path = "tests/data/pbench.src"
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge({
ScenarioExecutor.EXEC: {
"executor": "pbench",
"scenario": {
"script": "tests/data/pbench.src"
}
},
"provisioning": "test",
})
obj.execution = obj.engine.config['execution']
obj.settings.merge({
"path":
os.path.join(os.path.dirname(__file__), '..', "phantom.sh"),
"enhanced":
True,
})
obj.prepare()
pbench_conf = os.path.join(obj.engine.artifacts_dir, "pbench.conf")
with open(pbench_conf) as conf_fds:
config = conf_fds.read()
self.assertIn(script_path, config)
def test_same_address_port(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge(yaml.load(open(__dir__() + "/../resources/yaml/phantom_request_same_address.yml").read()))
obj.execution = obj.engine.config['execution'][0]
obj.settings.merge({
"path": os.path.join(os.path.dirname(__file__), "..", "resources", "pbench", "phantom.sh"),
})
self.assertRaises(TaurusConfigError, obj.prepare)
def test_pbench_file_lister(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge(
{ScenarioExecutor.EXEC: {"executor": "pbench", "scenario": {"script": "/opt/data/script.src"}}}
)
obj.execution = obj.engine.config["execution"]
obj.settings.merge({"path": os.path.join(os.path.dirname(__file__), "..", "phantom.sh")})
resource_files = obj.resource_files()
self.assertEqual(1, len(resource_files))
self.assertEqual(resource_files[0], "script.src")
def test_same_address_port(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge(yaml.load(open(__dir__() + "/../yaml/phantom_request_same_address.yml").read()))
obj.execution = obj.engine.config["execution"][0]
obj.settings.merge({"path": os.path.join(os.path.dirname(__file__), "..", "phantom.sh")})
try:
obj.prepare()
self.fail()
except ValueError:
pass
def test_pbench_file_lister(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge(
{'execution': {"executor": "pbench", "scenario": {"script": "script.src"}}})
obj.execution = obj.engine.config['execution']
obj.settings.merge({
"path": os.path.join(os.path.dirname(__file__), "..", "resources", "pbench", "phantom.sh"),
})
resource_files = obj.resource_files()
self.assertEqual(1, len(resource_files))
self.assertEqual(resource_files[0], 'script.src')
def test_same_address_port(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge(
yaml.load(
open(RESOURCES_DIR +
"yaml/phantom_request_same_address.yml").read()))
obj.execution = obj.engine.config['execution'][0]
obj.settings.merge({
"path": RESOURCES_DIR + "pbench/phantom.sh",
})
self.assertRaises(TaurusConfigError, obj.prepare)
def test_pbench_script(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge({
ScenarioExecutor.EXEC: {
"executor": "pbench",
"scenario": {"script": __dir__() + "/../resources/pbench/pbench.src"}
},
"provisioning": "test"
})
obj.execution = obj.engine.config['execution']
obj.settings.merge({
"path": os.path.join(os.path.dirname(__file__), "..", "resources", "pbench", "phantom.sh"),
})
obj.prepare()
def test_pbench_script(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge({
ScenarioExecutor.EXEC: {
"executor": "pbench",
"scenario": {
"script": RESOURCES_DIR + "pbench/pbench.src"
}
},
"provisioning": "test"
})
obj.execution = obj.engine.config['execution']
obj.settings.merge({"path": RESOURCES_DIR + "pbench/phantom.sh"})
obj.prepare()
def test_pbench_payload_py3_crash(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge({
ScenarioExecutor.EXEC: {
"executor": "pbench",
"scenario": {"requests": ["test%d" % i for i in range(20)]}
},
"provisioning": "test",
})
obj.execution = obj.engine.config['execution']
obj.settings.merge({
"path": os.path.join(os.path.dirname(__file__), '..', "phantom.sh"),
})
obj.prepare()
def test_pbench_payload_py3_crash(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge({
ScenarioExecutor.EXEC: {
"executor": "pbench",
"scenario": {"requests": ["test%d" % i for i in range(20)]}
},
"provisioning": "test",
})
obj.execution = obj.engine.config['execution']
obj.settings.merge({
"path": os.path.join(os.path.dirname(__file__), "..", "resources", "pbench", "phantom.sh"),
})
obj.prepare()
def test_pbench_script(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge({
ScenarioExecutor.EXEC: {
"executor": "pbench",
"scenario": {"script": __dir__() + "/../data/pbench.src"}
},
"provisioning": "test"
})
obj.execution = obj.engine.config['execution']
obj.settings.merge({
"path": os.path.join(os.path.dirname(__file__), '..', "phantom.sh"),
})
obj.prepare()
def test_pbench_payload_py3_crash(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge({
ScenarioExecutor.EXEC: {
"executor": "pbench",
"scenario": {
"requests": ["test%d" % i for i in range(20)]
}
},
"provisioning": "test",
})
obj.execution = obj.engine.config['execution']
obj.settings.merge({
"path": RESOURCES_DIR + "pbench/phantom.sh",
})
obj.prepare()
def test_same_address_port(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge(
yaml.load(
open(__dir__() +
"/../yaml/phantom_request_same_address.yml").read()))
obj.execution = obj.engine.config['execution'][0]
obj.settings.merge({
"path":
os.path.join(os.path.dirname(__file__), '..', "phantom.sh"),
})
try:
obj.prepare()
self.fail()
except ValueError:
pass
def test_improved_request_building(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge(yaml.load(open(__dir__() + "/../yaml/phantom_improved_request.yml").read()))
obj.execution = obj.engine.config['execution'][0]
obj.settings.merge({
"path": os.path.join(os.path.dirname(__file__), '..', "phantom.sh"),
})
obj.prepare()
with open(obj.pbench.schedule_file) as fds:
config = fds.readlines()
get_requests = [req_str.split(" ")[1] for req_str in config if req_str.startswith("GET")]
self.assertEqual(len(get_requests), 2)
for get_req in get_requests:
self.assertEqual(dict(parse.parse_qsl(parse.urlsplit(get_req).query)),
{"get_param1": "value1", "get_param2": "value2"})
def test_pbench_file_lister(self):
obj = PBenchExecutor()
obj.engine = EngineEmul()
obj.settings = BetterDict()
obj.engine.config = BetterDict()
obj.engine.config.merge({
'execution': {
"executor": "pbench",
"scenario": {
"script": "script.src"
}
}
})
obj.execution = obj.engine.config['execution']
obj.settings.merge({
"path": RESOURCES_DIR + "pbench/phantom.sh",
})
resource_files = obj.resource_files()
self.assertEqual(1, len(resource_files))
self.assertEqual(resource_files[0], 'script.src')
