def test_img_list_gen_incorrect_ext(): """test img_list_gen func but with incorrect extension.""" m_ext = '.mp3' m_root = mock.Mock() m_file = mock.Mock() with mock.patch('melissa.actions.imgur_handler.os') as m_os: # pre run. m_os.walk.return_value = [(m_root, '', [m_file])] m_os.path.splitext.return_value = ['', m_ext] # run from melissa.actions import imgur_handler res = imgur_handler.img_list_gen() # test assert res == [] m_os.assert_has_calls([ mock.call.walk(M_IMAGES_PATH), mock.call.path.splitext(m_file) ])
def test_img_list_gen(m_ext): """test img_list_gen func.""" m_root = mock.Mock() m_file = mock.Mock() with mock.patch('melissa.profile_loader.load_profile'): with mock.patch('melissa.actions.imgur_handler.os') as m_os: # pre run. m_os.walk.return_value = [(m_root, '', [m_file])] m_os.path.splitext.return_value = ['', m_ext] # run from melissa.actions import imgur_handler imgur_handler.profile.data = { 'images_path': M_IMAGES_PATH } res = imgur_handler.img_list_gen() # test m_os.assert_has_calls([ mock.call.walk(M_IMAGES_PATH), mock.call.path.splitext(m_file), mock.call.path.join(m_root, m_file.lower.return_value) ]) assert res == [m_os.path.join.return_value]