split of modify function in modify_test

This commit is contained in:
Peter Kessen 2016-01-27 19:42:53 +01:00
parent d486b2a70c
commit 8954f1a494

View file

@ -158,10 +158,13 @@ class ModifyTest(unittest.TestCase, TestHelper):
def tearDown(self):
self.teardown_beets()
def modify(self, *args):
with control_stdin('y'):
def modify_inp(self, inp, *args):
with control_stdin(inp):
ui._raw_main(['modify'] + list(args), self.lib)
def modify(self, *args):
self.modify_inp('y', *args)
# Item tests
def test_modify_item(self):