def attachDeviceFlags(dom, xml, flags=0): myxml = objectify.fromstring(xml) if myxml.tag == 'memory': unit = myxml.target.size.get('unit') myxml.target.size.set('unit', 'KiB') myxml.target.size._setText( str(int(convert_data_size(myxml.target.size.text, unit, 'KiB')))) xml = ET.tostring(myxml).decode('utf-8') dom.setMaxMemory(int(dom.maxMemory() + myxml.target.size)) MockModel._mock_vms[dom.name()].append(xml)
def attachDeviceFlags(dom, xml, flags=0): myxml = objectify.fromstring(xml) if myxml.tag == 'memory': unit = myxml.target.size.get('unit') myxml.target.size.set('unit', 'KiB') myxml.target.size._setText( str(int(convert_data_size(myxml.target.size.text, unit, 'KiB'))) ) xml = ET.tostring(myxml).decode('utf-8') dom.setMaxMemory(int(dom.maxMemory() + myxml.target.size)) MockModel._mock_vms[dom.name()].append(xml)
def test_convert_data_size(self): failure_data = [ {'val': None, 'from': 'MiB'}, {'val': self, 'from': 'MiB'}, {'val': 1, 'from': None}, {'val': 1, 'from': ''}, {'val': 1, 'from': 'foo'}, {'val': 1, 'from': 'kib'}, {'val': 1, 'from': 'MiB', 'to': None}, {'val': 1, 'from': 'MiB', 'to': ''}, {'val': 1, 'from': 'MiB', 'to': 'foo'}, {'val': 1, 'from': 'MiB', 'to': 'kib'}, ] for d in failure_data: if 'to' in d: self.assertRaises( InvalidParameter, convert_data_size, d['val'], d['from'], d['to'] ) else: self.assertRaises( InvalidParameter, convert_data_size, d['val'], d['from'] ) success_data = [ {'got': convert_data_size(5, 'MiB', 'MiB'), 'want': 5}, {'got': convert_data_size(5, 'MiB', 'KiB'), 'want': 5120}, {'got': convert_data_size(5, 'MiB', 'M'), 'want': 5.24288}, {'got': convert_data_size(5, 'MiB', 'GiB'), 'want': 0.0048828125}, {'got': convert_data_size(5, 'MiB', 'Tb'), 'want': 4.194304e-05}, {'got': convert_data_size(5, 'KiB', 'MiB'), 'want': 0.0048828125}, {'got': convert_data_size(5, 'M', 'MiB'), 'want': 4.76837158203125}, {'got': convert_data_size(5, 'GiB', 'MiB'), 'want': 5120}, {'got': convert_data_size(5, 'Tb', 'MiB'), 'want': 596046.4477539062}, { 'got': convert_data_size(5, 'MiB'), 'want': convert_data_size(5, 'MiB', 'B'), }, ] for d in success_data: self.assertEqual(d['got'], d['want'])
def test_convert_data_size(self): failure_data = [{'val': None, 'from': 'MiB'}, {'val': self, 'from': 'MiB'}, {'val': 1, 'from': None}, {'val': 1, 'from': ''}, {'val': 1, 'from': 'foo'}, {'val': 1, 'from': 'kib'}, {'val': 1, 'from': 'MiB', 'to': None}, {'val': 1, 'from': 'MiB', 'to': ''}, {'val': 1, 'from': 'MiB', 'to': 'foo'}, {'val': 1, 'from': 'MiB', 'to': 'kib'}] for d in failure_data: if 'to' in d: self.assertRaises(InvalidParameter, convert_data_size, d['val'], d['from'], d['to']) else: self.assertRaises(InvalidParameter, convert_data_size, d['val'], d['from']) success_data = [{'got': convert_data_size(5, 'MiB', 'MiB'), 'want': 5}, {'got': convert_data_size(5, 'MiB', 'KiB'), 'want': 5120}, {'got': convert_data_size(5, 'MiB', 'M'), 'want': 5.24288}, {'got': convert_data_size(5, 'MiB', 'GiB'), 'want': 0.0048828125}, {'got': convert_data_size(5, 'MiB', 'Tb'), 'want': 4.194304e-05}, {'got': convert_data_size(5, 'KiB', 'MiB'), 'want': 0.0048828125}, {'got': convert_data_size(5, 'M', 'MiB'), 'want': 4.76837158203125}, {'got': convert_data_size(5, 'GiB', 'MiB'), 'want': 5120}, {'got': convert_data_size(5, 'Tb', 'MiB'), 'want': 596046.4477539062}, {'got': convert_data_size(5, 'MiB'), 'want': convert_data_size(5, 'MiB', 'B')}] for d in success_data: self.assertEquals(d['got'], d['want'])