1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
# vim: ts=4:sw=4:expandtab
# BleachBit
# Copyright (C) 2008-2025 Andrew Ziem
# https://www.bleachbit.org
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
Test case for Command
"""
import os
from unittest import mock
from tests import common
from bleachbit import FileUtilities
from bleachbit.Command import Delete, Function, Shred
class CommandTestCase(common.BleachbitTestCase):
"""Test case for Command"""
def test_Delete(self, cls=Delete):
"""Unit test for Delete"""
path = self.write_file('test_Delete', b'foo')
cmd = cls(path)
self.assertExists(path)
# preview
ret = next(cmd.execute(really_delete=False))
self.assertGreater(ret['size'], 0)
self.assertEqual(ret['path'], path)
self.assertExists(path)
# delete
ret = next(cmd.execute(really_delete=True))
self.assertGreater(ret['size'], 0)
self.assertEqual(ret['path'], path)
self.assertNotExists(path)
def test_Function(self):
"""Unit test for Function"""
path = self.write_file('test_Function', b'foo')
cmd = Function(path, FileUtilities.delete, 'bar')
self.assertExists(path)
self.assertGreater(os.path.getsize(path), 0)
# preview
ret = next(cmd.execute(False))
self.assertExists(path)
self.assertGreater(os.path.getsize(path), 0)
# delete
ret = next(cmd.execute(True))
self.assertGreater(ret['size'], 0)
self.assertEqual(ret['path'], path)
self.assertNotExists(path)
def test_Function_no_collation(self):
"""Unit test for Function with no collation
See https://github.com/bleachbit/bleachbit/issues/1866
"""
path = self.write_file('test_Function_no_collation', b'')
cmd = Function(path,
lambda p: FileUtilities.execute_sqlite3(
p, 'CREATE TABLE test (name TEXT COLLATE foo);'),
'test_no_collation')
with mock.patch('bleachbit.Command.logger.debug') as mock_debug:
with self.assertRaises(StopIteration):
next(cmd.execute(True))
mock_debug.assert_called_with(mock.ANY)
def test_Shred(self):
"""Unit test for Shred"""
self.test_Delete(Shred)
|
