tests/unit/test_cli.py

@@ -0,0 +1,24 @@
+import pytest
+
+from scrape_movies.presentation import cli, soup_steps
+
+
+def test_main_when_soup_step_is_given_prints_output(
+    capsys: pytest.CaptureFixture[str],
+) -> None:
+    for step in soup_steps.STEPS:
+        assert cli.main(["soup", step]) == 0
+        assert capsys.readouterr().out.strip(), step
+
+
+def test_main_when_no_command_prints_help(
+    capsys: pytest.CaptureFixture[str],
+) -> None:
+    assert cli.main([]) == 0
+    assert "usage:" in capsys.readouterr().out
+
+
+def test_render_step_when_title_returns_page_title() -> None:
+    assert soup_steps.render_step("title") == (
+        "soup.title.string -> Angela's Personal Site"
+    )

tests/unit/test_soup_basics.py

@@ -0,0 +1,65 @@
+from bs4 import BeautifulSoup
+
+from scrape_movies.constants import WEBSITE_HTML_PATH
+from scrape_movies.domain.link import Link
+from scrape_movies.infrastructure import soup_basics
+
+
+def _soup() -> BeautifulSoup:
+    return soup_basics.make_soup(soup_basics.load_html(WEBSITE_HTML_PATH))
+
+
+def test_get_title_when_page_has_title_returns_text() -> None:
+    assert soup_basics.get_title(_soup()) == "Angela's Personal Site"
+
+
+def test_get_title_when_page_has_no_title_returns_none() -> None:
+    assert soup_basics.get_title(soup_basics.make_soup("<p>x</p>")) is None
+
+
+def test_prettify_html_when_called_puts_tags_on_own_lines() -> None:
+    pretty = soup_basics.prettify_html(soup_basics.make_soup("<p><b>x</b></p>"))
+    assert pretty.splitlines()[:3] == ["<p>", " <b>", "  x"]
+
+
+def test_find_all_anchors_when_page_has_links_returns_text_and_url() -> None:
+    links = soup_basics.find_all_anchors(_soup())
+    assert links[0] == Link("The App Brewery", "https://www.appbrewery.co/")
+    assert [link.text for link in links] == [
+        "The App Brewery",
+        "My Hobbies",
+        "Contact Me",
+    ]
+
+
+def test_find_all_anchors_when_anchor_has_no_href_skips_it() -> None:
+    soup = soup_basics.make_soup("<a>no link</a><a href='/x'>x</a>")
+    assert soup_basics.find_all_anchors(soup) == [Link("x", "/x")]
+
+
+def test_find_text_by_id_when_id_exists_returns_text() -> None:
+    assert soup_basics.find_text_by_id(_soup(), "name") == "Angela Yu"
+
+
+def test_find_text_by_id_when_id_is_missing_returns_none() -> None:
+    assert soup_basics.find_text_by_id(_soup(), "nobody") is None
+
+
+def test_find_all_texts_by_class_when_class_matches_returns_texts() -> None:
+    texts = soup_basics.find_all_texts_by_class(_soup(), "h3", "heading")
+    assert texts == ["Books and Teaching", "Other Pages"]
+
+
+def test_select_one_text_when_nested_selector_returns_first_match() -> None:
+    assert soup_basics.select_one_text(_soup(), "p a") == "The App Brewery"
+
+
+def test_select_one_text_when_nothing_matches_returns_none() -> None:
+    assert soup_basics.select_one_text(_soup(), "table") is None
+
+
+def test_select_texts_when_class_selector_returns_all_matches() -> None:
+    assert soup_basics.select_texts(_soup(), ".heading") == [
+        "Books and Teaching",
+        "Other Pages",
+    ]