TheLovinator/discord-rss-bot
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add type hints and other small stuff

Browse Source
This commit is contained in:
Joakim Hellsén
2022-12-14 02:16:33 +01:00
parent 1fcb9ffbf8
commit 3dbe6ba79d
8 changed files with 65 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_search.py
View File

@ -1,10 +1,11 @@
import os
import tempfile
from pathlib import Path
from typing import Iterable
from reader import make_reader
from reader import EntrySearchResult, Feed, Reader, make_reader
from discord_rss_bot.search import add_span_with_slice, create_html_for_search_results
from discord_rss_bot.search import create_html_for_search_results
def test_create_html_for_search_results() -> None:
@ -16,7 +17,7 @@ def test_create_html_for_search_results() -> None:
assert os.path.exists(temp_dir)
# Create a temporary reader.
reader = make_reader(url=str(Path(temp_dir, "test_db.sqlite")))
reader: Reader = make_reader(url=str(Path(temp_dir, "test_db.sqlite")))
assert reader is not None
# Add a feed to the reader.
@ -26,7 +27,7 @@ def test_create_html_for_search_results() -> None:
reader.update_feeds()
# Get the feed.
feed = reader.get_feed("https://www.reddit.com/r/Python/.rss")
feed: Feed = reader.get_feed("https://www.reddit.com/r/Python/.rss")
assert feed is not None
# Update the search index.
@ -34,7 +35,7 @@ def test_create_html_for_search_results() -> None:
reader.update_search()
# Get the HTML for the search results.
search_results = reader.search_entries("a", feed=feed)
search_results: Iterable[EntrySearchResult] = reader.search_entries("a", feed=feed)
# Create the HTML and check if it is not empty.
search_html: str = create_html_for_search_results(search_results, reader)