Quelle test_base_diskutils.py Sprache: Python

import unittest
from unittest import mock

import mozunit
from mozharness.base.diskutils import DiskInfo, DiskSize, DiskutilsError, convert_to

class TestDiskutils(unittest.TestCase):
    def test_convert_to(self):
        # 0 is 0 regardless from_unit/to_unit
        self.assertTrue(convert_to(size=0, from_unit="GB", to_unit="MB") == 0)
        size = 524288  # 512 * 1024
        # converting from/to same unit
        self.assertTrue(convert_to(size=size, from_unit="MB", to_unit="MB") == size)

        self.assertTrue(convert_to(size=size, from_unit="MB", to_unit="GB") == 512)

        self.assertRaises(
            DiskutilsError,
            lambda: convert_to(size="a string", from_unit="MB", to_unit="MB"),
        )
        self.assertRaises(
            DiskutilsError, lambda: convert_to(size=0, from_unit="foo", to_unit="MB")
        )
        self.assertRaises(
            DiskutilsError, lambda: convert_to(size=0, from_unit="MB", to_unit="foo")
        )

class TestDiskInfo(unittest.TestCase):
    def testDiskinfo_to(self):
        di = DiskInfo()
        self.assertTrue(di.unit == "bytes")
        self.assertTrue(di.free == 0)
        self.assertTrue(di.used == 0)
        self.assertTrue(di.total == 0)
        # convert to GB
        di._to("GB")
        self.assertTrue(di.unit == "GB")
        self.assertTrue(di.free == 0)
        self.assertTrue(di.used == 0)
        self.assertTrue(di.total == 0)

class MockStatvfs(object):
    def __init__(self):
        self.f_bsize = 0
        self.f_frsize = 0
        self.f_blocks = 0
        self.f_bfree = 0
        self.f_bavail = 0
        self.f_files = 0
        self.f_ffree = 0
        self.f_favail = 0
        self.f_flag = 0
        self.f_namemax = 0

class TestDiskSpace(unittest.TestCase):
    @mock.patch("mozharness.base.diskutils.os")
    def testDiskSpacePosix(self, mock_os):
        ds = MockStatvfs()
        mock_os.statvfs.return_value = ds
        di = DiskSize()._posix_size("/")
        self.assertTrue(di.unit == "bytes")
        self.assertTrue(di.free == 0)
        self.assertTrue(di.used == 0)
        self.assertTrue(di.total == 0)

    @mock.patch("mozharness.base.diskutils.ctypes")
    def testDiskSpaceWindows(self, mock_ctypes):
        mock_ctypes.windll.kernel32.GetDiskFreeSpaceExA.return_value = 0
        mock_ctypes.windll.kernel32.GetDiskFreeSpaceExW.return_value = 0
        di = DiskSize()._windows_size("/c/")
        self.assertTrue(di.unit == "bytes")
        self.assertTrue(di.free == 0)
        self.assertTrue(di.used == 0)
        self.assertTrue(di.total == 0)

    @mock.patch("mozharness.base.diskutils.os")
    @mock.patch("mozharness.base.diskutils.ctypes")
    def testUnspportedPlafrom(self, mock_ctypes, mock_os):
        mock_os.statvfs.side_effect = AttributeError("")
        self.assertRaises(AttributeError, lambda: DiskSize()._posix_size("/"))
        mock_ctypes.windll.kernel32.GetDiskFreeSpaceExW.side_effect = AttributeError("")
        mock_ctypes.windll.kernel32.GetDiskFreeSpaceExA.side_effect = AttributeError("")
        self.assertRaises(AttributeError, lambda: DiskSize()._windows_size("/"))
        self.assertRaises(
            DiskutilsError, lambda: DiskSize().get_size(path="/", unit="GB")
        )

if __name__ == "__main__":
    mozunit.main()

quality90%

¤ Dauer der Verarbeitung: 0.11 Sekunden (vorverarbeitet) ¤

Wurzel

Suchen

Beweissystem der NASA

Beweissystem Isabelle

NIST Cobol Testsuite

Cephes Mathematical Library

Wiener Entwicklungsmethode

Haftungshinweis

Die Informationen auf dieser Webseite wurden nach bestem Wissen sorgfältig zusammengestellt. Es wird jedoch weder Vollständigkeit, noch Richtigkeit, noch Qualität der bereit gestellten Informationen zugesichert.

Bemerkung:

Die farbliche Syntaxdarstellung ist noch experimentell.