"""Tests for the SOCKS5 proxy HTTP module."""

import ssl
import urllib.request

from socks import SOCKS5

from derp.http import _PROXY_ADDR, _PROXY_PORT, _ProxyHandler, build_opener


class TestProxyHandler:
    def test_uses_socks5(self):
        handler = _ProxyHandler()
        assert handler.args[0] == SOCKS5

    def test_proxy_address(self):
        handler = _ProxyHandler()
        assert handler.args[1] == _PROXY_ADDR
        assert handler.args[2] == _PROXY_PORT

    def test_rdns_enabled(self):
        handler = _ProxyHandler()
        assert handler.args[3] is True

    def test_default_ssl_context(self):
        handler = _ProxyHandler()
        assert isinstance(handler._ssl_context, ssl.SSLContext)

    def test_custom_ssl_context(self):
        ctx = ssl.create_default_context()
        ctx.check_hostname = False
        handler = _ProxyHandler(context=ctx)
        assert handler._ssl_context is ctx

    def test_is_https_handler(self):
        handler = _ProxyHandler()
        assert isinstance(handler, urllib.request.HTTPSHandler)


class TestBuildOpener:
    def test_includes_proxy_handler(self):
        opener = build_opener()
        proxy = [h for h in opener.handlers if isinstance(h, _ProxyHandler)]
        assert len(proxy) == 1

    def test_passes_extra_handlers(self):
        class Custom(urllib.request.HTTPRedirectHandler):
            pass

        opener = build_opener(Custom)
        custom = [h for h in opener.handlers if isinstance(h, Custom)]
        assert len(custom) == 1

    def test_passes_ssl_context(self):
        ctx = ssl.create_default_context()
        ctx.check_hostname = False
        opener = build_opener(context=ctx)
        proxy = [h for h in opener.handlers if isinstance(h, _ProxyHandler)][0]
        assert proxy._ssl_context is ctx