class TestDownloadInteractiveHandler(unittest.TestCase): def setUp(self): self.handler = DownloadInteractiveHandler(None) self.handler._download = Mock() def test_only_one_sub(self): sub = {'lang': 'fr', 'source': 'test'} self.handler._handle([sub], ['fr', 'en']) self.handler._download.assert_called_with(sub) @patch(MOCK_INPUT) def test_with_choice(self, input_): input_.return_value = '2' subs = [ { 'lang': 'fr', 'source': 'test' }, { 'lang': 'en', 'source': 'test1' }, ] self.handler._handle(subs, ['fr', 'en']) self.handler._download.assert_called_with({ 'lang': 'en', 'source': 'test1' })
class TestDownloadInteractiveHandler(unittest.TestCase): def setUp(self): self.handler = DownloadInteractiveHandler(None) self.handler._download = Mock() def test_only_one_sub(self): sub = {'lang': 'fr', 'source': 'test'} self.handler._handle([sub], ['fr', 'en']) self.handler._download.assert_called_with(sub) @patch(MOCK_INPUT) def test_with_choice(self, input_): input_.return_value = '2' subs = [ {'lang': 'fr', 'source': 'test'}, {'lang': 'en', 'source': 'test1'}, ] self.handler._handle(subs, ['fr', 'en']) self.handler._download.assert_called_with({'lang': 'en', 'source': 'test1'})
def setUp(self): self.handler = DownloadInteractiveHandler(None) self.handler._download = Mock()