Testing for providers requiring authorization (#11691)

* Set provider attributes for testing

* Add test cases for Animebytes

tests/providers/conftest.py

@@ -14,12 +14,12 @@
 
 
 def get_providers():
-    from medusa.providers.torrent import (anidex, btdb, limetorrents, nyaa,
+    from medusa.providers.torrent import (anidex, animebytes, btdb, limetorrents, nyaa,
                                           rarbg, shanaproject, thepiratebay,
                                           tokyotoshokan, torrentz2, zooqle)
     from medusa.providers.nzb import anizb, binsearch
 
-    return (anidex, btdb, limetorrents, nyaa, rarbg, shanaproject,
+    return (anidex, animebytes, btdb, limetorrents, nyaa, rarbg, shanaproject,
             thepiratebay, tokyotoshokan, torrentz2, zooqle, anizb, binsearch)
 
 
@@ -40,6 +40,11 @@ def get_provider_data():
         # Update provider with test data
         provider.data.update(test_data)
 
+        # Configure provider by overriding instance attributes
+        if test_data.get("_meta", {}).get("attr"):
+            for k, v in test_data["_meta"]["attr"].items():
+                setattr(provider.klass, k, v)
+
     return providers
 
 

tests/providers/test_search.py

@@ -28,7 +28,12 @@ def search(search_type, provider):
     cassette_filename = '{0}_{1}.yaml'.format(provider.name, search_type)
     cassette_path = os.path.join(__location__, provider.type, provider.name,
                                  cassette_filename)
-    with vcr.use_cassette(cassette_path, record_mode=record_mode):
+    if provider.data.get("_meta", {}).get("vcr"):
+        vcr_config = provider.data["_meta"]["vcr"]
+    else:
+        vcr_config = {}
+
+    with vcr.use_cassette(cassette_path, record_mode=record_mode, **vcr_config):
         actual = provider.klass.search(test_case['search_strings'])
 
     # Check if we got any results