From 31af88a81f3cf3f915e8016f72884147609c7897 Mon Sep 17 00:00:00 2001 From: Andrey Rakhmatullin Date: Sat, 20 Jun 2026 16:49:08 +0500 Subject: [PATCH] Implement more methods in CustomStatsCollector. --- tests/test_stats.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/test_stats.py b/tests/test_stats.py index 6e6aa0cc4..33ab75c97 100644 --- a/tests/test_stats.py +++ b/tests/test_stats.py @@ -141,10 +141,19 @@ class TestStatsCollector: def get_stats(self, spider=None): return self._stats + def get_value(self, key, default=None, spider=None): + return self._stats.get(key, default) + + def set_value(self, key, value, spider=None): + self._stats[key] = value + def inc_value(self, key, count=1, start=0, spider=None): d = self._stats d[key] = d.setdefault(key, start) + count + def max_value(self, key, value, spider=None) -> None: + self._stats[key] = max(self._stats.setdefault(key, value), value) + def close_spider(self, spider, reason): pass