Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2026-01-02 14:03:12 +01:00
Code Issues Projects Releases Wiki Activity

Tear down Item.write mock

Browse source
This commit is contained in:
Adrian Sampson 2015-11-17 10:26:36 -08:00
parent 0d459752d9
commit 8e9d335a87
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
test/test_edit.py
View file

@ -16,12 +16,10 @@ from __future__ import (division, absolute_import, print_function,
unicode_literals)
import codecs
from mock import Mock, patch
from mock import patch
from test._common import unittest
from test.helper import TestHelper, control_stdin
from beets import library
class ModifyFileMocker(object):
"""Helper for modifying a file, replacing or editing its contents. Used for
@ -84,9 +82,11 @@ class EditCommandTest(unittest.TestCase, TestHelper):
item in self.album.items()]
# keep track of write()s
library.Item.write = Mock()
self.write_patcher = patch('beets.library.Item.write')
self.mock_write = self.write_patcher.start()
def tearDown(self):
self.write_patcher.stop()
self.teardown_beets()
self.unload_plugins()
@ -103,7 +103,7 @@ class EditCommandTest(unittest.TestCase, TestHelper):
"""Several common assertions on Album, Track and call counts."""
self.assertEqual(len(self.lib.albums()), album_count)
self.assertEqual(len(self.lib.items()), track_count)
self.assertEqual(library.Item.write.call_count, write_call_count)
self.assertEqual(self.mock_write.call_count, write_call_count)
self.assertTrue(all(i.title.startswith(title_starts_with)
for i in self.lib.items()))