scrapy/tests/test_addons/test_builtins.py

24 lines
965 B
Python

import unittest
import scrapy
import scrapy.addons
from scrapy.addons.builtins import make_builtin_addon
from scrapy.settings import Settings
class BuiltinAddonsTest(unittest.TestCase):
def test_make_builtin_addon(self):
httpcache = make_builtin_addon("httpcache", {"enabled": True})
self.assertEqual(httpcache.name, "httpcache")
self.assertEqual(httpcache.default_config, {"enabled": True})
self.assertEqual(httpcache.version, scrapy.__version__)
httpcache = make_builtin_addon("httpcache", {"enabled": True}, "99.9")
self.assertEqual(httpcache.version, "99.9")
def test_defaultheaders_export_config(self):
settings = Settings()
dh = scrapy.addons.defaultheaders()
dh.export_config({"X-Test-Header": "val"}, settings)
self.assertIn("X-Test-Header", settings["DEFAULT_REQUEST_HEADERS"])
self.assertEqual(settings["DEFAULT_REQUEST_HEADERS"]["X-Test-Header"], "val")