Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-07 18:24:30 -04:00
Code Issues Packages Projects Releases Wiki Activity

MTP: Add mem leak test for file operations

Browse Source
This commit is contained in:
Kovid Goyal 2012-08-24 16:33:24 +05:30
parent 8fed10364f
commit b8f45adf28
1 changed files with 40 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
src/calibre/devices/mtp/test.py
View File

@ -180,7 +180,7 @@ class TestDeviceInteraction(unittest.TestCase):
@unittest.skipUnless(iswindows or islinux, 'Can only test for leaks on windows and linux')
def test_memory_leaks(self):
''' Test for memory leaks in the C modules '''
''' Test for memory leaks in the C module '''
self.check_setup()
# Test device scanning
@ -188,7 +188,7 @@ class TestDeviceInteraction(unittest.TestCase):
self.dev.detect_managed_devices, self.scanner.devices,
force_refresh=True)
used_by_many = self.measure_memory_usage(1000,
used_by_many = self.measure_memory_usage(100,
self.dev.detect_managed_devices, self.scanner.devices,
force_refresh=True)
@ -196,6 +196,42 @@ class TestDeviceInteraction(unittest.TestCase):
msg='Memory consumption during device scan: for one: %g for many:%g'%
(used_by_one, used_by_many))
# Test file transfer
size = 1024*100
raw = io.BytesIO(b'a'*size)
raw.seek(0)
name = 'zzz-test-file.txt'
def send_file(storage, name, raw, size):
raw.seek(0)
pc = ProgressCallback()
f = self.dev.put_file(storage, name, raw, size, callback=pc)
self.cleanup.append(f)
del pc
used_once = self.measure_memory_usage(1, send_file, self.storage, name,
raw, size)
used_many = self.measure_memory_usage(10, send_file, self.storage, name,
raw, size)
self.assertTrue(used_many <= used_once,
msg='Memory consumption during put_file: for one: %g for many:%g'%
(used_once, used_many))
def get_file(f):
raw = io.BytesIO()
pc = ProgressCallback()
self.dev.get_file(f, raw, callback=pc)
del raw
del pc
f = self.storage.file_named(name)
used_once = self.measure_memory_usage(1, get_file, f)
used_many = self.measure_memory_usage(10, get_file, f)
self.assertTrue(used_many <= used_once,
msg='Memory consumption during get_file: for one: %g for many:%g'%
(used_once, used_many))
# Test get_filesystem
used_by_one = self.measure_memory_usage(1,
self.dev.dev.get_filesystem, self.storage.object_id)
@ -207,9 +243,10 @@ class TestDeviceInteraction(unittest.TestCase):
msg='Memory consumption during get_filesystem: for one: %g for many:%g'%
(used_by_one, used_by_many))
def tests():
tl = unittest.TestLoader()
# return tl.loadTestsFromName('test.TestDeviceInteraction.test_file_transfer')
# return tl.loadTestsFromName('test.TestDeviceInteraction.test_memory_leaks')
return tl.loadTestsFromTestCase(TestDeviceInteraction)
def run():