1fb13bbf2SFam Zheng#!/usr/bin/env python 2fb13bbf2SFam Zheng# 3fb13bbf2SFam Zheng# Tests for IO throttling 4fb13bbf2SFam Zheng# 5fb13bbf2SFam Zheng# Copyright (C) 2015 Red Hat, Inc. 6a90cade0SAlberto Garcia# Copyright (C) 2015-2016 Igalia, S.L. 7fb13bbf2SFam Zheng# 8fb13bbf2SFam Zheng# This program is free software; you can redistribute it and/or modify 9fb13bbf2SFam Zheng# it under the terms of the GNU General Public License as published by 10fb13bbf2SFam Zheng# the Free Software Foundation; either version 2 of the License, or 11fb13bbf2SFam Zheng# (at your option) any later version. 12fb13bbf2SFam Zheng# 13fb13bbf2SFam Zheng# This program is distributed in the hope that it will be useful, 14fb13bbf2SFam Zheng# but WITHOUT ANY WARRANTY; without even the implied warranty of 15fb13bbf2SFam Zheng# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16fb13bbf2SFam Zheng# GNU General Public License for more details. 17fb13bbf2SFam Zheng# 18fb13bbf2SFam Zheng# You should have received a copy of the GNU General Public License 19fb13bbf2SFam Zheng# along with this program. If not, see <http://www.gnu.org/licenses/>. 20fb13bbf2SFam Zheng# 21fb13bbf2SFam Zheng 22fb13bbf2SFam Zhengimport iotests 23fb13bbf2SFam Zheng 24a90cade0SAlberto Garciansec_per_sec = 1000000000 25a90cade0SAlberto Garcia 26fb13bbf2SFam Zhengclass ThrottleTestCase(iotests.QMPTestCase): 27fb13bbf2SFam Zheng test_img = "null-aio://" 282db33f88SAlberto Garcia max_drives = 3 29fb13bbf2SFam Zheng 30fb13bbf2SFam Zheng def blockstats(self, device): 31fb13bbf2SFam Zheng result = self.vm.qmp("query-blockstats") 32fb13bbf2SFam Zheng for r in result['return']: 33fb13bbf2SFam Zheng if r['device'] == device: 34fb13bbf2SFam Zheng stat = r['stats'] 35fb13bbf2SFam Zheng return stat['rd_bytes'], stat['rd_operations'], stat['wr_bytes'], stat['wr_operations'] 36fb13bbf2SFam Zheng raise Exception("Device not found for blockstats: %s" % device) 37fb13bbf2SFam Zheng 38fb13bbf2SFam Zheng def setUp(self): 392db33f88SAlberto Garcia self.vm = iotests.VM() 402db33f88SAlberto Garcia for i in range(0, self.max_drives): 41*a6862418SAndrey Shinkevich self.vm.add_drive(self.test_img, "file.read-zeroes=on") 42fb13bbf2SFam Zheng self.vm.launch() 43fb13bbf2SFam Zheng 44fb13bbf2SFam Zheng def tearDown(self): 45fb13bbf2SFam Zheng self.vm.shutdown() 46fb13bbf2SFam Zheng 47a90cade0SAlberto Garcia def configure_throttle(self, ndrives, params): 482db33f88SAlberto Garcia params['group'] = 'test' 49fb13bbf2SFam Zheng 502db33f88SAlberto Garcia # Set the I/O throttling parameters to all drives 512db33f88SAlberto Garcia for i in range(0, ndrives): 522db33f88SAlberto Garcia params['device'] = 'drive%d' % i 53fb13bbf2SFam Zheng result = self.vm.qmp("block_set_io_throttle", conv_keys=False, **params) 54fb13bbf2SFam Zheng self.assert_qmp(result, 'return', {}) 55fb13bbf2SFam Zheng 56a26ddb43SAlberto Garcia def do_test_throttle(self, ndrives, seconds, params, first_drive = 0): 57a90cade0SAlberto Garcia def check_limit(limit, num): 58a90cade0SAlberto Garcia # IO throttling algorithm is discrete, allow 10% error so the test 59a90cade0SAlberto Garcia # is more robust 60a90cade0SAlberto Garcia return limit == 0 or \ 61a90cade0SAlberto Garcia (num < seconds * limit * 1.1 / ndrives 62a90cade0SAlberto Garcia and num > seconds * limit * 0.9 / ndrives) 63a90cade0SAlberto Garcia 64fb13bbf2SFam Zheng # Set vm clock to a known value 65fb13bbf2SFam Zheng ns = seconds * nsec_per_sec 66fb13bbf2SFam Zheng self.vm.qtest("clock_step %d" % ns) 67fb13bbf2SFam Zheng 68a90cade0SAlberto Garcia # Submit enough requests so the throttling mechanism kicks 69a90cade0SAlberto Garcia # in. The throttled requests won't be executed until we 70a90cade0SAlberto Garcia # advance the virtual clock. 71fb13bbf2SFam Zheng rq_size = 512 729a3a9a63SMax Reitz rd_nr = max(params['bps'] // rq_size // 2, 739a3a9a63SMax Reitz params['bps_rd'] // rq_size, 749a3a9a63SMax Reitz params['iops'] // 2, 75fb13bbf2SFam Zheng params['iops_rd']) 76fb13bbf2SFam Zheng rd_nr *= seconds * 2 779a3a9a63SMax Reitz rd_nr //= ndrives 789a3a9a63SMax Reitz wr_nr = max(params['bps'] // rq_size // 2, 799a3a9a63SMax Reitz params['bps_wr'] // rq_size, 809a3a9a63SMax Reitz params['iops'] // 2, 81fb13bbf2SFam Zheng params['iops_wr']) 82fb13bbf2SFam Zheng wr_nr *= seconds * 2 839a3a9a63SMax Reitz wr_nr //= ndrives 84fb13bbf2SFam Zheng 852db33f88SAlberto Garcia # Send I/O requests to all drives 862db33f88SAlberto Garcia for i in range(rd_nr): 872db33f88SAlberto Garcia for drive in range(0, ndrives): 88a26ddb43SAlberto Garcia idx = first_drive + drive 89a26ddb43SAlberto Garcia self.vm.hmp_qemu_io("drive%d" % idx, "aio_read %d %d" % 902db33f88SAlberto Garcia (i * rq_size, rq_size)) 912db33f88SAlberto Garcia 922db33f88SAlberto Garcia for i in range(wr_nr): 932db33f88SAlberto Garcia for drive in range(0, ndrives): 94a26ddb43SAlberto Garcia idx = first_drive + drive 95a26ddb43SAlberto Garcia self.vm.hmp_qemu_io("drive%d" % idx, "aio_write %d %d" % 962db33f88SAlberto Garcia (i * rq_size, rq_size)) 972db33f88SAlberto Garcia 982db33f88SAlberto Garcia # We'll store the I/O stats for each drive in these arrays 992db33f88SAlberto Garcia start_rd_bytes = [0] * ndrives 1002db33f88SAlberto Garcia start_rd_iops = [0] * ndrives 1012db33f88SAlberto Garcia start_wr_bytes = [0] * ndrives 1022db33f88SAlberto Garcia start_wr_iops = [0] * ndrives 1032db33f88SAlberto Garcia end_rd_bytes = [0] * ndrives 1042db33f88SAlberto Garcia end_rd_iops = [0] * ndrives 1052db33f88SAlberto Garcia end_wr_bytes = [0] * ndrives 1062db33f88SAlberto Garcia end_wr_iops = [0] * ndrives 1072db33f88SAlberto Garcia 1082db33f88SAlberto Garcia # Read the stats before advancing the clock 1092db33f88SAlberto Garcia for i in range(0, ndrives): 110a26ddb43SAlberto Garcia idx = first_drive + i 1112db33f88SAlberto Garcia start_rd_bytes[i], start_rd_iops[i], start_wr_bytes[i], \ 112a26ddb43SAlberto Garcia start_wr_iops[i] = self.blockstats('drive%d' % idx) 113fb13bbf2SFam Zheng 114fb13bbf2SFam Zheng self.vm.qtest("clock_step %d" % ns) 115fb13bbf2SFam Zheng 1162db33f88SAlberto Garcia # Read the stats after advancing the clock 1172db33f88SAlberto Garcia for i in range(0, ndrives): 118a26ddb43SAlberto Garcia idx = first_drive + i 1192db33f88SAlberto Garcia end_rd_bytes[i], end_rd_iops[i], end_wr_bytes[i], \ 120a26ddb43SAlberto Garcia end_wr_iops[i] = self.blockstats('drive%d' % idx) 1212db33f88SAlberto Garcia 1222db33f88SAlberto Garcia # Check that the I/O is within the limits and evenly distributed 1232db33f88SAlberto Garcia for i in range(0, ndrives): 1242db33f88SAlberto Garcia rd_bytes = end_rd_bytes[i] - start_rd_bytes[i] 1252db33f88SAlberto Garcia rd_iops = end_rd_iops[i] - start_rd_iops[i] 1262db33f88SAlberto Garcia wr_bytes = end_wr_bytes[i] - start_wr_bytes[i] 1272db33f88SAlberto Garcia wr_iops = end_wr_iops[i] - start_wr_iops[i] 128fb13bbf2SFam Zheng 129fb13bbf2SFam Zheng self.assertTrue(check_limit(params['bps'], rd_bytes + wr_bytes)) 130fb13bbf2SFam Zheng self.assertTrue(check_limit(params['bps_rd'], rd_bytes)) 131fb13bbf2SFam Zheng self.assertTrue(check_limit(params['bps_wr'], wr_bytes)) 132fb13bbf2SFam Zheng self.assertTrue(check_limit(params['iops'], rd_iops + wr_iops)) 133fb13bbf2SFam Zheng self.assertTrue(check_limit(params['iops_rd'], rd_iops)) 134fb13bbf2SFam Zheng self.assertTrue(check_limit(params['iops_wr'], wr_iops)) 135fb13bbf2SFam Zheng 136cbaddb25SStefan Hajnoczi # Allow remaining requests to finish. We submitted twice as many to 137cbaddb25SStefan Hajnoczi # ensure the throttle limit is reached. 138cbaddb25SStefan Hajnoczi self.vm.qtest("clock_step %d" % ns) 139cbaddb25SStefan Hajnoczi 140a26ddb43SAlberto Garcia # Connect N drives to a VM and test I/O in all of them 141fb13bbf2SFam Zheng def test_all(self): 142fb13bbf2SFam Zheng params = {"bps": 4096, 143fb13bbf2SFam Zheng "bps_rd": 4096, 144fb13bbf2SFam Zheng "bps_wr": 4096, 145fb13bbf2SFam Zheng "iops": 10, 146fb13bbf2SFam Zheng "iops_rd": 10, 147fb13bbf2SFam Zheng "iops_wr": 10, 148fb13bbf2SFam Zheng } 1492db33f88SAlberto Garcia # Repeat the test with different numbers of drives 1502db33f88SAlberto Garcia for ndrives in range(1, self.max_drives + 1): 151fb13bbf2SFam Zheng # Pick each out of all possible params and test 152fb13bbf2SFam Zheng for tk in params: 153fb13bbf2SFam Zheng limits = dict([(k, 0) for k in params]) 1542db33f88SAlberto Garcia limits[tk] = params[tk] * ndrives 155a90cade0SAlberto Garcia self.configure_throttle(ndrives, limits) 156a90cade0SAlberto Garcia self.do_test_throttle(ndrives, 5, limits) 157a90cade0SAlberto Garcia 158a26ddb43SAlberto Garcia # Connect N drives to a VM and test I/O in just one of them a time 159a26ddb43SAlberto Garcia def test_one(self): 160a26ddb43SAlberto Garcia params = {"bps": 4096, 161a26ddb43SAlberto Garcia "bps_rd": 4096, 162a26ddb43SAlberto Garcia "bps_wr": 4096, 163a26ddb43SAlberto Garcia "iops": 10, 164a26ddb43SAlberto Garcia "iops_rd": 10, 165a26ddb43SAlberto Garcia "iops_wr": 10, 166a26ddb43SAlberto Garcia } 167a26ddb43SAlberto Garcia # Repeat the test for each one of the drives 168a26ddb43SAlberto Garcia for drive in range(0, self.max_drives): 169a26ddb43SAlberto Garcia # Pick each out of all possible params and test 170a26ddb43SAlberto Garcia for tk in params: 171a26ddb43SAlberto Garcia limits = dict([(k, 0) for k in params]) 172a26ddb43SAlberto Garcia limits[tk] = params[tk] * self.max_drives 173a26ddb43SAlberto Garcia self.configure_throttle(self.max_drives, limits) 174a26ddb43SAlberto Garcia self.do_test_throttle(1, 5, limits, drive) 175a26ddb43SAlberto Garcia 176a90cade0SAlberto Garcia def test_burst(self): 177a90cade0SAlberto Garcia params = {"bps": 4096, 178a90cade0SAlberto Garcia "bps_rd": 4096, 179a90cade0SAlberto Garcia "bps_wr": 4096, 180a90cade0SAlberto Garcia "iops": 10, 181a90cade0SAlberto Garcia "iops_rd": 10, 182a90cade0SAlberto Garcia "iops_wr": 10, 183a90cade0SAlberto Garcia } 184a90cade0SAlberto Garcia ndrives = 1 185a90cade0SAlberto Garcia # Pick each out of all possible params and test 186a90cade0SAlberto Garcia for tk in params: 187a90cade0SAlberto Garcia rate = params[tk] * ndrives 188a90cade0SAlberto Garcia burst_rate = rate * 7 189a90cade0SAlberto Garcia burst_length = 4 190a90cade0SAlberto Garcia 191a90cade0SAlberto Garcia # Configure the throttling settings 192a90cade0SAlberto Garcia settings = dict([(k, 0) for k in params]) 193a90cade0SAlberto Garcia settings[tk] = rate 194a90cade0SAlberto Garcia settings['%s_max' % tk] = burst_rate 195a90cade0SAlberto Garcia settings['%s_max_length' % tk] = burst_length 196a90cade0SAlberto Garcia self.configure_throttle(ndrives, settings) 197a90cade0SAlberto Garcia 198a90cade0SAlberto Garcia # Wait for the bucket to empty so we can do bursts 1999a3a9a63SMax Reitz wait_ns = nsec_per_sec * burst_length * burst_rate // rate 200a90cade0SAlberto Garcia self.vm.qtest("clock_step %d" % wait_ns) 201a90cade0SAlberto Garcia 202a90cade0SAlberto Garcia # Test I/O at the max burst rate 203a90cade0SAlberto Garcia limits = dict([(k, 0) for k in params]) 204a90cade0SAlberto Garcia limits[tk] = burst_rate 205a90cade0SAlberto Garcia self.do_test_throttle(ndrives, burst_length, limits) 206a90cade0SAlberto Garcia 207a90cade0SAlberto Garcia # Now test I/O at the normal rate 208a90cade0SAlberto Garcia limits[tk] = rate 2092db33f88SAlberto Garcia self.do_test_throttle(ndrives, 5, limits) 210fb13bbf2SFam Zheng 211ef7a6a3cSAlberto Garcia # Test that removing a drive from a throttle group should not 212ef7a6a3cSAlberto Garcia # affect the remaining members of the group. 213ef7a6a3cSAlberto Garcia # https://bugzilla.redhat.com/show_bug.cgi?id=1535914 214ef7a6a3cSAlberto Garcia def test_remove_group_member(self): 215ef7a6a3cSAlberto Garcia # Create a throttle group with two drives 216ef7a6a3cSAlberto Garcia # and set a 4 KB/s read limit. 217ef7a6a3cSAlberto Garcia params = {"bps": 0, 218ef7a6a3cSAlberto Garcia "bps_rd": 4096, 219ef7a6a3cSAlberto Garcia "bps_wr": 0, 220ef7a6a3cSAlberto Garcia "iops": 0, 221ef7a6a3cSAlberto Garcia "iops_rd": 0, 222ef7a6a3cSAlberto Garcia "iops_wr": 0 } 223ef7a6a3cSAlberto Garcia self.configure_throttle(2, params) 224ef7a6a3cSAlberto Garcia 225ef7a6a3cSAlberto Garcia # Read 4KB from drive0. This is performed immediately. 226ef7a6a3cSAlberto Garcia self.vm.hmp_qemu_io("drive0", "aio_read 0 4096") 227ef7a6a3cSAlberto Garcia 2283db3e9c6SAlberto Garcia # Read 2KB. The I/O limit has been exceeded so this 229ef7a6a3cSAlberto Garcia # request is throttled and a timer is set to wake it up. 2303db3e9c6SAlberto Garcia self.vm.hmp_qemu_io("drive0", "aio_read 0 2048") 231ef7a6a3cSAlberto Garcia 2323db3e9c6SAlberto Garcia # Read 2KB again. We're still over the I/O limit so this is 2333db3e9c6SAlberto Garcia # request is also throttled, but no new timer is set since 2343db3e9c6SAlberto Garcia # there's already one. 2353db3e9c6SAlberto Garcia self.vm.hmp_qemu_io("drive0", "aio_read 0 2048") 2363db3e9c6SAlberto Garcia 2373db3e9c6SAlberto Garcia # Read from drive1. This request is also throttled, and no 2383db3e9c6SAlberto Garcia # timer is set in drive1 because there's already one in 2393db3e9c6SAlberto Garcia # drive0. 240ef7a6a3cSAlberto Garcia self.vm.hmp_qemu_io("drive1", "aio_read 0 4096") 241ef7a6a3cSAlberto Garcia 242ef7a6a3cSAlberto Garcia # At this point only the first 4KB have been read from drive0. 243ef7a6a3cSAlberto Garcia # The other requests are throttled. 244ef7a6a3cSAlberto Garcia self.assertEqual(self.blockstats('drive0')[0], 4096) 245ef7a6a3cSAlberto Garcia self.assertEqual(self.blockstats('drive1')[0], 0) 246ef7a6a3cSAlberto Garcia 247ef7a6a3cSAlberto Garcia # Remove drive0 from the throttle group and disable its I/O limits. 248ef7a6a3cSAlberto Garcia # drive1 remains in the group with a throttled request. 249ef7a6a3cSAlberto Garcia params['bps_rd'] = 0 250ef7a6a3cSAlberto Garcia params['device'] = 'drive0' 251ef7a6a3cSAlberto Garcia result = self.vm.qmp("block_set_io_throttle", conv_keys=False, **params) 252ef7a6a3cSAlberto Garcia self.assert_qmp(result, 'return', {}) 253ef7a6a3cSAlberto Garcia 2543db3e9c6SAlberto Garcia # Removing the I/O limits from drive0 drains its two pending requests. 255ef7a6a3cSAlberto Garcia # The read request in drive1 is still throttled. 256ef7a6a3cSAlberto Garcia self.assertEqual(self.blockstats('drive0')[0], 8192) 257ef7a6a3cSAlberto Garcia self.assertEqual(self.blockstats('drive1')[0], 0) 258ef7a6a3cSAlberto Garcia 259ef7a6a3cSAlberto Garcia # Advance the clock 5 seconds. This completes the request in drive1 260ef7a6a3cSAlberto Garcia self.vm.qtest("clock_step %d" % (5 * nsec_per_sec)) 261ef7a6a3cSAlberto Garcia 262ef7a6a3cSAlberto Garcia # Now all requests have been processed. 263ef7a6a3cSAlberto Garcia self.assertEqual(self.blockstats('drive0')[0], 8192) 264ef7a6a3cSAlberto Garcia self.assertEqual(self.blockstats('drive1')[0], 4096) 265ef7a6a3cSAlberto Garcia 266fb13bbf2SFam Zhengclass ThrottleTestCoroutine(ThrottleTestCase): 267fb13bbf2SFam Zheng test_img = "null-co://" 268fb13bbf2SFam Zheng 269435d5ee6SAlberto Garciaclass ThrottleTestGroupNames(iotests.QMPTestCase): 270435d5ee6SAlberto Garcia test_img = "null-aio://" 271435d5ee6SAlberto Garcia max_drives = 3 272435d5ee6SAlberto Garcia 273435d5ee6SAlberto Garcia def setUp(self): 274435d5ee6SAlberto Garcia self.vm = iotests.VM() 275435d5ee6SAlberto Garcia for i in range(0, self.max_drives): 276*a6862418SAndrey Shinkevich self.vm.add_drive(self.test_img, 277*a6862418SAndrey Shinkevich "throttling.iops-total=100,file.read-zeroes=on") 278435d5ee6SAlberto Garcia self.vm.launch() 279435d5ee6SAlberto Garcia 280435d5ee6SAlberto Garcia def tearDown(self): 281435d5ee6SAlberto Garcia self.vm.shutdown() 282435d5ee6SAlberto Garcia 283435d5ee6SAlberto Garcia def set_io_throttle(self, device, params): 284435d5ee6SAlberto Garcia params["device"] = device 285435d5ee6SAlberto Garcia result = self.vm.qmp("block_set_io_throttle", conv_keys=False, **params) 286435d5ee6SAlberto Garcia self.assert_qmp(result, 'return', {}) 287435d5ee6SAlberto Garcia 288435d5ee6SAlberto Garcia def verify_name(self, device, name): 289435d5ee6SAlberto Garcia result = self.vm.qmp("query-block") 290435d5ee6SAlberto Garcia for r in result["return"]: 291435d5ee6SAlberto Garcia if r["device"] == device: 292435d5ee6SAlberto Garcia info = r["inserted"] 293435d5ee6SAlberto Garcia if name: 294435d5ee6SAlberto Garcia self.assertEqual(info["group"], name) 295435d5ee6SAlberto Garcia else: 296d7a4228eSEduardo Habkost self.assertFalse('group' in info) 297435d5ee6SAlberto Garcia return 298435d5ee6SAlberto Garcia 299435d5ee6SAlberto Garcia raise Exception("No group information found for '%s'" % device) 300435d5ee6SAlberto Garcia 301435d5ee6SAlberto Garcia def test_group_naming(self): 302435d5ee6SAlberto Garcia params = {"bps": 0, 303435d5ee6SAlberto Garcia "bps_rd": 0, 304435d5ee6SAlberto Garcia "bps_wr": 0, 305435d5ee6SAlberto Garcia "iops": 0, 306435d5ee6SAlberto Garcia "iops_rd": 0, 307435d5ee6SAlberto Garcia "iops_wr": 0} 308435d5ee6SAlberto Garcia 309435d5ee6SAlberto Garcia # Check the drives added using the command line. 310435d5ee6SAlberto Garcia # The default throttling group name is the device name. 311435d5ee6SAlberto Garcia for i in range(self.max_drives): 312435d5ee6SAlberto Garcia devname = "drive%d" % i 313435d5ee6SAlberto Garcia self.verify_name(devname, devname) 314435d5ee6SAlberto Garcia 315435d5ee6SAlberto Garcia # Clear throttling settings => the group name is gone. 316435d5ee6SAlberto Garcia for i in range(self.max_drives): 317435d5ee6SAlberto Garcia devname = "drive%d" % i 318435d5ee6SAlberto Garcia self.set_io_throttle(devname, params) 319435d5ee6SAlberto Garcia self.verify_name(devname, None) 320435d5ee6SAlberto Garcia 321435d5ee6SAlberto Garcia # Set throttling settings using block_set_io_throttle and 322435d5ee6SAlberto Garcia # check the default group names. 323435d5ee6SAlberto Garcia params["iops"] = 10 324435d5ee6SAlberto Garcia for i in range(self.max_drives): 325435d5ee6SAlberto Garcia devname = "drive%d" % i 326435d5ee6SAlberto Garcia self.set_io_throttle(devname, params) 327435d5ee6SAlberto Garcia self.verify_name(devname, devname) 328435d5ee6SAlberto Garcia 329435d5ee6SAlberto Garcia # Set a custom group name for each device 330435d5ee6SAlberto Garcia for i in range(3): 331435d5ee6SAlberto Garcia devname = "drive%d" % i 332435d5ee6SAlberto Garcia groupname = "group%d" % i 333435d5ee6SAlberto Garcia params['group'] = groupname 334435d5ee6SAlberto Garcia self.set_io_throttle(devname, params) 335435d5ee6SAlberto Garcia self.verify_name(devname, groupname) 336435d5ee6SAlberto Garcia 337435d5ee6SAlberto Garcia # Put drive0 in group1 and check that all other devices remain 338435d5ee6SAlberto Garcia # unchanged 339435d5ee6SAlberto Garcia params['group'] = 'group1' 340435d5ee6SAlberto Garcia self.set_io_throttle('drive0', params) 341435d5ee6SAlberto Garcia self.verify_name('drive0', 'group1') 342435d5ee6SAlberto Garcia for i in range(1, self.max_drives): 343435d5ee6SAlberto Garcia devname = "drive%d" % i 344435d5ee6SAlberto Garcia groupname = "group%d" % i 345435d5ee6SAlberto Garcia self.verify_name(devname, groupname) 346435d5ee6SAlberto Garcia 347435d5ee6SAlberto Garcia # Put drive0 in group2 and check that all other devices remain 348435d5ee6SAlberto Garcia # unchanged 349435d5ee6SAlberto Garcia params['group'] = 'group2' 350435d5ee6SAlberto Garcia self.set_io_throttle('drive0', params) 351435d5ee6SAlberto Garcia self.verify_name('drive0', 'group2') 352435d5ee6SAlberto Garcia for i in range(1, self.max_drives): 353435d5ee6SAlberto Garcia devname = "drive%d" % i 354435d5ee6SAlberto Garcia groupname = "group%d" % i 355435d5ee6SAlberto Garcia self.verify_name(devname, groupname) 356435d5ee6SAlberto Garcia 357435d5ee6SAlberto Garcia # Clear throttling settings from drive0 check that all other 358435d5ee6SAlberto Garcia # devices remain unchanged 359435d5ee6SAlberto Garcia params["iops"] = 0 360435d5ee6SAlberto Garcia self.set_io_throttle('drive0', params) 361435d5ee6SAlberto Garcia self.verify_name('drive0', None) 362435d5ee6SAlberto Garcia for i in range(1, self.max_drives): 363435d5ee6SAlberto Garcia devname = "drive%d" % i 364435d5ee6SAlberto Garcia groupname = "group%d" % i 365435d5ee6SAlberto Garcia self.verify_name(devname, groupname) 366435d5ee6SAlberto Garcia 36707615626SAlberto Garciaclass ThrottleTestRemovableMedia(iotests.QMPTestCase): 36807615626SAlberto Garcia def setUp(self): 36907615626SAlberto Garcia self.vm = iotests.VM() 37007615626SAlberto Garcia if iotests.qemu_default_machine == 's390-ccw-virtio': 37107615626SAlberto Garcia self.vm.add_device("virtio-scsi-ccw,id=virtio-scsi") 37207615626SAlberto Garcia else: 37307615626SAlberto Garcia self.vm.add_device("virtio-scsi-pci,id=virtio-scsi") 37407615626SAlberto Garcia self.vm.launch() 37507615626SAlberto Garcia 37607615626SAlberto Garcia def tearDown(self): 37707615626SAlberto Garcia self.vm.shutdown() 37807615626SAlberto Garcia 37907615626SAlberto Garcia def test_removable_media(self): 38007615626SAlberto Garcia # Add a couple of dummy nodes named cd0 and cd1 38107615626SAlberto Garcia result = self.vm.qmp("blockdev-add", driver="null-aio", 382*a6862418SAndrey Shinkevich read_zeroes=True, node_name="cd0") 38307615626SAlberto Garcia self.assert_qmp(result, 'return', {}) 38407615626SAlberto Garcia result = self.vm.qmp("blockdev-add", driver="null-aio", 385*a6862418SAndrey Shinkevich read_zeroes=True, node_name="cd1") 38607615626SAlberto Garcia self.assert_qmp(result, 'return', {}) 38707615626SAlberto Garcia 38807615626SAlberto Garcia # Attach a CD drive with cd0 inserted 38907615626SAlberto Garcia result = self.vm.qmp("device_add", driver="scsi-cd", 39007615626SAlberto Garcia id="dev0", drive="cd0") 39107615626SAlberto Garcia self.assert_qmp(result, 'return', {}) 39207615626SAlberto Garcia 39307615626SAlberto Garcia # Set I/O limits 39407615626SAlberto Garcia args = { "id": "dev0", "iops": 100, "iops_rd": 0, "iops_wr": 0, 39507615626SAlberto Garcia "bps": 50, "bps_rd": 0, "bps_wr": 0 } 39607615626SAlberto Garcia result = self.vm.qmp("block_set_io_throttle", conv_keys=False, **args) 39707615626SAlberto Garcia self.assert_qmp(result, 'return', {}) 39807615626SAlberto Garcia 39907615626SAlberto Garcia # Check that the I/O limits have been set 40007615626SAlberto Garcia result = self.vm.qmp("query-block") 40107615626SAlberto Garcia self.assert_qmp(result, 'return[0]/inserted/iops', 100) 40207615626SAlberto Garcia self.assert_qmp(result, 'return[0]/inserted/bps', 50) 40307615626SAlberto Garcia 40407615626SAlberto Garcia # Now eject cd0 and insert cd1 40507615626SAlberto Garcia result = self.vm.qmp("blockdev-open-tray", id='dev0') 40607615626SAlberto Garcia self.assert_qmp(result, 'return', {}) 40734ce1111SMax Reitz result = self.vm.qmp("blockdev-remove-medium", id='dev0') 40807615626SAlberto Garcia self.assert_qmp(result, 'return', {}) 40934ce1111SMax Reitz result = self.vm.qmp("blockdev-insert-medium", id='dev0', node_name='cd1') 41007615626SAlberto Garcia self.assert_qmp(result, 'return', {}) 41107615626SAlberto Garcia 41207615626SAlberto Garcia # Check that the I/O limits are still the same 41307615626SAlberto Garcia result = self.vm.qmp("query-block") 41407615626SAlberto Garcia self.assert_qmp(result, 'return[0]/inserted/iops', 100) 41507615626SAlberto Garcia self.assert_qmp(result, 'return[0]/inserted/bps', 50) 41607615626SAlberto Garcia 41707615626SAlberto Garcia # Eject cd1 41834ce1111SMax Reitz result = self.vm.qmp("blockdev-remove-medium", id='dev0') 41907615626SAlberto Garcia self.assert_qmp(result, 'return', {}) 42007615626SAlberto Garcia 42107615626SAlberto Garcia # Check that we can't set limits if the device has no medium 42207615626SAlberto Garcia result = self.vm.qmp("block_set_io_throttle", conv_keys=False, **args) 42307615626SAlberto Garcia self.assert_qmp(result, 'error/class', 'GenericError') 42407615626SAlberto Garcia 42507615626SAlberto Garcia # Remove the CD drive 42607615626SAlberto Garcia result = self.vm.qmp("device_del", id='dev0') 42707615626SAlberto Garcia self.assert_qmp(result, 'return', {}) 42807615626SAlberto Garcia 429435d5ee6SAlberto Garcia 430fb13bbf2SFam Zhengif __name__ == '__main__': 431fb13bbf2SFam Zheng iotests.main(supported_fmts=["raw"]) 432