def test_mount(self): # # None to mount # dev = None fs_type = 'ext4' option = '' self.assertRaises(Exception, main.mount, dev, fs_type, option) # # fstype undefine # dev = '/dev/Xda1' fs_type = None option = '' self.assertRaises(Exception, main.mount, dev, fs_type, option) # # mount failure # dev = '/dev/Xda1' fstype = 'ext4' options = '' with patch('tempfile.mkdtemp', return_value='/mnt'): self.assertRaises(Exception, main.mount, dev, fstype, options) # # mount successfully # def create_temp_directory(*args, **kwargs): return '/mnt' dev = '/dev/Xda1' fstype = 'ext4' options = '' patcher = patch('tempfile.mkdtemp') create_tmpdir = patcher.start() create_tmpdir.side_effect = create_temp_directory with patch.multiple( main, create_tmpdir, command_check_call=lambda cmd: True, ): main.mount(dev, fstype, options)