From 2567ef297cd4e2c0d12c4129217e16d95551ec82 Mon Sep 17 00:00:00 2001 From: Julio-Oliveira-Encora Date: Wed, 1 May 2024 09:17:37 -0300 Subject: [PATCH] Added test for convert_byte_size. --- netbox/netbox/tests/test_convert_byte_size.py | 25 +++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100644 netbox/netbox/tests/test_convert_byte_size.py diff --git a/netbox/netbox/tests/test_convert_byte_size.py b/netbox/netbox/tests/test_convert_byte_size.py new file mode 100644 index 000000000..f5d8b4a55 --- /dev/null +++ b/netbox/netbox/tests/test_convert_byte_size.py @@ -0,0 +1,25 @@ +from netbox.utils import convert_byte_size +from utilities.testing import TestCase + + +class TestConvertByteSize(TestCase): + def test_convert_byte_size_returns_original_value_for_small_numbers(self): + self.assertEqual(convert_byte_size(500, "kilo"), 500) + + def test_convert_byte_size_converts_kilobytes_to_bytes(self): + self.assertEqual(convert_byte_size(1024, "kilo"), 1024) + + def test_convert_byte_size_converts_megabytes_to_bytes(self): + self.assertEqual(convert_byte_size(1048576, "mega"), 1) + + def test_convert_byte_size_converts_gigabytes_to_bytes(self): + self.assertEqual(convert_byte_size(1073741824, "giga"), 1) + + def test_convert_byte_size_converts_terabytes_to_bytes(self): + self.assertEqual(convert_byte_size(1099511627776, "tera"), 1) + + def test_convert_byte_size_returns_zero_for_none(self): + self.assertEqual(convert_byte_size(None, "mega"), 0) + + def test_convert_byte_size_without_unit(self): + self.assertEqual(round(convert_byte_size(123456789), 2), 117.74)