def setup(self, mock_size):
size = mock.Mock()
size.accumulate_mbyte_file_sizes = mock.Mock(return_value=42)
size.customize = mock.Mock(return_value=42)
mock_size.return_value = size
self.xml_state = mock.Mock()
self.xml_state.get_build_type_name = mock.Mock(return_value='ext4')
self.setup = FileSystemSetup(self.xml_state, 'root_dir')
def test_setup_with_pxe_type(self):
self.xml_state.get_build_type_name = mock.Mock(return_value='pxe')
self.xml_state.build_type.get_filesystem = mock.Mock(
return_value='xfs')
setup = FileSystemSetup(self.xml_state, 'root_dir')
assert setup.requested_filesystem == 'xfs'
