core: Update tests with cases for filtering out None results.
This commit is contained in:
parent
5dd7f4b07a
commit
f1bd092e63
@ -90,6 +90,22 @@ class CoreLibraryTest(unittest.TestCase):
|
|||||||
self.library1.find_exact.assert_called_once_with(any=['a'])
|
self.library1.find_exact.assert_called_once_with(any=['a'])
|
||||||
self.library2.find_exact.assert_called_once_with(any=['a'])
|
self.library2.find_exact.assert_called_once_with(any=['a'])
|
||||||
|
|
||||||
|
def test_find_exact_filters_out_none(self):
|
||||||
|
track1 = Track(uri='dummy1:a')
|
||||||
|
result1 = SearchResult(tracks=[track1])
|
||||||
|
|
||||||
|
self.library1.find_exact().get.return_value = result1
|
||||||
|
self.library1.find_exact.reset_mock()
|
||||||
|
self.library2.find_exact().get.return_value = None
|
||||||
|
self.library2.find_exact.reset_mock()
|
||||||
|
|
||||||
|
result = self.core.library.find_exact(any=['a'])
|
||||||
|
|
||||||
|
self.assertIn(result1, result)
|
||||||
|
self.assertNotIn(None, result)
|
||||||
|
self.library1.find_exact.assert_called_once_with(any=['a'])
|
||||||
|
self.library2.find_exact.assert_called_once_with(any=['a'])
|
||||||
|
|
||||||
def test_find_accepts_query_dict_instead_of_kwargs(self):
|
def test_find_accepts_query_dict_instead_of_kwargs(self):
|
||||||
track1 = Track(uri='dummy1:a')
|
track1 = Track(uri='dummy1:a')
|
||||||
track2 = Track(uri='dummy2:a')
|
track2 = Track(uri='dummy2:a')
|
||||||
@ -126,6 +142,22 @@ class CoreLibraryTest(unittest.TestCase):
|
|||||||
self.library1.search.assert_called_once_with(any=['a'])
|
self.library1.search.assert_called_once_with(any=['a'])
|
||||||
self.library2.search.assert_called_once_with(any=['a'])
|
self.library2.search.assert_called_once_with(any=['a'])
|
||||||
|
|
||||||
|
def test_search_filters_out_none(self):
|
||||||
|
track1 = Track(uri='dummy1:a')
|
||||||
|
result1 = SearchResult(tracks=[track1])
|
||||||
|
|
||||||
|
self.library1.search().get.return_value = result1
|
||||||
|
self.library1.search.reset_mock()
|
||||||
|
self.library2.search().get.return_value = None
|
||||||
|
self.library2.search.reset_mock()
|
||||||
|
|
||||||
|
result = self.core.library.search(any=['a'])
|
||||||
|
|
||||||
|
self.assertIn(result1, result)
|
||||||
|
self.assertNotIn(None, result)
|
||||||
|
self.library1.search.assert_called_once_with(any=['a'])
|
||||||
|
self.library2.search.assert_called_once_with(any=['a'])
|
||||||
|
|
||||||
def test_search_accepts_query_dict_instead_of_kwargs(self):
|
def test_search_accepts_query_dict_instead_of_kwargs(self):
|
||||||
track1 = Track(uri='dummy1:a')
|
track1 = Track(uri='dummy1:a')
|
||||||
track2 = Track(uri='dummy2:a')
|
track2 = Track(uri='dummy2:a')
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user