From 1cb23e5c5c1953f5f0ad91784b5a19d530c08ba6 Mon Sep 17 00:00:00 2001
From: soergeld <soergeldorian@gmail.com>
Date: Mon, 13 Jul 2020 21:48:06 +0200
Subject: [PATCH] no hybrid mocking, two classes

---
 test/test_parentwork.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/test/test_parentwork.py b/test/test_parentwork.py
index 7db3acdcb..395e6c107 100644
--- a/test/test_parentwork.py
+++ b/test/test_parentwork.py
@@ -60,8 +60,6 @@ def mock_workid_response(mbid):
         return dp_work
     elif mbid == '3':
         return p_work
-    else:
-        return musicbrainzngs.get_work_by_id(mbid)
 
 
 class ParentWorkTest(unittest.TestCase, TestHelper):
@@ -132,6 +130,17 @@ class ParentWorkTest(unittest.TestCase, TestHelper):
         self.assertEqual(u'45afb3b2-18ac-4187-bc72-beb1b1c194ba',
                          parentwork.work_parent_id(mb_workid)[0])
 
+
+class ParentWorkTest_mock(unittest.TestCase, TestHelper):
+    def setUp(self):
+        """Set up configuration"""
+        self.setup_beets()
+        self.load_plugins('parentwork')
+
+    def tearDown(self):
+        self.unload_plugins()
+        self.teardown_beets()
+
     @mock.patch('musicbrainzngs.get_work_by_id',
                 side_effect=mock_workid_response)
     def test_normal_case(self):