xref: /openbmc/qemu/tests/qemu-iotests/093 (revision ef7a6a3c2a7725b169d054aa7487f9738bd6c4a6)
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):
412db33f88SAlberto Garcia            self.vm.add_drive(self.test_img)
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
72fb13bbf2SFam Zheng        rd_nr = max(params['bps'] / rq_size / 2,
73fb13bbf2SFam Zheng                    params['bps_rd'] / rq_size,
74fb13bbf2SFam Zheng                    params['iops'] / 2,
75fb13bbf2SFam Zheng                    params['iops_rd'])
76fb13bbf2SFam Zheng        rd_nr *= seconds * 2
772db33f88SAlberto Garcia        rd_nr /= ndrives
78fb13bbf2SFam Zheng        wr_nr = max(params['bps'] / rq_size / 2,
79fb13bbf2SFam Zheng                    params['bps_wr'] / rq_size,
80fb13bbf2SFam Zheng                    params['iops'] / 2,
81fb13bbf2SFam Zheng                    params['iops_wr'])
82fb13bbf2SFam Zheng        wr_nr *= seconds * 2
832db33f88SAlberto Garcia        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
199a90cade0SAlberto Garcia            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
211*ef7a6a3cSAlberto Garcia    # Test that removing a drive from a throttle group should not
212*ef7a6a3cSAlberto Garcia    # affect the remaining members of the group.
213*ef7a6a3cSAlberto Garcia    # https://bugzilla.redhat.com/show_bug.cgi?id=1535914
214*ef7a6a3cSAlberto Garcia    def test_remove_group_member(self):
215*ef7a6a3cSAlberto Garcia        # Create a throttle group with two drives
216*ef7a6a3cSAlberto Garcia        # and set a 4 KB/s read limit.
217*ef7a6a3cSAlberto Garcia        params = {"bps": 0,
218*ef7a6a3cSAlberto Garcia                  "bps_rd": 4096,
219*ef7a6a3cSAlberto Garcia                  "bps_wr": 0,
220*ef7a6a3cSAlberto Garcia                  "iops": 0,
221*ef7a6a3cSAlberto Garcia                  "iops_rd": 0,
222*ef7a6a3cSAlberto Garcia                  "iops_wr": 0 }
223*ef7a6a3cSAlberto Garcia        self.configure_throttle(2, params)
224*ef7a6a3cSAlberto Garcia
225*ef7a6a3cSAlberto Garcia        # Read 4KB from drive0. This is performed immediately.
226*ef7a6a3cSAlberto Garcia        self.vm.hmp_qemu_io("drive0", "aio_read 0 4096")
227*ef7a6a3cSAlberto Garcia
228*ef7a6a3cSAlberto Garcia        # Read 4KB again. The I/O limit has been exceeded so this
229*ef7a6a3cSAlberto Garcia        # request is throttled and a timer is set to wake it up.
230*ef7a6a3cSAlberto Garcia        self.vm.hmp_qemu_io("drive0", "aio_read 0 4096")
231*ef7a6a3cSAlberto Garcia
232*ef7a6a3cSAlberto Garcia        # Read from drive1. We're still over the I/O limit so this
233*ef7a6a3cSAlberto Garcia        # request is also throttled. There's no timer set in drive1
234*ef7a6a3cSAlberto Garcia        # because there's already one in drive0. Once the timer in
235*ef7a6a3cSAlberto Garcia        # drive0 fires and its throttled request is processed then the
236*ef7a6a3cSAlberto Garcia        # next request in the queue will be scheduled: this one.
237*ef7a6a3cSAlberto Garcia        self.vm.hmp_qemu_io("drive1", "aio_read 0 4096")
238*ef7a6a3cSAlberto Garcia
239*ef7a6a3cSAlberto Garcia        # At this point only the first 4KB have been read from drive0.
240*ef7a6a3cSAlberto Garcia        # The other requests are throttled.
241*ef7a6a3cSAlberto Garcia        self.assertEqual(self.blockstats('drive0')[0], 4096)
242*ef7a6a3cSAlberto Garcia        self.assertEqual(self.blockstats('drive1')[0], 0)
243*ef7a6a3cSAlberto Garcia
244*ef7a6a3cSAlberto Garcia        # Remove drive0 from the throttle group and disable its I/O limits.
245*ef7a6a3cSAlberto Garcia        # drive1 remains in the group with a throttled request.
246*ef7a6a3cSAlberto Garcia        params['bps_rd'] = 0
247*ef7a6a3cSAlberto Garcia        params['device'] = 'drive0'
248*ef7a6a3cSAlberto Garcia        result = self.vm.qmp("block_set_io_throttle", conv_keys=False, **params)
249*ef7a6a3cSAlberto Garcia        self.assert_qmp(result, 'return', {})
250*ef7a6a3cSAlberto Garcia
251*ef7a6a3cSAlberto Garcia        # Removing the I/O limits from drive0 drains its pending request.
252*ef7a6a3cSAlberto Garcia        # The read request in drive1 is still throttled.
253*ef7a6a3cSAlberto Garcia        self.assertEqual(self.blockstats('drive0')[0], 8192)
254*ef7a6a3cSAlberto Garcia        self.assertEqual(self.blockstats('drive1')[0], 0)
255*ef7a6a3cSAlberto Garcia
256*ef7a6a3cSAlberto Garcia        # Advance the clock 5 seconds. This completes the request in drive1
257*ef7a6a3cSAlberto Garcia        self.vm.qtest("clock_step %d" % (5 * nsec_per_sec))
258*ef7a6a3cSAlberto Garcia
259*ef7a6a3cSAlberto Garcia        # Now all requests have been processed.
260*ef7a6a3cSAlberto Garcia        self.assertEqual(self.blockstats('drive0')[0], 8192)
261*ef7a6a3cSAlberto Garcia        self.assertEqual(self.blockstats('drive1')[0], 4096)
262*ef7a6a3cSAlberto Garcia
263fb13bbf2SFam Zhengclass ThrottleTestCoroutine(ThrottleTestCase):
264fb13bbf2SFam Zheng    test_img = "null-co://"
265fb13bbf2SFam Zheng
266435d5ee6SAlberto Garciaclass ThrottleTestGroupNames(iotests.QMPTestCase):
267435d5ee6SAlberto Garcia    test_img = "null-aio://"
268435d5ee6SAlberto Garcia    max_drives = 3
269435d5ee6SAlberto Garcia
270435d5ee6SAlberto Garcia    def setUp(self):
271435d5ee6SAlberto Garcia        self.vm = iotests.VM()
272435d5ee6SAlberto Garcia        for i in range(0, self.max_drives):
273435d5ee6SAlberto Garcia            self.vm.add_drive(self.test_img, "throttling.iops-total=100")
274435d5ee6SAlberto Garcia        self.vm.launch()
275435d5ee6SAlberto Garcia
276435d5ee6SAlberto Garcia    def tearDown(self):
277435d5ee6SAlberto Garcia        self.vm.shutdown()
278435d5ee6SAlberto Garcia
279435d5ee6SAlberto Garcia    def set_io_throttle(self, device, params):
280435d5ee6SAlberto Garcia        params["device"] = device
281435d5ee6SAlberto Garcia        result = self.vm.qmp("block_set_io_throttle", conv_keys=False, **params)
282435d5ee6SAlberto Garcia        self.assert_qmp(result, 'return', {})
283435d5ee6SAlberto Garcia
284435d5ee6SAlberto Garcia    def verify_name(self, device, name):
285435d5ee6SAlberto Garcia        result = self.vm.qmp("query-block")
286435d5ee6SAlberto Garcia        for r in result["return"]:
287435d5ee6SAlberto Garcia            if r["device"] == device:
288435d5ee6SAlberto Garcia                info = r["inserted"]
289435d5ee6SAlberto Garcia                if name:
290435d5ee6SAlberto Garcia                    self.assertEqual(info["group"], name)
291435d5ee6SAlberto Garcia                else:
292d7a4228eSEduardo Habkost                    self.assertFalse('group' in info)
293435d5ee6SAlberto Garcia                return
294435d5ee6SAlberto Garcia
295435d5ee6SAlberto Garcia        raise Exception("No group information found for '%s'" % device)
296435d5ee6SAlberto Garcia
297435d5ee6SAlberto Garcia    def test_group_naming(self):
298435d5ee6SAlberto Garcia        params = {"bps": 0,
299435d5ee6SAlberto Garcia                  "bps_rd": 0,
300435d5ee6SAlberto Garcia                  "bps_wr": 0,
301435d5ee6SAlberto Garcia                  "iops": 0,
302435d5ee6SAlberto Garcia                  "iops_rd": 0,
303435d5ee6SAlberto Garcia                  "iops_wr": 0}
304435d5ee6SAlberto Garcia
305435d5ee6SAlberto Garcia        # Check the drives added using the command line.
306435d5ee6SAlberto Garcia        # The default throttling group name is the device name.
307435d5ee6SAlberto Garcia        for i in range(self.max_drives):
308435d5ee6SAlberto Garcia            devname = "drive%d" % i
309435d5ee6SAlberto Garcia            self.verify_name(devname, devname)
310435d5ee6SAlberto Garcia
311435d5ee6SAlberto Garcia        # Clear throttling settings => the group name is gone.
312435d5ee6SAlberto Garcia        for i in range(self.max_drives):
313435d5ee6SAlberto Garcia            devname = "drive%d" % i
314435d5ee6SAlberto Garcia            self.set_io_throttle(devname, params)
315435d5ee6SAlberto Garcia            self.verify_name(devname, None)
316435d5ee6SAlberto Garcia
317435d5ee6SAlberto Garcia        # Set throttling settings using block_set_io_throttle and
318435d5ee6SAlberto Garcia        # check the default group names.
319435d5ee6SAlberto Garcia        params["iops"] = 10
320435d5ee6SAlberto Garcia        for i in range(self.max_drives):
321435d5ee6SAlberto Garcia            devname = "drive%d" % i
322435d5ee6SAlberto Garcia            self.set_io_throttle(devname, params)
323435d5ee6SAlberto Garcia            self.verify_name(devname, devname)
324435d5ee6SAlberto Garcia
325435d5ee6SAlberto Garcia        # Set a custom group name for each device
326435d5ee6SAlberto Garcia        for i in range(3):
327435d5ee6SAlberto Garcia            devname = "drive%d" % i
328435d5ee6SAlberto Garcia            groupname = "group%d" % i
329435d5ee6SAlberto Garcia            params['group'] = groupname
330435d5ee6SAlberto Garcia            self.set_io_throttle(devname, params)
331435d5ee6SAlberto Garcia            self.verify_name(devname, groupname)
332435d5ee6SAlberto Garcia
333435d5ee6SAlberto Garcia        # Put drive0 in group1 and check that all other devices remain
334435d5ee6SAlberto Garcia        # unchanged
335435d5ee6SAlberto Garcia        params['group'] = 'group1'
336435d5ee6SAlberto Garcia        self.set_io_throttle('drive0', params)
337435d5ee6SAlberto Garcia        self.verify_name('drive0', 'group1')
338435d5ee6SAlberto Garcia        for i in range(1, self.max_drives):
339435d5ee6SAlberto Garcia            devname = "drive%d" % i
340435d5ee6SAlberto Garcia            groupname = "group%d" % i
341435d5ee6SAlberto Garcia            self.verify_name(devname, groupname)
342435d5ee6SAlberto Garcia
343435d5ee6SAlberto Garcia        # Put drive0 in group2 and check that all other devices remain
344435d5ee6SAlberto Garcia        # unchanged
345435d5ee6SAlberto Garcia        params['group'] = 'group2'
346435d5ee6SAlberto Garcia        self.set_io_throttle('drive0', params)
347435d5ee6SAlberto Garcia        self.verify_name('drive0', 'group2')
348435d5ee6SAlberto Garcia        for i in range(1, self.max_drives):
349435d5ee6SAlberto Garcia            devname = "drive%d" % i
350435d5ee6SAlberto Garcia            groupname = "group%d" % i
351435d5ee6SAlberto Garcia            self.verify_name(devname, groupname)
352435d5ee6SAlberto Garcia
353435d5ee6SAlberto Garcia        # Clear throttling settings from drive0 check that all other
354435d5ee6SAlberto Garcia        # devices remain unchanged
355435d5ee6SAlberto Garcia        params["iops"] = 0
356435d5ee6SAlberto Garcia        self.set_io_throttle('drive0', params)
357435d5ee6SAlberto Garcia        self.verify_name('drive0', None)
358435d5ee6SAlberto Garcia        for i in range(1, self.max_drives):
359435d5ee6SAlberto Garcia            devname = "drive%d" % i
360435d5ee6SAlberto Garcia            groupname = "group%d" % i
361435d5ee6SAlberto Garcia            self.verify_name(devname, groupname)
362435d5ee6SAlberto Garcia
36307615626SAlberto Garciaclass ThrottleTestRemovableMedia(iotests.QMPTestCase):
36407615626SAlberto Garcia    def setUp(self):
36507615626SAlberto Garcia        self.vm = iotests.VM()
36607615626SAlberto Garcia        if iotests.qemu_default_machine == 's390-ccw-virtio':
36707615626SAlberto Garcia            self.vm.add_device("virtio-scsi-ccw,id=virtio-scsi")
36807615626SAlberto Garcia        else:
36907615626SAlberto Garcia            self.vm.add_device("virtio-scsi-pci,id=virtio-scsi")
37007615626SAlberto Garcia        self.vm.launch()
37107615626SAlberto Garcia
37207615626SAlberto Garcia    def tearDown(self):
37307615626SAlberto Garcia        self.vm.shutdown()
37407615626SAlberto Garcia
37507615626SAlberto Garcia    def test_removable_media(self):
37607615626SAlberto Garcia        # Add a couple of dummy nodes named cd0 and cd1
37707615626SAlberto Garcia        result = self.vm.qmp("blockdev-add", driver="null-aio",
37807615626SAlberto Garcia                             node_name="cd0")
37907615626SAlberto Garcia        self.assert_qmp(result, 'return', {})
38007615626SAlberto Garcia        result = self.vm.qmp("blockdev-add", driver="null-aio",
38107615626SAlberto Garcia                             node_name="cd1")
38207615626SAlberto Garcia        self.assert_qmp(result, 'return', {})
38307615626SAlberto Garcia
38407615626SAlberto Garcia        # Attach a CD drive with cd0 inserted
38507615626SAlberto Garcia        result = self.vm.qmp("device_add", driver="scsi-cd",
38607615626SAlberto Garcia                             id="dev0", drive="cd0")
38707615626SAlberto Garcia        self.assert_qmp(result, 'return', {})
38807615626SAlberto Garcia
38907615626SAlberto Garcia        # Set I/O limits
39007615626SAlberto Garcia        args = { "id": "dev0", "iops": 100, "iops_rd": 0, "iops_wr": 0,
39107615626SAlberto Garcia                                "bps":  50,  "bps_rd": 0,  "bps_wr": 0 }
39207615626SAlberto Garcia        result = self.vm.qmp("block_set_io_throttle", conv_keys=False, **args)
39307615626SAlberto Garcia        self.assert_qmp(result, 'return', {})
39407615626SAlberto Garcia
39507615626SAlberto Garcia        # Check that the I/O limits have been set
39607615626SAlberto Garcia        result = self.vm.qmp("query-block")
39707615626SAlberto Garcia        self.assert_qmp(result, 'return[0]/inserted/iops', 100)
39807615626SAlberto Garcia        self.assert_qmp(result, 'return[0]/inserted/bps',   50)
39907615626SAlberto Garcia
40007615626SAlberto Garcia        # Now eject cd0 and insert cd1
40107615626SAlberto Garcia        result = self.vm.qmp("blockdev-open-tray", id='dev0')
40207615626SAlberto Garcia        self.assert_qmp(result, 'return', {})
40334ce1111SMax Reitz        result = self.vm.qmp("blockdev-remove-medium", id='dev0')
40407615626SAlberto Garcia        self.assert_qmp(result, 'return', {})
40534ce1111SMax Reitz        result = self.vm.qmp("blockdev-insert-medium", id='dev0', node_name='cd1')
40607615626SAlberto Garcia        self.assert_qmp(result, 'return', {})
40707615626SAlberto Garcia
40807615626SAlberto Garcia        # Check that the I/O limits are still the same
40907615626SAlberto Garcia        result = self.vm.qmp("query-block")
41007615626SAlberto Garcia        self.assert_qmp(result, 'return[0]/inserted/iops', 100)
41107615626SAlberto Garcia        self.assert_qmp(result, 'return[0]/inserted/bps',   50)
41207615626SAlberto Garcia
41307615626SAlberto Garcia        # Eject cd1
41434ce1111SMax Reitz        result = self.vm.qmp("blockdev-remove-medium", id='dev0')
41507615626SAlberto Garcia        self.assert_qmp(result, 'return', {})
41607615626SAlberto Garcia
41707615626SAlberto Garcia        # Check that we can't set limits if the device has no medium
41807615626SAlberto Garcia        result = self.vm.qmp("block_set_io_throttle", conv_keys=False, **args)
41907615626SAlberto Garcia        self.assert_qmp(result, 'error/class', 'GenericError')
42007615626SAlberto Garcia
42107615626SAlberto Garcia        # Remove the CD drive
42207615626SAlberto Garcia        result = self.vm.qmp("device_del", id='dev0')
42307615626SAlberto Garcia        self.assert_qmp(result, 'return', {})
42407615626SAlberto Garcia
425435d5ee6SAlberto Garcia
426fb13bbf2SFam Zhengif __name__ == '__main__':
427fb13bbf2SFam Zheng    iotests.main(supported_fmts=["raw"])
428