#!/usr/bin/env python ############################################################################## # Copyright (c) 2016 Huawei Technologies Co.,Ltd. # # All rights reserved. This program and the accompanying materials # are made available under the terms of the Apache License, Version 2.0 # which accompanies this distribution, and is available at # http://www.apache.org/licenses/LICENSE-2.0 ############################################################################## # Unittest for yardstick.benchmark.scenarios.storage.storperf.StorPerf import mock import unittest import requests import json from yardstick.benchmark.scenarios.storage import storperf def mocked_requests_config_post(*args, **kwargs): class MockResponseConfigPost: def __init__(self, json_data, status_code): self.content = json_data self.status_code = status_code return MockResponseConfigPost('{"stack_id": "dac27db1-3502-4300-b301-91c64e6a1622","stack_created": "false"}', 200) def mocked_requests_config_get(*args, **kwargs): class MockResponseConfigGet: def __init__(self, json_data, status_code): self.content = json_data self.status_code = status_code return MockResponseConfigGet('{"stack_id": "dac27db1-3502-4300-b301-91c64e6a1622","stack_created": "true"}', 200) def mocked_requests_job_get(*args, **kwargs): class MockResponseJobGet: def __init__(self, json_data, status_code): self.content = json_data self.status_code = status_code return MockResponseJobGet('{"_ssd_preconditioning.queue-depth.8.block-size.16384.duration": 6}', 200) def mocked_requests_job_post(*args, **kwargs): class MockResponseJobPost: def __init__(self, json_data, status_code): self.content = json_data self.status_code = status_code return MockResponseJobPost('{"job_id": \ "d46bfb8c-36f4-4a40-813b-c4b4a437f728"}', 200) def mocked_requests_job_delete(*args, **kwargs): class MockResponseJobDelete: def __init__(self, json_data, status_code): self.content = json_data self.status_code = status_code return MockResponseJobDelete('{}', 200) def mocked_requests_delete(*args, **kwargs): class MockResponseDelete: def __init__(self, json_data, status_code): self.json_data = json_data self.status_code = status_code return MockResponseDelete('{}', 200) def mocked_requests_delete_failed(*args, **kwargs): class MockResponseDeleteFailed: def __init__(self, json_data, status_code): self.json_data = json_data self.status_code = status_code if args[0] == "http://172.16.0.137:5000/api/v1.0/configurations": return MockResponseDeleteFailed('{"message": "Teardown failed"}', 400) return MockResponseDeleteFailed('{}', 404) class StorPerfTestCase(unittest.TestCase): def setUp(self): self.ctx = { 'host': { 'ip': '172.16.0.137', 'user': 'cirros', 'key_filename': "mykey.key" } } self.result = {} @mock.patch('yardstick.benchmark.scenarios.storage.storperf.requests.post', side_effect=mocked_requests_config_post) @mock.patch('yardstick.benchmark.scenarios.storage.storperf.requests.get', side_effect=mocked_requests_config_get) def test_successful_setup(self, mock_post, mock_get): options = { "agent_count": 8, "public_network": 'ext-net', "volume_size": 10, "block_sizes": 4096, "queue_depths": 4, "workload": "rs", "StorPerf_ip": "192.168.23.2", "query_interval": 10, "timeout": 60 } args = { "options": options } s = storperf.StorPerf(args, self.ctx) s.setup() self.assertTrue(s.setup_done) @mock.patch('yardstick.benchmark.scenarios.storage.storperf.requests.post', side_effect=mocked_requests_job_post) @mock.patch('yardstick.benchmark.scenarios.storage.storperf.requests.get', side_effect=mocked_requests_job_get) @mock.patch('yardstick.benchmark.scenarios.storage.storperf.requests.delete', side_effect=mocked_requests_job_delete) def test_successful_run(self, mock_post, mock_get, mock_delete): options = { "agent_count": 8, "public_network": 'ext-net', "volume_size": 10, "block_sizes": 4096, "queue_depths": 4, "workload": "rs", "StorPerf_ip": "192.168.23.2", "query_interval": 10, "timeout": 60 } args = { "options": options } s = storperf.StorPerf(args, self.ctx) s.setup_done = True sample_output = '{"_ssd_preconditioning.queue-depth.8.block-size.16384.duration": 6}' expected_result = json.loads(sample_output) s.run(self.result) self.assertEqual(self.result, expected_result) @mock.patch('yardstick.benchmark.scenarios.storage.storperf.requests.delete', side_effect=mocked_requests_delete) def test_successful_teardown(self, mock_delete): options = { "agent_count": 8, "public_network": 'ext-net', "volume_size": 10, "block_sizes": 4096, "queue_depths": 4, "workload": "rs", "StorPerf_ip": "192.168.23.2", "query_interval": 10, "timeout": 60 } args = { "options": options } s = storperf.StorPerf(args, self.ctx) s.teardown() self.assertFalse(s.setup_done) @mock.patch('yardstick.benchmark.scenarios.storage.storperf.requests.delete', side_effect=mocked_requests_delete_failed) def test_failed_teardown(self, mock_delete): options = { "agent_count": 8, "public_network": 'ext-net', "volume_size": 10, "block_sizes": 4096, "queue_depths": 4, "workload": "rs", "StorPerf_ip": "192.168.23.2", "query_interval": 10, "timeout": 60 } args = { "options": options } s = storperf.StorPerf(args, self.ctx) self.assertRaises(AssertionError, s.teardown(), self.result) def main(): unittest.main() if __name__ == '__main__': main()