From aaefd023eb5ef0c397fbacaf1beff9ff20447e3f Mon Sep 17 00:00:00 2001 From: Thomas Scholtes Date: Tue, 28 Jan 2014 14:35:20 +0100 Subject: [PATCH] Extract choose tracks function --- test/test_importer.py | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/test/test_importer.py b/test/test_importer.py index 5073640d6..fc79e9746 100644 --- a/test/test_importer.py +++ b/test/test_importer.py @@ -130,6 +130,9 @@ class ImportHelper(object): def _choose_asis(self, task): return importer.action.ASIS + def _choose_as_track(self, task): + return importer.action.TRACKS + def _choose_candidate(self, task): return task.candidates[0] @@ -308,11 +311,8 @@ class ImportTest(_common.TestCase, ImportHelper): self.assertNotExists(import_file) def test_apply_tracks_adds_singleton(self): - def choose_as_tracks(task): return importer.action.TRACKS - self.importer.choose_match = choose_as_tracks - def choose_first_item(task): - return task.candidates[0] - self.importer.choose_item = choose_first_item + self.importer.choose_match = self._choose_as_track + self.importer.choose_item = self._choose_candidate self.assertEqual(self.lib.items().get(), None) self.assertEqual(self.lib.albums().get(), None) @@ -321,11 +321,8 @@ class ImportTest(_common.TestCase, ImportHelper): self.assertEqual(self.lib.albums().get(), None) def test_apply_tracks_adds_singleton_path(self): - def choose_as_tracks(task): return importer.action.TRACKS - self.importer.choose_match = choose_as_tracks - def choose_first_item(task): - return task.candidates[0] - self.importer.choose_item = choose_first_item + self.importer.choose_match = self._choose_as_track + self.importer.choose_item = self._choose_candidate self.assert_lib_dir_empty() self.importer.run()