From 002eeb511d1b8bcd2d6531987a05a94025cbfd86 Mon Sep 17 00:00:00 2001
From: fei <204683769+feiiiiii5@users.noreply.github.com>
Date: Sun, 11 Oct 2026 02:14:38 +0800
Subject: [PATCH] Make the DEFAULT_TIMEOUT environment variable work in the CLI

The CLI passed it to validate_email() as default_timeout, a keyword the
function does not have, so setting it raised TypeError for every address.
Hand the value to the DNS resolver that actually uses it.
---
 email_validator/__main__.py | 13 ++++++++--
 tests/test_main.py          | 49 +++++++++++++++++++++++++++++++++++++
 2 files changed, 60 insertions(+), 2 deletions(-)

diff --git a/email_validator/__main__.py b/email_validator/__main__.py
index 107b08b..daa777d 100644
--- a/email_validator/__main__.py
+++ b/email_validator/__main__.py
@@ -36,11 +36,20 @@ def main(dns_resolver: Optional[_Resolver] = None) -> None:
             options[varname.lower()] = bool(os.environ[varname])
     for varname in ('DEFAULT_TIMEOUT',):
         if varname in os.environ:
-            options[varname.lower()] = float(os.environ[varname])
+            # The environment variable is named after the DEFAULT_TIMEOUT module
+            # constant, but the validate_email() keyword argument for the DNS
+            # timeout is `timeout`.
+            options['timeout'] = float(os.environ[varname])
+
+    # validate_email() does not accept `timeout` together with a `dns_resolver`,
+    # and a resolver passed in by the caller already carries its own timeout.
+    if dns_resolver is not None:
+        options.pop('timeout', None)
 
     if len(sys.argv) == 1:
         # Validate the email addresses passed line-by-line on STDIN.
-        dns_resolver = dns_resolver or caching_resolver()
+        # The same resolver is used for every address, so the timeout goes on it.
+        dns_resolver = dns_resolver or caching_resolver(timeout=options.pop('timeout', None))
         for line in sys.stdin:
             email = line.strip()
             try:
diff --git a/tests/test_main.py b/tests/test_main.py
index b8c6583..56668df 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -81,6 +81,55 @@ def test_main_multi_input(monkeypatch: pytest.MonkeyPatch, capsys: pytest.Captur
     assert test_cases[3] in stdout
 
 
+def test_main_default_timeout_env_arg(monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]) -> None:
+    # The DEFAULT_TIMEOUT environment variable is named after the module constant,
+    # but the validate_email() keyword argument for the DNS timeout is `timeout`.
+    # Regression test: the CLI crashed for every address when it was set.
+    import json
+    test_email = "google@google.com"
+    monkeypatch.setenv('DEFAULT_TIMEOUT', '5')
+    monkeypatch.setattr('sys.argv', ['email_validator', test_email])
+    validator_command_line_tool(dns_resolver=RESOLVER)
+    stdout, _ = capsys.readouterr()
+    output = json.loads(str(stdout))
+    assert output["original"] == test_email
+
+
+def test_main_default_timeout_env_stdin(monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]) -> None:
+    # Same as above, for the STDIN path, which uses a resolver for every line.
+    import io
+    test_cases = ["google1@google.com", "test@.com"]
+    test_input = io.StringIO("\n".join(test_cases))
+    monkeypatch.setenv('DEFAULT_TIMEOUT', '5')
+    monkeypatch.setattr('sys.stdin', test_input)
+    monkeypatch.setattr('sys.argv', ['email_validator'])
+    validator_command_line_tool(dns_resolver=RESOLVER)
+    stdout, _ = capsys.readouterr()
+    assert test_cases[0] not in stdout
+    assert test_cases[1] in stdout
+
+
+def test_main_default_timeout_env_sets_resolver_timeout(monkeypatch: pytest.MonkeyPatch, capsys: pytest.CaptureFixture[str]) -> None:
+    # On the STDIN path the CLI builds the resolver itself, so the timeout has to be
+    # handed to caching_resolver() rather than passed on to validate_email(), which
+    # refuses a timeout and a resolver together.
+    import io
+    captured: typing.Dict[str, typing.Any] = {}
+
+    def fake_caching_resolver(timeout: typing.Optional[float] = None) -> typing.Any:
+        captured['timeout'] = timeout
+        return RESOLVER
+
+    monkeypatch.setenv('DEFAULT_TIMEOUT', '5')
+    monkeypatch.setattr('email_validator.__main__.caching_resolver', fake_caching_resolver)
+    monkeypatch.setattr('sys.stdin', io.StringIO("google1@google.com\n"))
+    monkeypatch.setattr('sys.argv', ['email_validator'])
+    validator_command_line_tool()
+    stdout, _ = capsys.readouterr()
+    assert captured['timeout'] == 5.0
+    assert stdout == ''
+
+
 def test_bytes_input() -> None:
     input_email = b"testaddr@example.tld"
     valid_email = validate_email(input_email, check_deliverability=False)