trex Bot
d8034278d8
- read [ignored_tags] instead of [ignored_domains], so tag-based ignoring
actually works
- match ignored tags as whole tags, not substrings ('ipt' no longer matches
'iptorrents')
- send the login-error notification through the Pushover client instead of
the non-existent module-level send_message()
- handle an empty torrent list instead of raising IndexError before any
summary is sent
- tag a new torrent once instead of once per code path
- honour toml_path without DOCKER in AppriseClient
- close replaced log handlers so the file descriptor is released
Also add the missing [telemetry] section to config.toml.example, which
qbit-maid.py requires at startup.
The tests that pinned each bug now assert the fixed behaviour, and the
suite still reports 100% coverage.
131 lines
5.3 KiB
Python
131 lines
5.3 KiB
Python
"""Tests for qlist.build_tor_list — the tagging and sorting pass."""
|
|
import unittest
|
|
|
|
from qbt_fakes import FakeQbt, torrent
|
|
from qlist import build_tor_list, torrent_tag_for
|
|
|
|
|
|
class TestTorrentTagFor(unittest.TestCase):
|
|
def test_protected_tracker_gets_the_protected_tag(self):
|
|
self.assertEqual(
|
|
torrent_tag_for("https://tracker.example/a", "ipt", "public",
|
|
["tracker.example"]),
|
|
"ipt",
|
|
)
|
|
|
|
def test_unknown_tracker_gets_the_public_tag(self):
|
|
self.assertEqual(
|
|
torrent_tag_for("https://other.example/a", "ipt", "public",
|
|
["tracker.example"]),
|
|
"public",
|
|
)
|
|
|
|
def test_blank_tracker_gets_no_tag(self):
|
|
self.assertIsNone(
|
|
torrent_tag_for("", "ipt", "public", ["tracker.example"])
|
|
)
|
|
|
|
|
|
class TestBuildTorList(unittest.TestCase):
|
|
def run_build(self, torrents, **overrides):
|
|
qbt = FakeQbt(torrent_list=list(torrents), **overrides).reset_counters()
|
|
build_tor_list(qbt)
|
|
return qbt
|
|
|
|
def test_sets_total_torrents(self):
|
|
qbt = self.run_build([torrent(infohash="a"), torrent(infohash="b")])
|
|
self.assertEqual(qbt.total_torrents, 2)
|
|
|
|
def test_blank_tracker_is_ignored(self):
|
|
qbt = self.run_build([torrent(tracker="")])
|
|
self.assertEqual(qbt.ignored_counter, 1)
|
|
self.assertEqual(qbt.tracker_list, [])
|
|
|
|
def test_ignored_category_is_ignored(self):
|
|
qbt = self.run_build(
|
|
[torrent(category="tech")],
|
|
cat_whitelist={"tech": "tech"},
|
|
)
|
|
self.assertEqual(qbt.ignored_counter, 1)
|
|
self.assertEqual(qbt.tracker_list, [])
|
|
|
|
def test_ignored_tag_is_ignored(self):
|
|
qbt = self.run_build(
|
|
[torrent(tags="save,public")],
|
|
ignored_tags={"save": "save"},
|
|
)
|
|
self.assertEqual(qbt.ignored_counter, 1)
|
|
self.assertEqual(qbt.tracker_list, [])
|
|
|
|
def test_ignored_tag_does_not_match_a_longer_tag(self):
|
|
qbt = self.run_build(
|
|
[torrent(tags="iptorrents", tracker="https://other.example/a")],
|
|
ignored_tags={"ipt": "ipt"},
|
|
)
|
|
self.assertEqual(qbt.ignored_counter, 0)
|
|
self.assertEqual(len(qbt.tracker_list), 1)
|
|
|
|
def test_blank_tag_on_protected_tracker_gets_protected_tag(self):
|
|
qbt = self.run_build([torrent(tags="", infohash="h1")])
|
|
self.assertIn(("ipt", "h1"), qbt.qbt_client.added_tags)
|
|
self.assertEqual(len(qbt.tracker_list), 1)
|
|
|
|
def test_blank_tag_on_non_protected_tracker_gets_public_tag(self):
|
|
qbt = self.run_build(
|
|
[torrent(tags="", tracker="https://other.example/announce", infohash="h2")]
|
|
)
|
|
self.assertIn(("public", "h2"), qbt.qbt_client.added_tags)
|
|
self.assertEqual(len(qbt.tracker_list), 1)
|
|
|
|
def test_blank_tag_is_applied_once_for_non_premature_torrents(self):
|
|
"""A blank-tag torrent is tagged once, not once per code path."""
|
|
qbt = self.run_build([torrent(tags="", infohash="h1")])
|
|
self.assertEqual(qbt.qbt_client.added_tags, [("ipt", "h1")])
|
|
|
|
def test_premature_blank_tag_torrent_is_tagged_once(self):
|
|
"""Premature torrents are tagged by the first block, then skipped."""
|
|
qbt = self.run_build([torrent(tags="", seeding_time=1, infohash="h3")])
|
|
self.assertEqual(qbt.qbt_client.added_tags, [("ipt", "h3")])
|
|
self.assertEqual(qbt.preme_tor_counter, 1)
|
|
self.assertEqual(qbt.tracker_list, [])
|
|
|
|
def test_premature_torrent_is_counted_and_not_queued(self):
|
|
qbt = self.run_build([torrent(seeding_time=10, tags="ipt")])
|
|
self.assertEqual(qbt.preme_tor_counter, 1)
|
|
self.assertEqual(qbt.tracker_list, [])
|
|
|
|
def test_protected_torrent_with_tag_is_queued_without_retagging(self):
|
|
qbt = self.run_build([torrent(tags="ipt")])
|
|
self.assertEqual(qbt.qbt_client.added_tags, [])
|
|
self.assertEqual(len(qbt.tracker_list), 1)
|
|
|
|
def test_non_protected_torrent_with_tag_is_queued_without_retagging(self):
|
|
qbt = self.run_build(
|
|
[torrent(tags="public", tracker="https://other.example/announce")]
|
|
)
|
|
self.assertEqual(qbt.qbt_client.added_tags, [])
|
|
self.assertEqual(len(qbt.tracker_list), 1)
|
|
|
|
def test_mixed_batch_counts_each_bucket(self):
|
|
qbt = self.run_build([
|
|
torrent(tracker=""), # ignored
|
|
torrent(category="tech"), # ignored
|
|
torrent(tags="save"), # ignored
|
|
torrent(seeding_time=1, tags="ipt"), # premature
|
|
torrent(tags="ipt", infohash="p1"), # queued
|
|
torrent(tags="public", tracker="https://o.example/a",
|
|
infohash="n1"), # queued
|
|
], cat_whitelist={"tech": "tech"}, ignored_tags={"save": "save"})
|
|
self.assertEqual(qbt.ignored_counter, 3)
|
|
self.assertEqual(qbt.preme_tor_counter, 1)
|
|
self.assertEqual(len(qbt.tracker_list), 2)
|
|
self.assertEqual(qbt.total_torrents, 6)
|
|
|
|
def test_consumes_the_input_list(self):
|
|
qbt = self.run_build([torrent(tags="ipt")])
|
|
self.assertEqual(qbt.torrent_list, [])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|