6203541ac6
If the media is bootable, we can not split it. The limit is not used in that case and we may overflow it with an warning message. The warning should correctly mention what is going on instead of printing a non-sensical message about free space on media being some huge number. Signed-off-by: Lubomír Sedlář <lsedlar@redhat.com>
148 lines
4.8 KiB
Python
148 lines
4.8 KiB
Python
#!/usr/bin/env python2
|
|
# -*- coding: utf-8 -*-
|
|
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest
|
|
import os
|
|
import sys
|
|
import mock
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
|
|
|
|
from pungi import media_split
|
|
|
|
|
|
class ConvertMediaSizeTestCase(unittest.TestCase):
|
|
|
|
def test_size_parser_correct_number_as_int(self):
|
|
self.assertEqual(media_split.convert_media_size(123), 123)
|
|
|
|
def test_size_parser_correct_number_as_str(self):
|
|
self.assertEqual(media_split.convert_media_size('123'), 123)
|
|
|
|
def test_size_parser_with_unit_b(self):
|
|
self.assertEqual(media_split.convert_media_size('123b'), 123)
|
|
|
|
def test_size_parser_with_unit_k(self):
|
|
self.assertEqual(media_split.convert_media_size('123k'), 123 * 1024)
|
|
|
|
def test_size_parser_with_unit_M(self):
|
|
self.assertEqual(media_split.convert_media_size('123M'),
|
|
123 * 1024 * 1024)
|
|
|
|
def test_size_parser_with_unit_G(self):
|
|
self.assertEqual(media_split.convert_media_size('123G'),
|
|
123 * 1024 * 1024 * 1024)
|
|
|
|
def test_size_parser_with_negative_number(self):
|
|
with self.assertRaises(ValueError):
|
|
media_split.convert_media_size(-15)
|
|
|
|
def test_size_parser_with_unknown_unit(self):
|
|
with self.assertRaises(ValueError):
|
|
media_split.convert_media_size('123X')
|
|
|
|
|
|
class ConvertFileSizeTestCase(unittest.TestCase):
|
|
|
|
def test_round_up(self):
|
|
self.assertEqual(media_split.convert_file_size(123, 2048), 2048)
|
|
|
|
def test_exactly_block_size(self):
|
|
self.assertEqual(media_split.convert_file_size(100, 100), 100)
|
|
|
|
|
|
def bl(s):
|
|
return s * 2048
|
|
|
|
|
|
class MediaSplitterTestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.compose = mock.Mock()
|
|
|
|
def assertFreeSpace(self, free, total):
|
|
self.assertEqual(
|
|
self.compose.mock_calls,
|
|
[mock.call.log_debug('MediaSplitter: free space on single media would be %s. '
|
|
'Total size of single medium: %s.' % (free, total))])
|
|
|
|
def assertUnlimited(self, total):
|
|
self.assertEqual(
|
|
self.compose.mock_calls,
|
|
[mock.call.log_debug('MediaSplitter: Total size of single medium: %s.' % total)])
|
|
|
|
def test_sum_size(self):
|
|
ms = media_split.MediaSplitter(bl(100))
|
|
ms.add_file('first', bl(20))
|
|
ms.add_file('second', bl(30))
|
|
ms.add_file('third', 10)
|
|
|
|
self.assertEqual(ms.total_size, bl(50) + 10)
|
|
self.assertEqual(ms.total_size_in_blocks, bl(51))
|
|
|
|
def test_add_same_file_twice(self):
|
|
ms = media_split.MediaSplitter(bl(100))
|
|
ms.add_file('first', bl(20))
|
|
ms.add_file('first', bl(20))
|
|
|
|
self.assertEqual(ms.total_size, bl(20))
|
|
|
|
def test_add_same_file_twice_with_different_size(self):
|
|
ms = media_split.MediaSplitter(bl(100))
|
|
ms.add_file('first', bl(20))
|
|
with self.assertRaises(ValueError):
|
|
ms.add_file('first', bl(30))
|
|
|
|
def test_add_too_big_file(self):
|
|
ms = media_split.MediaSplitter(bl(100))
|
|
with self.assertRaises(ValueError):
|
|
ms.add_file('too-big', bl(300))
|
|
|
|
def test_fit_on_one(self):
|
|
ms = media_split.MediaSplitter(bl(100), compose=self.compose)
|
|
ms.add_file('first', bl(20))
|
|
ms.add_file('second', bl(30))
|
|
|
|
self.assertEqual(ms.split(),
|
|
[{'files': ['first', 'second'], 'size': bl(50)}])
|
|
self.assertFreeSpace(bl(50), bl(50))
|
|
|
|
def test_split_on_two_discs(self):
|
|
ms = media_split.MediaSplitter(bl(100), compose=self.compose)
|
|
ms.add_file('first', bl(25))
|
|
ms.add_file('second', bl(40))
|
|
ms.add_file('third', bl(80))
|
|
|
|
self.assertEqual(ms.split(),
|
|
[{'files': ['first', 'second'], 'size': bl(65)},
|
|
{'files': ['third'], 'size': bl(80)}])
|
|
self.assertFreeSpace(bl(100 - 25 - 40 - 80), bl(25 + 40 + 80))
|
|
|
|
def test_split_with_sticky_file(self):
|
|
ms = media_split.MediaSplitter(bl(100))
|
|
ms.add_file('sticky', bl(15), sticky=True)
|
|
ms.add_file('first', bl(25))
|
|
ms.add_file('second', bl(40))
|
|
ms.add_file('third', bl(80))
|
|
|
|
self.assertEqual(ms.split(),
|
|
[{'files': ['sticky', 'first', 'second'], 'size': bl(80)},
|
|
{'files': ['sticky', 'third'], 'size': bl(95)}])
|
|
|
|
def test_split_unlimited_media(self):
|
|
ms = media_split.MediaSplitter(None, compose=self.compose)
|
|
ms.add_file('first', bl(25))
|
|
ms.add_file('second', bl(40))
|
|
ms.add_file('third', bl(80))
|
|
|
|
self.assertEqual(ms.split(),
|
|
[{'files': ['first', 'second', 'third'], 'size': bl(145)}])
|
|
self.assertUnlimited(bl(25 + 40 + 80))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|