diff options
Diffstat (limited to 'test/test_netrc.py')
-rw-r--r-- | test/test_netrc.py | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/test/test_netrc.py b/test/test_netrc.py index 50b9e5b..c7f5272 100644 --- a/test/test_netrc.py +++ b/test/test_netrc.py @@ -7,18 +7,19 @@ import unittest sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) -from hypervideo_dl.extractor import ( - gen_extractors, -) +from hypervideo_dl.extractor import gen_extractor_classes +from hypervideo_dl.extractor.common import InfoExtractor + +NO_LOGIN = InfoExtractor._perform_login class TestNetRc(unittest.TestCase): def test_netrc_present(self): - for ie in gen_extractors(): - if not hasattr(ie, '_login'): + for ie in gen_extractor_classes(): + if ie._perform_login is NO_LOGIN: continue self.assertTrue( - hasattr(ie, '_NETRC_MACHINE'), + ie._NETRC_MACHINE, 'Extractor %s supports login, but is missing a _NETRC_MACHINE property' % ie.IE_NAME) |