ariadne/tests/test_name_generator.py

11 lines
359 B
Python
Raw Permalink Normal View History

from __future__ import annotations
from ariadne.utils.name_generator import NameGenerator
def test_name_generator_returns_none_after_exhausting_attempts(monkeypatch) -> None:
generator = NameGenerator(max_attempts=2)
monkeypatch.setattr(NameGenerator, "generate", lambda self: "already-used")
assert generator.unique({"already-used"}) is None