get_virt_memory_opt: support str values like "16GB"

This commit is contained in:
Hans-Christoph Steiner 2025-10-24 12:30:03 +02:00
parent 2386bcc64f
commit 8421d61369
2 changed files with 34 additions and 7 deletions

View file

@ -142,3 +142,25 @@ class Up_run_vagrant(UpTest):
up.run_vagrant(APPID, VERCODE, 1, 1)
vagrant_destroy.assert_called_once()
self.assertNotEqual(ctime, os.path.getctime(vagrantfile))
class Up_options(UpTest):
def test_get_virt_memory_opt_default(self):
self.assertEqual(up.get_virt_memory_opt(None), 6 * 1024**3)
def test_get_virt_memory_opt_int(self):
testvalue = 1234567890
self.assertEqual(up.get_virt_memory_opt(testvalue), testvalue)
def test_get_virt_memory_opt_str_int(self):
testvalue = 1234567890
self.assertEqual(up.get_virt_memory_opt(str(testvalue)), testvalue)
def test_get_virt_memory_opt_str_upper(self):
self.assertEqual(up.get_virt_memory_opt('1GB'), 1024**3)
def test_get_virt_memory_opt_str_lower(self):
self.assertEqual(up.get_virt_memory_opt('1tib'), 1024**4)
def test_get_virt_memory_opt_str_mixed(self):
self.assertEqual(up.get_virt_memory_opt('1MiB'), 1024**2)