def test_extract_urls_from_units(all_units):
"""
Make sure that urls are grabbed from both mp4_urls and from
resources_urls of Unit class.
"""
urls = edx_dl.extract_urls_from_units(all_units, "%(url)s")
expected = ["1\n", "2\n", "3\n"]
assert sorted(urls) == sorted(expected)
def test_extract_urls_from_units(all_units):
"""
Make sure that urls are grabbed from both mp4_urls and from
resources_urls of Unit class.
"""
urls = edx_dl.extract_urls_from_units(all_units, '%(url)s')
expected = ['1\n', '2\n', '3\n']
assert sorted(urls) == sorted(expected)
def test_extract_urls_from_units_unknown_videos(unknown_videos):
"""
Make sure that we only expect Video in the list of Unit videos.
"""
with pytest.raises(TypeError):
edx_dl.extract_urls_from_units(unknown_videos, "%(url)s")
def test_extract_urls_from_units_unknown_videos(unknown_videos):
"""
Make sure that we only expect Video in the list of Unit videos.
"""
with pytest.raises(TypeError):
edx_dl.extract_urls_from_units(unknown_videos, '%(url)s')
def test_extract_urls_from_units_unknown_units(unknown_units):
"""
Make sure that we only expect Units in the list of units.
"""
with pytest.raises(TypeError):
edx_dl.extract_urls_from_units(unknown_units, '%(url)s')
